tiya1012 commited on
Commit
310a939
·
1 Parent(s): b1fd5f9

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -5
README.md CHANGED
@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.1375
20
- - Accuracy: 0.9286
21
 
22
  ## Model description
23
 
@@ -42,14 +42,17 @@ The following hyperparameters were used during training:
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
- - num_epochs: 4
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | 0.3926 | 2.0 | 100 | 0.2469 | 0.8980 |
52
- | 0.0904 | 4.0 | 200 | 0.1375 | 0.9286 |
 
 
 
53
 
54
 
55
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.2051
20
+ - Accuracy: 0.9388
21
 
22
  ## Model description
23
 
 
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
+ - num_epochs: 10
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 0.5025 | 2.0 | 100 | 0.3267 | 0.8776 |
52
+ | 0.204 | 4.0 | 200 | 0.2941 | 0.9082 |
53
+ | 0.1425 | 6.0 | 300 | 0.3407 | 0.9082 |
54
+ | 0.091 | 8.0 | 400 | 0.1722 | 0.9592 |
55
+ | 0.0328 | 10.0 | 500 | 0.2051 | 0.9388 |
56
 
57
 
58
  ### Framework versions