shubhangikarade commited on
Commit
07b37c1
·
1 Parent(s): d26bd09

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -10
README.md CHANGED
@@ -1,8 +1,6 @@
1
  ---
2
  license: apache-2.0
3
  tags:
4
- - image-classification
5
- - vision
6
  - generated_from_trainer
7
  datasets:
8
  - beans
@@ -33,8 +31,8 @@ should probably proofread and complete it, then remove this comment. -->
33
 
34
  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 beans dataset.
35
  It achieves the following results on the evaluation set:
36
- - Loss: 0.1332
37
  - Accuracy: 0.9699
 
38
 
39
  ## Model description
40
 
@@ -63,13 +61,13 @@ The following hyperparameters were used during training:
63
 
64
  ### Training results
65
 
66
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | 0.4885 | 1.0 | 65 | 0.3982 | 0.9549 |
69
- | 0.271 | 2.0 | 130 | 0.1864 | 0.9774 |
70
- | 0.1419 | 3.0 | 195 | 0.1587 | 0.9774 |
71
- | 0.1658 | 4.0 | 260 | 0.1332 | 0.9699 |
72
- | 0.1767 | 5.0 | 325 | 0.1349 | 0.9699 |
73
 
74
 
75
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
  tags:
 
 
4
  - generated_from_trainer
5
  datasets:
6
  - beans
 
31
 
32
  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 beans dataset.
33
  It achieves the following results on the evaluation set:
 
34
  - Accuracy: 0.9699
35
+ - Loss: 0.1349
36
 
37
  ## Model description
38
 
 
61
 
62
  ### Training results
63
 
64
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
65
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
66
+ | 0.4885 | 1.0 | 65 | 0.9549 | 0.3982 |
67
+ | 0.271 | 2.0 | 130 | 0.9774 | 0.1864 |
68
+ | 0.1419 | 3.0 | 195 | 0.9774 | 0.1587 |
69
+ | 0.1658 | 4.0 | 260 | 0.9699 | 0.1332 |
70
+ | 0.1767 | 5.0 | 325 | 0.9699 | 0.1349 |
71
 
72
 
73
  ### Framework versions