Hugging Face's logo

Hemg
/
Melanoma-Cancer-Image-Classification

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Model card Files
xet
Metrics Community
Hemg commited on
Commit
fca5583
·
verified ·
1 Parent(s): ad2db7e

Model save

Browse files
Files changed (3) hide show
  1. README.md +19 -11
  2. model.safetensors +1 -1
  3. runs/Mar18_06-14-26_c66e66e08f91/events.out.tfevents.1710742467.c66e66e08f91.34.0 +2 -2
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.1937
21
- - Accuracy: 0.9264
22
 
23
  ## Model description
24
 
@@ -46,21 +46,29 @@ The following hyperparameters were used during training:
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
- - num_epochs: 8
50
  - mixed_precision_training: Native AMP
51
 
52
  ### Training results
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
56
- | 0.4884 | 0.99 | 68 | 0.2696 | 0.8923 |
57
- | 0.2361 | 1.99 | 137 | 0.2285 | 0.905 |
58
- | 0.1927 | 3.0 | 206 | 0.2121 | 0.9182 |
59
- | 0.1673 | 4.0 | 275 | 0.2066 | 0.9114 |
60
- | 0.1498 | 4.99 | 343 | 0.2038 | 0.9209 |
61
- | 0.1243 | 5.99 | 412 | 0.1965 | 0.9218 |
62
- | 0.1106 | 7.0 | 481 | 0.1829 | 0.9318 |
63
- | 0.1095 | 7.91 | 544 | 0.1937 | 0.9264 |
 
 
 
 
 
 
 
 
64
 
65
 
66
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.1954
21
+ - Accuracy: 0.9395
22
 
23
  ## Model description
24
 
 
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 16
50
  - mixed_precision_training: Native AMP
51
 
52
  ### Training results
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
56
+ | 0.5451 | 0.99 | 68 | 0.2960 | 0.8936 |
57
+ | 0.2488 | 1.99 | 137 | 0.2254 | 0.9105 |
58
+ | 0.1986 | 3.0 | 206 | 0.1913 | 0.9282 |
59
+ | 0.1714 | 4.0 | 275 | 0.1906 | 0.9264 |
60
+ | 0.1576 | 4.99 | 343 | 0.1825 | 0.9323 |
61
+ | 0.1359 | 5.99 | 412 | 0.1973 | 0.9318 |
62
+ | 0.1193 | 7.0 | 481 | 0.1756 | 0.9368 |
63
+ | 0.1062 | 8.0 | 550 | 0.1743 | 0.9382 |
64
+ | 0.0983 | 8.99 | 618 | 0.1885 | 0.9395 |
65
+ | 0.0797 | 9.99 | 687 | 0.1931 | 0.9309 |
66
+ | 0.0698 | 11.0 | 756 | 0.1895 | 0.9359 |
67
+ | 0.0657 | 12.0 | 825 | 0.1861 | 0.9368 |
68
+ | 0.0587 | 12.99 | 893 | 0.1837 | 0.9414 |
69
+ | 0.056 | 13.99 | 962 | 0.1936 | 0.9377 |
70
+ | 0.0592 | 15.0 | 1031 | 0.1958 | 0.935 |
71
+ | 0.0508 | 15.83 | 1088 | 0.1954 | 0.9395 |
72
 
73
 
74
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8fcdda47b3dbd43aacece6cf0f93c19d55581315c4a27e0ef0170a24833e9be7
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:416d6a2619b2a44197037a064c4ae0dbfa10f0637d0c4fc2cd9c9b8ca257ff8b
3
  size 343223968
runs/Mar18_06-14-26_c66e66e08f91/events.out.tfevents.1710742467.c66e66e08f91.34.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5b42befcf0ce031df5b42c06cef7b156fc42fe51cdcdf1448fc269b9e43fe87b
3
- size 13160
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef12816d940c5d5dc9de3e126b5b06a04cb52669168b96eea5237292acb6a1c0
3
+ size 13514