Hugging Face's logo

pkr7098
/
vit-base-beans

Image Classification
PEFT
Safetensors
vit
vision
Generated from Trainer
Model card Files Community
pkr7098 commited on
Commit
117a687
·
verified ·
1 Parent(s): e93f5a5

Model save

Browse files
Files changed (2) hide show
  1. README.md +9 -11
  2. adapter_model.safetensors +1 -1
README.md CHANGED
@@ -5,8 +5,6 @@ license: apache-2.0
5
  metrics:
6
  - accuracy
7
  tags:
8
- - image-classification
9
- - vision
10
  - generated_from_trainer
11
  model-index:
12
  - name: vit-base-beans
@@ -18,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  # vit-base-beans
20
 
21
- 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.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.8347
24
- - Accuracy: 0.8045
25
 
26
  ## Model description
27
 
@@ -40,7 +38,7 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - learning_rate: 2e-05
44
  - train_batch_size: 8
45
  - eval_batch_size: 8
46
  - seed: 1337
@@ -52,11 +50,11 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 1.0299 | 1.0 | 130 | 1.0292 | 0.6241 |
56
- | 0.9616 | 2.0 | 260 | 0.9589 | 0.7444 |
57
- | 0.8783 | 3.0 | 390 | 0.8892 | 0.7895 |
58
- | 0.836 | 4.0 | 520 | 0.8480 | 0.8045 |
59
- | 0.8494 | 5.0 | 650 | 0.8347 | 0.8045 |
60
 
61
 
62
  ### Framework versions
 
5
  metrics:
6
  - accuracy
7
  tags:
 
 
8
  - generated_from_trainer
9
  model-index:
10
  - name: vit-base-beans
 
16
 
17
  # vit-base-beans
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.0804
22
+ - Accuracy: 0.9699
23
 
24
  ## Model description
25
 
 
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
+ - learning_rate: 0.0005
42
  - train_batch_size: 8
43
  - eval_batch_size: 8
44
  - seed: 1337
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 0.2167 | 1.0 | 130 | 0.1714 | 0.9474 |
54
+ | 0.1288 | 2.0 | 260 | 0.1307 | 0.9549 |
55
+ | 0.1056 | 3.0 | 390 | 0.0669 | 0.9774 |
56
+ | 0.0772 | 4.0 | 520 | 0.0466 | 0.9774 |
57
+ | 0.0941 | 5.0 | 650 | 0.0804 | 0.9699 |
58
 
59
 
60
  ### Framework versions
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:650fafaaf52836fd8f0842525397a217e3780e22c46d55e95b942c84ef597716
3
  size 3558756
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:501eea3127ecb6ac45e830080cd17200bf34998dd0b225726555307e24a8bcc1
3
  size 3558756