Hugging Face's logo

sck
/
vca

Image Classification
Transformers
PyTorch
vit
Generated from Trainer
Inference Endpoints
Model card Files Community
sck commited on
Commit
ade7109
·
1 Parent(s): 58c380f

Model save

Browse files
Files changed (2) hide show
  1. README.md +60 -29
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -1,5 +1,6 @@
1
  ---
2
  license: apache-2.0
 
3
  tags:
4
  - generated_from_trainer
5
  datasets:
@@ -21,7 +22,7 @@ model-index:
21
  metrics:
22
  - name: Recall
23
  type: recall
24
- value: 0.7866666666666666
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  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 imagefolder dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 0.2021
35
- - Recall: 0.7867
36
 
37
  ## Model description
38
 
@@ -60,38 +61,68 @@ The following hyperparameters were used during training:
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: cosine_with_restarts
62
  - lr_scheduler_warmup_ratio: 0.1
63
- - num_epochs: 30
64
 
65
  ### Training results
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Recall |
68
  |:-------------:|:-----:|:----:|:---------------:|:------:|
69
- | No log | 1.0 | 9 | 0.4676 | 0.0 |
70
- | No log | 2.0 | 18 | 0.2918 | 0.0 |
71
- | No log | 3.0 | 27 | 0.2191 | 0.0 |
72
- | No log | 4.0 | 36 | 0.1971 | 0.1733 |
73
- | No log | 5.0 | 45 | 0.1695 | 0.4133 |
74
- | No log | 6.0 | 54 | 0.1693 | 0.52 |
75
- | No log | 7.0 | 63 | 0.1597 | 0.5867 |
76
- | No log | 8.0 | 72 | 0.1863 | 0.7733 |
77
- | No log | 9.0 | 81 | 0.1591 | 0.72 |
78
- | No log | 10.0 | 90 | 0.1543 | 0.72 |
79
- | No log | 11.0 | 99 | 0.1559 | 0.6933 |
80
- | No log | 12.0 | 108 | 0.1658 | 0.7333 |
81
- | No log | 13.0 | 117 | 0.1691 | 0.6533 |
82
- | No log | 14.0 | 126 | 0.1779 | 0.68 |
83
- | No log | 15.0 | 135 | 0.1635 | 0.8133 |
84
- | No log | 16.0 | 144 | 0.1765 | 0.6933 |
85
- | No log | 17.0 | 153 | 0.1679 | 0.7333 |
86
- | No log | 18.0 | 162 | 0.1694 | 0.7467 |
87
- | No log | 19.0 | 171 | 0.1770 | 0.8133 |
88
- | No log | 20.0 | 180 | 0.1692 | 0.7867 |
89
- | No log | 21.0 | 189 | 0.2021 | 0.7867 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
90
 
91
 
92
  ### Framework versions
93
 
94
- - Transformers 4.31.0.dev0
95
- - Pytorch 1.13.1+cu117
96
- - Datasets 2.11.0
97
- - Tokenizers 0.13.3
 
1
  ---
2
  license: apache-2.0
3
+ base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_trainer
6
  datasets:
 
22
  metrics:
23
  - name: Recall
24
  type: recall
25
+ value: 0.6533333333333333
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  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 imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.2071
36
+ - Recall: 0.6533
37
 
38
  ## Model description
39
 
 
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: cosine_with_restarts
63
  - lr_scheduler_warmup_ratio: 0.1
64
+ - num_epochs: 100
65
 
66
  ### Training results
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Recall |
69
  |:-------------:|:-----:|:----:|:---------------:|:------:|
