mariamoracrossitcr commited on
Commit
3cb0005
1 Parent(s): acba45f

Model save

Browse files
Files changed (2) hide show
  1. README.md +5 -6
  2. model.safetensors +1 -1
README.md CHANGED
@@ -3,7 +3,6 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
6
- - image-classification
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
@@ -17,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # vit-base-beans-demo-v18Set
19
 
20
- 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.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.0183
23
- - Accuracy: 0.9925
24
 
25
  ## Model description
26
 
@@ -52,8 +51,8 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:------:|:----:|:---------------:|:--------:|
55
- | 0.078 | 1.5385 | 100 | 0.2393 | 0.9023 |
56
- | 0.0146 | 3.0769 | 200 | 0.0183 | 0.9925 |
57
 
58
 
59
  ### Framework versions
 
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
 
16
 
17
  # vit-base-beans-demo-v18Set
18
 
19
+ This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0673
22
+ - Accuracy: 0.9850
23
 
24
  ## Model description
25
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:------:|:----:|:---------------:|:--------:|
54
+ | 0.0625 | 1.5385 | 100 | 0.1025 | 0.9774 |
55
+ | 0.0119 | 3.0769 | 200 | 0.0673 | 0.9850 |
56
 
57
 
58
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc1c34f306686c6a2550bb2f04dc8b442ae2ee328f66a178104c8299fbd229ba
3
  size 343227052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27bd7f6208fcb9c2202470e06720bbfb6b4b5d9c19281f8c7d7187f1c2df887c
3
  size 343227052