Hugging Face's logo

jaypratap
/
vit-pretraining-2024_03_10

Transformers
TensorBoard
Safetensors
vit_mae
pretraining
masked-auto-encoding
generated_from_trainer
Inference Endpoints
Model card Files Metrics Community
jaypratap commited on
Commit
44e53ed
•
1 Parent(s): 756b917

Model save

Browse files
Files changed (3) hide show
  1. README.md +259 -0
  2. model.safetensors +1 -1
  3. runs/Mar12_04-48-19_ip-172-16-71-114.us-west-2.compute.internal/events.out.tfevents.1710218909.ip-172-16-71-114.us-west-2.compute.internal.18339.0 +2 -2
README.md ADDED
@@ -0,0 +1,259 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google/vit-base-patch16-224
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - imagefolder
8
+ model-index:
9
+ - name: vit-pretraining-2024_03_10
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # vit-pretraining-2024_03_10
17
+
18
+ This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.4430
21
+
22
+ ## Model description
23
+
24
+ More information needed
25
+
26
+ ## Intended uses & limitations
27
+
28
+ More information needed
29
+
30
+ ## Training and evaluation data
31
+
32
+ More information needed
33
+
34
+ ## Training procedure
35
+
36
+ ### Training hyperparameters
37
+
38
+ The following hyperparameters were used during training:
39
+ - learning_rate: 4.6875e-06
40
+ - train_batch_size: 8
41
+ - eval_batch_size: 8
42
+ - seed: 42
43
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
+ - lr_scheduler_type: cosine
45
+ - lr_scheduler_warmup_ratio: 0.05
46
+ - num_epochs: 200.0
47
+
48
+ ### Training results
49
+
50
+ | Training Loss | Epoch | Step | Validation Loss |
51
+ |:-------------:|:-----:|:------:|:---------------:|
52
+ | 1.0002 | 1.0 | 2443 | 1.0000 |
53
+ | 0.9832 | 2.0 | 4886 | 0.9753 |
54
+ | 0.9246 | 3.0 | 7329 | 0.9304 |
55
+ | 0.8979 | 4.0 | 9772 | 0.8855 |
56
+ | 0.8307 | 5.0 | 12215 | 0.8077 |
57
+ | 0.7861 | 6.0 | 14658 | 0.7776 |
58
+ | 0.7665 | 7.0 | 17101 | 0.7557 |
59
+ | 0.7421 | 8.0 | 19544 | 0.7337 |
60
+ | 0.6841 | 9.0 | 21987 | 0.7133 |
61
+ | 0.6875 | 10.0 | 24430 | 0.7001 |
62
+ | 0.6991 | 11.0 | 26873 | 0.6887 |
63
+ | 0.6991 | 12.0 | 29316 | 0.6711 |
64
+ | 0.6584 | 13.0 | 31759 | 0.6674 |
65
+ | 0.6619 | 14.0 | 34202 | 0.6507 |
66
+ | 0.6389 | 15.0 | 36645 | 0.6462 |
67
+ | 0.6381 | 16.0 | 39088 | 0.6370 |
68
+ | 0.616 | 17.0 | 41531 | 0.6248 |
69
+ | 0.627 | 18.0 | 43974 | 0.6213 |
70
+ | 0.6179 | 19.0 | 46417 | 0.6150 |
71
+ | 0.6226 | 20.0 | 48860 | 0.6112 |
72
+ | 0.5876 | 21.0 | 51303 | 0.6062 |
73
+ | 0.613 | 22.0 | 53746 | 0.5990 |
74
+ | 0.5864 | 23.0 | 56189 | 0.5948 |
75
+ | 0.5741 | 24.0 | 58632 | 0.5940 |
76
+ | 0.5886 | 25.0 | 61075 | 0.5883 |
77
+ | 0.6028 | 26.0 | 63518 | 0.5890 |
78
+ | 0.578 | 27.0 | 65961 | 0.5841 |
79
+ | 0.5846 | 28.0 | 68404 | 0.5779 |
80
+ | 0.5725 | 29.0 | 70847 | 0.5766 |
81
+ | 0.5684 | 30.0 | 73290 | 0.5791 |
82
+ | 0.5689 | 31.0 | 75733 | 0.5726 |
83
+ | 0.5478 | 32.0 | 78176 | 0.5708 |
84
+ | 0.5739 | 33.0 | 80619 | 0.5697 |
85
+ | 0.5578 | 34.0 | 83062 | 0.5629 |
86
+ | 0.568 | 35.0 | 85505 | 0.5696 |
87
+ | 0.5819 | 36.0 | 87948 | 0.5649 |
88
+ | 0.5442 | 37.0 | 90391 | 0.5649 |
89
+ | 0.5616 | 38.0 | 92834 | 0.5626 |
90
+ | 0.5386 | 39.0 | 95277 | 0.5617 |
91
+ | 0.5725 | 40.0 | 97720 | 0.5552 |
92
+ | 0.549 | 41.0 | 100163 | 0.5621 |
93
+ | 0.5539 | 42.0 | 102606 | 0.5535 |
94
+ | 0.5513 | 43.0 | 105049 | 0.5514 |
95
+ | 0.5538 | 44.0 | 107492 | 0.5480 |
96
+ | 0.5423 | 45.0 | 109935 | 0.5488 |
97
+ | 0.5431 | 46.0 | 112378 | 0.5466 |
98
+ | 0.5495 | 47.0 | 114821 | 0.5442 |
99
+ | 0.5593 | 48.0 | 117264 | 0.5447 |
100
+ | 0.5488 | 49.0 | 119707 | 0.5431 |
101
+ | 0.5203 | 50.0 | 122150 | 0.5391 |
102
+ | 0.5386 | 51.0 | 124593 | 0.5384 |
103
+ | 0.5498 | 52.0 | 127036 | 0.5393 |
104
+ | 0.5391 | 53.0 | 129479 | 0.5372 |
105
+ | 0.5361 | 54.0 | 131922 | 0.5363 |
106
+ | 0.5295 | 55.0 | 134365 | 0.5343 |
107
+ | 0.5227 | 56.0 | 136808 | 0.5345 |
108
+ | 0.5182 | 57.0 | 139251 | 0.5287 |
109
+ | 0.5103 | 58.0 | 141694 | 0.5303 |
110
+ | 0.5411 | 59.0 | 144137 | 0.5278 |
111
+ | 0.5187 | 60.0 | 146580 | 0.5259 |
112
+ | 0.5272 | 61.0 | 149023 | 0.5254 |
113
+ | 0.5352 | 62.0 | 151466 | 0.5264 |
114
+ | 0.5243 | 63.0 | 153909 | 0.5214 |
115
+ | 0.5134 | 64.0 | 156352 | 0.5210 |
116
+ | 0.5305 | 65.0 | 158795 | 0.5238 |
117
+ | 0.5507 | 66.0 | 161238 | 0.5210 |
118
+ | 0.5179 | 67.0 | 163681 | 0.5217 |
119
+ | 0.5162 | 68.0 | 166124 | 0.5166 |
120
+ | 0.5192 | 69.0 | 168567 | 0.5201 |
121
+ | 0.5231 | 70.0 | 171010 | 0.5175 |
122
+ | 0.5095 | 71.0 | 173453 | 0.5138 |
123
+ | 0.5205 | 72.0 | 175896 | 0.5135 |
124
+ | 0.5299 | 73.0 | 178339 | 0.5147 |
125
+ | 0.4947 | 74.0 | 180782 | 0.5112 |
126
+ | 0.5133 | 75.0 | 183225 | 0.5115 |
127
+ | 0.4886 | 76.0 | 185668 | 0.5090 |
128
+ | 0.5288 | 77.0 | 188111 | 0.5105 |
129
+ | 0.514 | 78.0 | 190554 | 0.5072 |
130
+ | 0.4803 | 79.0 | 192997 | 0.5053 |
131
+ | 0.4882 | 80.0 | 195440 | 0.5075 |
132
+ | 0.5037 | 81.0 | 197883 | 0.5063 |
133
+ | 0.5314 | 82.0 | 200326 | 0.5027 |
134
+ | 0.5181 | 83.0 | 202769 | 0.5013 |
135
+ | 0.5191 | 84.0 | 205212 | 0.5009 |
136
+ | 0.503 | 85.0 | 207655 | 0.4980 |
137
+ | 0.4894 | 86.0 | 210098 | 0.4993 |
138
+ | 0.4801 | 87.0 | 212541 | 0.4964 |
139
+ | 0.5019 | 88.0 | 214984 | 0.4956 |
140
+ | 0.5036 | 89.0 | 217427 | 0.4927 |
141
+ | 0.4844 | 90.0 | 219870 | 0.4932 |
142
+ | 0.4656 | 91.0 | 222313 | 0.4890 |
143
+ | 0.4839 | 92.0 | 224756 | 0.4881 |
144
+ | 0.4955 | 93.0 | 227199 | 0.4880 |
145
+ | 0.4792 | 94.0 | 229642 | 0.4877 |
146
+ | 0.4655 | 95.0 | 232085 | 0.4833 |
147
+ | 0.4811 | 96.0 | 234528 | 0.4835 |
148
+ | 0.5118 | 97.0 | 236971 | 0.4842 |
149
+ | 0.479 | 98.0 | 239414 | 0.4830 |
150
+ | 0.4693 | 99.0 | 241857 | 0.4827 |
151
+ | 0.46 | 100.0 | 244300 | 0.4785 |
152
+ | 0.479 | 101.0 | 246743 | 0.4792 |
153
+ | 0.4702 | 102.0 | 249186 | 0.4793 |
154
+ | 0.4683 | 103.0 | 251629 | 0.4757 |
155
+ | 0.4682 | 104.0 | 254072 | 0.4750 |
156
+ | 0.4749 | 105.0 | 256515 | 0.4747 |
157
+ | 0.4915 | 106.0 | 258958 | 0.4719 |
158
+ | 0.4832 | 107.0 | 261401 | 0.4729 |
159
+ | 0.4371 | 108.0 | 263844 | 0.4720 |
160
+ | 0.4779 | 109.0 | 266287 | 0.4710 |
161
+ | 0.4796 | 110.0 | 268730 | 0.4693 |
162
+ | 0.463 | 111.0 | 271173 | 0.4696 |
163
+ | 0.4722 | 112.0 | 273616 | 0.4679 |
164
+ | 0.4689 | 113.0 | 276059 | 0.4693 |
165
+ | 0.4644 | 114.0 | 278502 | 0.4665 |
166
+ | 0.4688 | 115.0 | 280945 | 0.4674 |
167
+ | 0.4619 | 116.0 | 283388 | 0.4644 |
168
+ | 0.4533 | 117.0 | 285831 | 0.4663 |
169
+ | 0.4604 | 118.0 | 288274 | 0.4634 |
170
+ | 0.4722 | 119.0 | 290717 | 0.4637 |
171
+ | 0.4622 | 120.0 | 293160 | 0.4634 |
172
+ | 0.4575 | 121.0 | 295603 | 0.4628 |
173
+ | 0.4824 | 122.0 | 298046 | 0.4631 |
174
+ | 0.4757 | 123.0 | 300489 | 0.4620 |
175
+ | 0.4457 | 124.0 | 302932 | 0.4620 |
176
+ | 0.4471 | 125.0 | 305375 | 0.4599 |
177
+ | 0.444 | 126.0 | 307818 | 0.4575 |
178
+ | 0.4521 | 127.0 | 310261 | 0.4599 |
179
+ | 0.4441 | 128.0 | 312704 | 0.4588 |
180
+ | 0.4432 | 129.0 | 315147 | 0.4596 |
181
+ | 0.4518 | 130.0 | 317590 | 0.4550 |
182
+ | 0.4457 | 131.0 | 320033 | 0.4578 |
183
+ | 0.4529 | 132.0 | 322476 | 0.4543 |
184
+ | 0.4871 | 133.0 | 324919 | 0.4560 |
185
+ | 0.4482 | 134.0 | 327362 | 0.4546 |
186
+ | 0.4648 | 135.0 | 329805 | 0.4574 |
187
+ | 0.4372 | 136.0 | 332248 | 0.4546 |
188
+ | 0.4353 | 137.0 | 334691 | 0.4531 |
189
+ | 0.4446 | 138.0 | 337134 | 0.4539 |
190
+ | 0.4666 | 139.0 | 339577 | 0.4518 |
191
+ | 0.4734 | 140.0 | 342020 | 0.4528 |
192
+ | 0.4601 | 141.0 | 344463 | 0.4540 |
193
+ | 0.4415 | 142.0 | 346906 | 0.4528 |
194
+ | 0.459 | 143.0 | 349349 | 0.4505 |
195
+ | 0.454 | 144.0 | 351792 | 0.4514 |
196
+ | 0.4606 | 145.0 | 354235 | 0.4511 |
197
+ | 0.4315 | 146.0 | 356678 | 0.4514 |
198
+ | 0.4583 | 147.0 | 359121 | 0.4520 |
199
+ | 0.452 | 148.0 | 361564 | 0.4495 |
200
+ | 0.4449 | 149.0 | 364007 | 0.4508 |
201
+ | 0.4272 | 150.0 | 366450 | 0.4489 |
202
+ | 0.439 | 151.0 | 368893 | 0.4504 |
203
+ | 0.4586 | 152.0 | 371336 | 0.4503 |
204
+ | 0.4559 | 153.0 | 373779 | 0.4500 |
205
+ | 0.4527 | 154.0 | 376222 | 0.4492 |
206
+ | 0.4511 | 155.0 | 378665 | 0.4491 |
207
+ | 0.4405 | 156.0 | 381108 | 0.4488 |
208
+ | 0.4509 | 157.0 | 383551 | 0.4482 |
209
+ | 0.4713 | 158.0 | 385994 | 0.4480 |
210
+ | 0.4578 | 159.0 | 388437 | 0.4465 |
211
+ | 0.4154 | 160.0 | 390880 | 0.4464 |
212
+ | 0.4399 | 161.0 | 393323 | 0.4488 |
213
+ | 0.4547 | 162.0 | 395766 | 0.4476 |
214
+ | 0.4426 | 163.0 | 398209 | 0.4456 |
215
+ | 0.4517 | 164.0 | 400652 | 0.4484 |
216
+ | 0.4376 | 165.0 | 403095 | 0.4455 |
217
+ | 0.4463 | 166.0 | 405538 | 0.4463 |
218
+ | 0.4289 | 167.0 | 407981 | 0.4466 |
219
+ | 0.4291 | 168.0 | 410424 | 0.4469 |
220
+ | 0.4623 | 169.0 | 412867 | 0.4455 |
221
+ | 0.4673 | 170.0 | 415310 | 0.4455 |
222
+ | 0.4609 | 171.0 | 417753 | 0.4456 |
223
+ | 0.4478 | 172.0 | 420196 | 0.4468 |
224
+ | 0.4521 | 173.0 | 422639 | 0.4437 |
225
+ | 0.4378 | 174.0 | 425082 | 0.4460 |
226
+ | 0.4361 | 175.0 | 427525 | 0.4446 |
227
+ | 0.4321 | 176.0 | 429968 | 0.4451 |
228
+ | 0.4369 | 177.0 | 432411 | 0.4451 |
229
+ | 0.4381 | 178.0 | 434854 | 0.4443 |
230
+ | 0.4408 | 179.0 | 437297 | 0.4449 |
231
+ | 0.4414 | 180.0 | 439740 | 0.4448 |
232
+ | 0.4333 | 181.0 | 442183 | 0.4438 |
233
+ | 0.4468 | 182.0 | 444626 | 0.4452 |
234
+ | 0.4394 | 183.0 | 447069 | 0.4440 |
235
+ | 0.441 | 184.0 | 449512 | 0.4434 |
236
+ | 0.4546 | 185.0 | 451955 | 0.4462 |
237
+ | 0.4455 | 186.0 | 454398 | 0.4458 |
238
+ | 0.4431 | 187.0 | 456841 | 0.4426 |
239
+ | 0.4489 | 188.0 | 459284 | 0.4433 |
240
+ | 0.4485 | 189.0 | 461727 | 0.4435 |
241
+ | 0.4449 | 190.0 | 464170 | 0.4433 |
242
+ | 0.4482 | 191.0 | 466613 | 0.4449 |
243
+ | 0.4395 | 192.0 | 469056 | 0.4433 |
244
+ | 0.4557 | 193.0 | 471499 | 0.4436 |
245
+ | 0.4208 | 194.0 | 473942 | 0.4450 |
246
+ | 0.4274 | 195.0 | 476385 | 0.4429 |
247
+ | 0.4423 | 196.0 | 478828 | 0.4434 |
248
+ | 0.4331 | 197.0 | 481271 | 0.4453 |
249
+ | 0.43 | 198.0 | 483714 | 0.4448 |
250
+ | 0.4308 | 199.0 | 486157 | 0.4460 |
251
+ | 0.4373 | 200.0 | 488600 | 0.4430 |
252
+
253
+
254
+ ### Framework versions
255
+
256
+ - Transformers 4.39.0.dev0
257
+ - Pytorch 2.2.1+cu121
258
+ - Datasets 2.18.0
259
+ - Tokenizers 0.15.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:08697cb7b1638a7e82820ca80bf33805f90db6c98cd465f3b73c14f14843e69d
3
  size 447670680
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27c68a82f5debe48dc38f11814ecd5234a5709f59135bdbe1678a686732aad46
3
  size 447670680
runs/Mar12_04-48-19_ip-172-16-71-114.us-west-2.compute.internal/events.out.tfevents.1710218909.ip-172-16-71-114.us-west-2.compute.internal.18339.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7dc703d94be1614b64722a4fcb27443ef31f5254bf7cafaa63242fa00e2d7389
3
- size 10627437
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:419f6b3b0e368a095213d4b60cb2c4ea55f8b2d2e08a8b57329dfaa35eab2800
3
+ size 10627797