steven commited on
Commit
9d50315
1 Parent(s): d6da07f

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -13
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
@@ -23,7 +21,7 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.9924812030075187
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -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.0818
37
- - Accuracy: 0.9925
38
 
39
  ## Model description
40
 
@@ -63,18 +61,18 @@ The following hyperparameters were used during training:
63
 
64
  ### Training results
65
 
66
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | 0.254 | 1.0 | 130 | 0.2244 | 0.9624 |
69
- | 0.2885 | 2.0 | 260 | 0.1098 | 0.9925 |
70
- | 0.1555 | 3.0 | 390 | 0.1041 | 0.9850 |
71
- | 0.092 | 4.0 | 520 | 0.0818 | 0.9925 |
72
- | 0.1591 | 5.0 | 650 | 0.0826 | 0.9850 |
73
 
74
 
75
  ### Framework versions
76
 
77
  - Transformers 4.29.0.dev0
78
- - Pytorch 1.13.1+cu117
79
  - Datasets 2.11.0
80
  - Tokenizers 0.13.3
 
1
  ---
2
  license: apache-2.0
3
  tags:
 
 
4
  - generated_from_trainer
5
  datasets:
6
  - beans
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.9849624060150376
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
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.9850
35
+ - Loss: 0.0826
36
 
37
  ## Model description
38
 
 
61
 
62
  ### Training results
63
 
64
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
65
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
66
+ | 0.254 | 1.0 | 130 | 0.9624 | 0.2244 |
67
+ | 0.2885 | 2.0 | 260 | 0.9925 | 0.1098 |
68
+ | 0.1555 | 3.0 | 390 | 0.9850 | 0.1041 |
69
+ | 0.092 | 4.0 | 520 | 0.9925 | 0.0818 |
70
+ | 0.1591 | 5.0 | 650 | 0.9850 | 0.0826 |
71
 
72
 
73
  ### Framework versions
74
 
75
  - Transformers 4.29.0.dev0
76
+ - Pytorch 2.0.0+cu117
77
  - Datasets 2.11.0
78
  - Tokenizers 0.13.3