simlaharma commited on
Commit
8cf9d44
1 Parent(s): 69c5227

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.1328
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.49 | 1.0 | 65 | 0.4050 | 0.9624 |
69
- | 0.2769 | 2.0 | 130 | 0.1862 | 0.9850 |
70
- | 0.1441 | 3.0 | 195 | 0.1554 | 0.9774 |
71
- | 0.1661 | 4.0 | 260 | 0.1333 | 0.9774 |
72
- | 0.1754 | 5.0 | 325 | 0.1328 | 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.1328
36
 
37
  ## Model description
38
 
 
61
 
62
  ### Training results
63
 
64
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
65
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
66
+ | 0.49 | 1.0 | 65 | 0.9624 | 0.4050 |
67
+ | 0.2769 | 2.0 | 130 | 0.9850 | 0.1862 |
68
+ | 0.1441 | 3.0 | 195 | 0.9774 | 0.1554 |
69
+ | 0.1661 | 4.0 | 260 | 0.9774 | 0.1333 |
70
+ | 0.1754 | 5.0 | 325 | 0.9699 | 0.1328 |
71
 
72
 
73
  ### Framework versions