Hugging Face's logo

YufeiWeng
/
vit-base-beans

Image Classification
Transformers
Safetensors
vit
vision
Generated from Trainer
Model card Files Community
YufeiWeng commited on
Commit
f7fec75
·
verified ·
1 Parent(s): e654074

Model save

Browse files
Files changed (3) hide show
  1. README.md +9 -11
  2. config.json +0 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -2,8 +2,6 @@
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
- - image-classification
6
- - vision
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
@@ -17,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # vit-base-beans
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.1921
23
  - Accuracy: 0.9699
 
24
 
25
  ## Model description
26
 
@@ -49,13 +47,13 @@ The following hyperparameters were used during training:
49
 
50
  ### Training results
51
 
52
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | 0.7212 | 1.0 | 33 | 0.5818 | 0.9624 |
55
- | 0.4355 | 2.0 | 66 | 0.3262 | 0.9774 |
56
- | 0.3134 | 3.0 | 99 | 0.2371 | 0.9699 |
57
- | 0.2207 | 4.0 | 132 | 0.1955 | 0.9774 |
58
- | 0.2149 | 5.0 | 165 | 0.1921 | 0.9699 |
59
 
60
 
61
  ### Framework versions
 
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
 
 
5
  - generated_from_trainer
6
  metrics:
7
  - accuracy
 
15
 
16
  # vit-base-beans
17
 
18
+ 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.
19
  It achieves the following results on the evaluation set:
 
20
  - Accuracy: 0.9699
21
+ - Loss: 0.1921
22
 
23
  ## Model description
24
 
 
47
 
48
  ### Training results
49
 
50
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
51
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
52
+ | 0.7212 | 1.0 | 33 | 0.9624 | 0.5818 |
53
+ | 0.4355 | 2.0 | 66 | 0.9774 | 0.3262 |
54
+ | 0.3134 | 3.0 | 99 | 0.9699 | 0.2371 |
55
+ | 0.2207 | 4.0 | 132 | 0.9774 | 0.1955 |
56
+ | 0.2149 | 5.0 | 165 | 0.9699 | 0.1921 |
57
 
58
 
59
  ### Framework versions
config.json CHANGED
@@ -28,7 +28,6 @@
28
  "num_channels": 3,
29
  "num_hidden_layers": 12,
30
  "patch_size": 16,
31
- "problem_type": "single_label_classification",
32
  "qkv_bias": true,
33
  "torch_dtype": "float32",
34
  "transformers_version": "4.43.0.dev0"
 
28
  "num_channels": 3,
29
  "num_hidden_layers": 12,
30
  "patch_size": 16,
 
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
  "transformers_version": "4.43.0.dev0"
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b398c74fa116e966f844b41dc73b1a82558f49afb06ade12699f70ff47986689
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91348e70859975dda7deb5af26f29f0250a0055a64bfae34d5d5a54827fd1300
3
  size 5240