Woleek commited on
Commit
916f86d
1 Parent(s): d39a0a1

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.6744
36
  - Accuracy: 0.8182
37
 
38
  ## Model description
@@ -64,11 +64,11 @@ The following hyperparameters were used during training:
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
- | 0.0099 | 3.57 | 50 | 0.6155 | 0.8182 |
68
- | 0.0007 | 7.14 | 100 | 0.7441 | 0.7576 |
69
- | 0.0004 | 10.71 | 150 | 0.6925 | 0.8182 |
70
- | 0.0003 | 14.29 | 200 | 0.6793 | 0.8182 |
71
- | 0.0003 | 17.86 | 250 | 0.6744 | 0.8182 |
72
 
73
 
74
  ### Framework versions
 
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.7675
36
  - Accuracy: 0.8182
37
 
38
  ## Model description
 
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
+ | 0.004 | 3.57 | 50 | 1.0744 | 0.7576 |
68
+ | 0.0005 | 7.14 | 100 | 0.8182 | 0.7879 |
69
+ | 0.0004 | 10.71 | 150 | 0.7808 | 0.7879 |
70
+ | 0.0003 | 14.29 | 200 | 0.7710 | 0.8182 |
71
+ | 0.0003 | 17.86 | 250 | 0.7675 | 0.8182 |
72
 
73
 
74
  ### Framework versions