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
a61e263
·
verified ·
1 Parent(s): 65ce60e

Model save

Browse files
Files changed (3) hide show
  1. README.md +10 -12
  2. adapter_config.json +2 -2
  3. training_args.bin +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.5642
24
  - Accuracy: 0.8797
 
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: 4e-05
44
  - train_batch_size: 8
45
  - eval_batch_size: 8
46
  - seed: 1337
@@ -50,13 +48,13 @@ The following hyperparameters were used during training:
50
 
51
  ### Training results
52
 
53
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 0.9653 | 1.0 | 130 | 0.9567 | 0.7519 |
56
- | 0.7802 | 2.0 | 260 | 0.7600 | 0.8496 |
57
- | 0.6269 | 3.0 | 390 | 0.6368 | 0.8797 |
58
- | 0.5764 | 4.0 | 520 | 0.5801 | 0.8872 |
59
- | 0.5977 | 5.0 | 650 | 0.5642 | 0.8797 |
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
  - Accuracy: 0.8797
22
+ - Loss: 0.5642
23
 
24
  ## Model description
25
 
 
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
+ - learning_rate: 2e-05
42
  - train_batch_size: 8
43
  - eval_batch_size: 8
44
  - seed: 1337
 
48
 
49
  ### Training results
50
 
51
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
52
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
53
+ | 0.9653 | 1.0 | 130 | 0.7519 | 0.9567 |
54
+ | 0.7802 | 2.0 | 260 | 0.8496 | 0.7600 |
55
+ | 0.6269 | 3.0 | 390 | 0.8797 | 0.6368 |
56
+ | 0.5764 | 4.0 | 520 | 0.8872 | 0.5801 |
57
+ | 0.5977 | 5.0 | 650 | 0.8797 | 0.5642 |
58
 
59
 
60
  ### Framework versions
adapter_config.json CHANGED
@@ -25,8 +25,8 @@
25
  "rank_pattern": {},
26
  "revision": null,
27
  "target_modules": [
28
- "query",
29
- "value"
30
  ],
31
  "task_type": null,
32
  "use_dora": false,
 
25
  "rank_pattern": {},
26
  "revision": null,
27
  "target_modules": [
28
+ "value",
29
+ "query"
30
  ],
31
  "task_type": null,
32
  "use_dora": false,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:81360db78c728a20e61d6184752632054de74ccf9c73c9aa9393bd7d32381182
3
  size 4859
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b154d42a21248b9d6f5e659a45ce5379a1258e1cf3137e4f97e49aeb5b1ef398
3
  size 4859