Hugging Face's logo

tonyassi
/
vogue-fashion-collection-15

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
tonyassi commited on
Commit
d5440c0
•
1 Parent(s): 69be3bc

End of training

Browse files
Files changed (2) hide show
  1. README.md +15 -10
  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: 2.4719
21
- - Accuracy: 0.8380
22
 
23
  ## Model description
24
 
@@ -46,17 +46,22 @@ 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: 5
50
 
51
  ### Training results
52
 
53
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 6.1358 | 1.0 | 1094 | 6.1121 | 0.5762 |
56
- | 4.5926 | 2.0 | 2189 | 4.5302 | 0.7306 |
57
- | 3.386 | 3.0 | 3283 | 3.3786 | 0.7971 |
58
- | 2.7142 | 4.0 | 4378 | 2.7066 | 0.8231 |
59
- | 2.4032 | 5.0 | 5470 | 2.4719 | 0.8380 |
 
 
 
 
 
60
 
61
 
62
  ### 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.1795
21
+ - Accuracy: 0.9454
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: 10
50
 
51
  ### Training results
52
 
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|
55
+ | 1.6546 | 1.0 | 1094 | 1.8069 | 0.8558 |
56
+ | 0.8991 | 2.0 | 2189 | 0.9520 | 0.8966 |
57
+ | 0.589 | 3.0 | 3283 | 0.5906 | 0.9158 |
58
+ | 0.4017 | 4.0 | 4378 | 0.4167 | 0.9273 |
59
+ | 0.2594 | 5.0 | 5472 | 0.3189 | 0.9358 |
60
+ | 0.308 | 6.0 | 6567 | 0.2758 | 0.9363 |
61
+ | 0.2114 | 7.0 | 7661 | 0.2338 | 0.9419 |
62
+ | 0.1439 | 8.0 | 8756 | 0.2131 | 0.9443 |
63
+ | 0.1568 | 9.0 | 9850 | 0.1943 | 0.9454 |
64
+ | 0.1388 | 10.0 | 10940 | 0.1795 | 0.9454 |
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3a1b514b27537eceff8bff8f529a44233faef87e2a444f7d5544bc7a3d02c076
3
  size 348376316
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd84cde3f121b0d1d7386bfbd47a00904706f733a80014dd51e7d4dc82a39cb8
3
  size 348376316