Hugging Face's logo

Countigo
/
vit-base-beans

Image Classification
Transformers
TensorBoard
Safetensors
vit
vision
Generated from Trainer
Model card Files
xet
Metrics Community
Countigo commited on
Commit
d457b97
·
verified ·
1 Parent(s): 38396f4

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.3436
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.9867 | 1.0 | 17 | 0.7573 | 0.9248 |
55
- | 0.6898 | 2.0 | 34 | 0.5536 | 0.9624 |
56
- | 0.4979 | 3.0 | 51 | 0.4195 | 0.9699 |
57
- | 0.4541 | 4.0 | 68 | 0.3605 | 0.9624 |
58
- | 0.3922 | 5.0 | 85 | 0.3436 | 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.3436
22
 
23
  ## Model description
24
 
 
47
 
48
  ### Training results
49
 
50
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
51
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
52
+ | 0.9867 | 1.0 | 17 | 0.9248 | 0.7573 |
53
+ | 0.6898 | 2.0 | 34 | 0.9624 | 0.5536 |
54
+ | 0.4979 | 3.0 | 51 | 0.9699 | 0.4195 |
55
+ | 0.4541 | 4.0 | 68 | 0.9624 | 0.3605 |
56
+ | 0.3922 | 5.0 | 85 | 0.9699 | 0.3436 |
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:918d210cc6ebc413a0d0804d0200565f030d0be9decc98451ebf09ceccb47030
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2013bda08a76559c057cb1122464533e61219ef151dcf4e8a40ba7b498d09cc9
3
  size 5176