70
+ | No log | 1.0 | 9 | 0.6362 | 0.3333 |
71
+ | No log | 2.0 | 18 | 0.4641 | 0.0 |
72
+ | No log | 3.0 | 27 | 0.3251 | 0.0 |
73
+ | No log | 4.0 | 36 | 0.2605 | 0.0 |
74
+ | No log | 5.0 | 45 | 0.2100 | 0.0 |
75
+ | No log | 6.0 | 54 | 0.1943 | 0.08 |
76
+ | No log | 7.0 | 63 | 0.1986 | 0.64 |
77
+ | No log | 8.0 | 72 | 0.1856 | 0.6933 |
78
+ | No log | 9.0 | 81 | 0.1654 | 0.6933 |
79
+ | No log | 10.0 | 90 | 0.1593 | 0.72 |
80
+ | No log | 11.0 | 99 | 0.1638 | 0.68 |
81
+ | No log | 12.0 | 108 | 0.1732 | 0.6933 |
82
+ | No log | 13.0 | 117 | 0.1748 | 0.56 |
83
+ | No log | 14.0 | 126 | 0.1792 | 0.6533 |
84
+ | No log | 15.0 | 135 | 0.1743 | 0.84 |
85
+ | No log | 16.0 | 144 | 0.1760 | 0.5733 |
86
+ | No log | 17.0 | 153 | 0.1641 | 0.6 |
87
+ | No log | 18.0 | 162 | 0.1558 | 0.76 |
88
+ | No log | 19.0 | 171 | 0.2121 | 0.7867 |
89
+ | No log | 20.0 | 180 | 0.1765 | 0.56 |
90
+ | No log | 21.0 | 189 | 0.1802 | 0.7733 |
91
+ | No log | 22.0 | 198 | 0.1729 | 0.7467 |
92
+ | No log | 23.0 | 207 | 0.2004 | 0.48 |
93
+ | No log | 24.0 | 216 | 0.1794 | 0.72 |
94
+ | No log | 25.0 | 225 | 0.2185 | 0.7867 |
95
+ | No log | 26.0 | 234 | 0.2115 | 0.8533 |
96
+ | No log | 27.0 | 243 | 0.1999 | 0.7067 |
97
+ | No log | 28.0 | 252 | 0.1900 | 0.5467 |
98
+ | No log | 29.0 | 261 | 0.2158 | 0.72 |
99
+ | No log | 30.0 | 270 | 0.2515 | 0.8533 |
100
+ | No log | 31.0 | 279 | 0.2322 | 0.7733 |
101
+ | No log | 32.0 | 288 | 0.2024 | 0.8 |
102
+ | No log | 33.0 | 297 | 0.2342 | 0.76 |
103
+ | No log | 34.0 | 306 | 0.2205 | 0.7467 |
104
+ | No log | 35.0 | 315 | 0.1820 | 0.7067 |
105
+ | No log | 36.0 | 324 | 0.2169 | 0.68 |
106
+ | No log | 37.0 | 333 | 0.2170 | 0.6133 |
107
+ | No log | 38.0 | 342 | 0.1767 | 0.68 |
108
+ | No log | 39.0 | 351 | 0.2326 | 0.8133 |
109
+ | No log | 40.0 | 360 | 0.2386 | 0.76 |
110
+ | No log | 41.0 | 369 | 0.2431 | 0.68 |
111
+ | No log | 42.0 | 378 | 0.2160 | 0.6933 |
112
+ | No log | 43.0 | 387 | 0.2234 | 0.76 |
113
+ | No log | 44.0 | 396 | 0.2491 | 0.7467 |
114
+ | No log | 45.0 | 405 | 0.2342 | 0.6933 |
115
+ | No log | 46.0 | 414 | 0.2124 | 0.7333 |
116
+ | No log | 47.0 | 423 | 0.2602 | 0.6533 |
117
+ | No log | 48.0 | 432 | 0.2702 | 0.6133 |
118
+ | No log | 49.0 | 441 | 0.2258 | 0.6533 |
119
+ | No log | 50.0 | 450 | 0.2158 | 0.64 |
120
+ | No log | 51.0 | 459 | 0.2071 | 0.6533 |
121
 
122
 
123
  ### Framework versions
124
 
125
+ - Transformers 4.34.1
126
+ - Pytorch 2.0.1+cu117
127
+ - Datasets 2.14.6
128
+ - Tokenizers 0.14.1
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:285fe32d9d54062d785d1914df5e6d01aa89787307e8654407534e2589bfc4fb
3
  size 343268717
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c82945a9820eedd828d225088ec728ac1cf7daf8bdb70e08d3c5923d5057323
3
  size 343268717