Hugging Face's logo

tonyassi
/
celebrity-classifier

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Model card Files Metrics Community
tonyassi commited on
Commit
a321fcc
·
1 Parent(s): c04eb81

End of training

Browse files
Files changed (2) hide show
  1. README.md +22 -22
  2. model.safetensors +1 -1
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: 1.6266
21
- - Accuracy: 0.7355
22
 
23
  ## Model description
24
 
@@ -52,26 +52,26 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 3.2092 | 1.0 | 227 | 3.8429 | 0.4847 |
56
- | 3.1076 | 2.0 | 455 | 3.6825 | 0.4990 |
57
- | 2.7996 | 3.0 | 682 | 3.4241 | 0.5194 |
58
- | 2.4653 | 4.0 | 910 | 3.1673 | 0.5513 |
59
- | 2.117 | 5.0 | 1137 | 2.9653 | 0.5733 |
60
- | 1.977 | 6.0 | 1365 | 2.7396 | 0.6057 |
61
- | 1.7053 | 7.0 | 1592 | 2.6115 | 0.6107 |
62
- | 1.5829 | 8.0 | 1820 | 2.4370 | 0.6395 |
63
- | 1.4496 | 9.0 | 2047 | 2.2758 | 0.6635 |
64
- | 1.3663 | 10.0 | 2275 | 2.1700 | 0.6816 |
65
- | 1.2573 | 11.0 | 2502 | 2.0498 | 0.6910 |
66
- | 1.1949 | 12.0 | 2730 | 1.9767 | 0.6956 |
67
- | 1.1401 | 13.0 | 2957 | 1.8944 | 0.7050 |
68
- | 1.0346 | 14.0 | 3185 | 1.8601 | 0.7072 |
69
- | 0.9015 | 15.0 | 3412 | 1.7747 | 0.7215 |
70
- | 0.8774 | 16.0 | 3640 | 1.7055 | 0.7239 |
71
- | 0.875 | 17.0 | 3867 | 1.7231 | 0.7294 |
72
- | 0.9291 | 18.0 | 4095 | 1.6697 | 0.7303 |
73
- | 0.8888 | 19.0 | 4322 | 1.6108 | 0.7404 |
74
- | 0.8496 | 19.96 | 4540 | 1.6266 | 0.7355 |
75
 
76
 
77
  ### 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: 1.0002
21
+ - Accuracy: 0.7850
22
 
23
  ## Model description
24
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 0.7184 | 1.0 | 227 | 1.6196 | 0.7327 |
56
+ | 0.6575 | 2.0 | 455 | 1.6466 | 0.7102 |
57
+ | 0.6033 | 3.0 | 682 | 1.5743 | 0.7226 |
58
+ | 0.4692 | 4.0 | 910 | 1.4970 | 0.7289 |
59
+ | 0.3551 | 5.0 | 1137 | 1.4109 | 0.7393 |
60
+ | 0.4104 | 6.0 | 1365 | 1.3546 | 0.7352 |
61
+ | 0.3038 | 7.0 | 1592 | 1.3356 | 0.7393 |
62
+ | 0.3195 | 8.0 | 1820 | 1.3145 | 0.7371 |
63
+ | 0.3254 | 9.0 | 2047 | 1.2262 | 0.7580 |
64
+ | 0.3182 | 10.0 | 2275 | 1.1926 | 0.7630 |
65
+ | 0.3278 | 11.0 | 2502 | 1.1491 | 0.7704 |
66
+ | 0.2805 | 12.0 | 2730 | 1.1514 | 0.7627 |
67
+ | 0.3188 | 13.0 | 2957 | 1.0948 | 0.7795 |
68
+ | 0.3161 | 14.0 | 3185 | 1.1231 | 0.7685 |
69
+ | 0.229 | 15.0 | 3412 | 1.0588 | 0.7789 |
70
+ | 0.2544 | 16.0 | 3640 | 1.0311 | 0.7806 |
71
+ | 0.2872 | 17.0 | 3867 | 1.0586 | 0.7762 |
72
+ | 0.3974 | 18.0 | 4095 | 1.0341 | 0.7833 |
73
+ | 0.337 | 19.0 | 4322 | 0.9834 | 0.7891 |
74
+ | 0.3165 | 19.96 | 4540 | 1.0002 | 0.7850 |
75
 
76
 
77
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce849c881318786dc6edc8a1ab4a20f28d9afac68b2ff3b743ce3e731e78cd9e
3
  size 346284620
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c8c5820a8e087253b7ad246cbf73e3c03d43ad503d5f99958bfc2d04d0b3795
3
  size 346284620