vuongnhathien commited on
Commit
a41b734
1 Parent(s): ed0d11e

End of training

Browse files
README.md CHANGED
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.9399602385685885
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [facebook/convnextv2-tiny-22k-384](https://huggingface.co/facebook/convnextv2-tiny-22k-384) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.2370
36
- - Accuracy: 0.9400
37
 
38
  ## Model description
39
 
 
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.9404761904761905
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 [facebook/convnextv2-tiny-22k-384](https://huggingface.co/facebook/convnextv2-tiny-22k-384) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.2265
36
+ - Accuracy: 0.9405
37
 
38
  ## Model description
39
 
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 1.301428412334932e+19,
4
+ "train_loss": 0.4333542574102228,
5
+ "train_runtime": 11179.012,
6
+ "train_samples_per_second": 15.727,
7
+ "train_steps_per_second": 0.492
8
+ }
runs/May29_18-17-22_be109efa84d7/events.out.tfevents.1717018065.be109efa84d7.24.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03aa95a9a5094989eb439c1e33f9cf41da91b104a38f1c64127fee18440af320
3
+ size 411
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 1.301428412334932e+19,
4
+ "train_loss": 0.4333542574102228,
5
+ "train_runtime": 11179.012,
6
+ "train_samples_per_second": 15.727,
7
+ "train_steps_per_second": 0.492
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,505 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.23698414862155914,
3
+ "best_model_checkpoint": "./convnext-tiny-new-1e-4-batch-32/checkpoint-5500",
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
6
+ "global_step": 5500,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.18,
13
+ "grad_norm": 18.67238426208496,
14
+ "learning_rate": 9.991845519630678e-05,
15
+ "loss": 2.1347,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.36,
20
+ "grad_norm": 24.724580764770508,
21
+ "learning_rate": 9.967408676742751e-05,
22
+ "loss": 1.1347,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.55,
27
+ "grad_norm": 23.250165939331055,
28
+ "learning_rate": 9.926769179238466e-05,
29
+ "loss": 0.9363,
30
+ "step": 300
31
+ },
32
+ {
33
+ "epoch": 0.73,
34
+ "grad_norm": 20.889720916748047,
35
+ "learning_rate": 9.870059584711668e-05,
36
+ "loss": 0.8421,
37
+ "step": 400
38
+ },
39
+ {
40
+ "epoch": 0.91,
41
+ "grad_norm": 16.967330932617188,
42
+ "learning_rate": 9.797464868072488e-05,
43
+ "loss": 0.7731,
44
+ "step": 500
45
+ },
46
+ {
47
+ "epoch": 1.0,
48
+ "eval_accuracy": 0.88389662027833,
49
+ "eval_loss": 0.42373502254486084,
50
+ "eval_runtime": 76.9715,
51
+ "eval_samples_per_second": 32.674,
52
+ "eval_steps_per_second": 1.026,
53
+ "step": 550
54
+ },
55
+ {
56
+ "epoch": 1.09,
57
+ "grad_norm": 19.625564575195312,
58
+ "learning_rate": 9.709221818197624e-05,
59
+ "loss": 0.7413,
60
+ "step": 600
61
+ },
62
+ {
63
+ "epoch": 1.27,
64
+ "grad_norm": 16.88126564025879,
65
+ "learning_rate": 9.60561826557425e-05,
66
+ "loss": 0.6635,
67
+ "step": 700
68
+ },
69
+ {
70
+ "epoch": 1.45,
71
+ "grad_norm": 14.250543594360352,
72
+ "learning_rate": 9.486992143456792e-05,
73
+ "loss": 0.6125,
74
+ "step": 800
75
+ },
76
+ {
77
+ "epoch": 1.64,
78
+ "grad_norm": 14.460857391357422,
79
+ "learning_rate": 9.353730385598887e-05,
80
+ "loss": 0.6703,
81
+ "step": 900
82
+ },
83
+ {
84
+ "epoch": 1.82,
85
+ "grad_norm": 16.239057540893555,
86
+ "learning_rate": 9.206267664155907e-05,
87
+ "loss": 0.5786,
88
+ "step": 1000
89
+ },
90
+ {
91
+ "epoch": 2.0,
92
+ "grad_norm": 25.278282165527344,
93
+ "learning_rate": 9.045084971874738e-05,
94
+ "loss": 0.6206,
95
+ "step": 1100
96
+ },
97
+ {
98
+ "epoch": 2.0,
99
+ "eval_accuracy": 0.8930417495029821,
100
+ "eval_loss": 0.37559983134269714,
101
+ "eval_runtime": 77.5566,
102
+ "eval_samples_per_second": 32.428,
103
+ "eval_steps_per_second": 1.019,
104
+ "step": 1100
105
+ },
106
+ {
107
+ "epoch": 2.18,
108
+ "grad_norm": 8.611919403076172,
109
+ "learning_rate": 8.870708053195413e-05,
110
+ "loss": 0.5204,
111
+ "step": 1200
112
+ },
113
+ {
114
+ "epoch": 2.36,
115
+ "grad_norm": 13.736300468444824,
116
+ "learning_rate": 8.683705689382024e-05,
117
+ "loss": 0.5279,
118
+ "step": 1300
119
+ },
120
+ {
121
+ "epoch": 2.55,
122
+ "grad_norm": 12.42533016204834,
123
+ "learning_rate": 8.484687843276469e-05,
124
+ "loss": 0.489,
125
+ "step": 1400
126
+ },
127
+ {
128
+ "epoch": 2.73,
129
+ "grad_norm": 15.548003196716309,
130
+ "learning_rate": 8.274303669726426e-05,
131
+ "loss": 0.5077,
132
+ "step": 1500
133
+ },
134
+ {
135
+ "epoch": 2.91,
136
+ "grad_norm": 14.506275177001953,
137
+ "learning_rate": 8.053239398177191e-05,
138
+ "loss": 0.5014,
139
+ "step": 1600
140
+ },
141
+ {
142
+ "epoch": 3.0,
143
+ "eval_accuracy": 0.9073558648111332,
144
+ "eval_loss": 0.3272249698638916,
145
+ "eval_runtime": 75.9338,
146
+ "eval_samples_per_second": 33.121,
147
+ "eval_steps_per_second": 1.04,
148
+ "step": 1650
149
+ },
150
+ {
151
+ "epoch": 3.09,
152
+ "grad_norm": 18.079133987426758,
153
+ "learning_rate": 7.822216094333847e-05,
154
+ "loss": 0.4807,
155
+ "step": 1700
156
+ },
157
+ {
158
+ "epoch": 3.27,
159
+ "grad_norm": 13.355294227600098,
160
+ "learning_rate": 7.58198730819481e-05,
161
+ "loss": 0.4562,
162
+ "step": 1800
163
+ },
164
+ {
165
+ "epoch": 3.45,
166
+ "grad_norm": 13.22020149230957,
167
+ "learning_rate": 7.333336616128369e-05,
168
+ "loss": 0.4124,
169
+ "step": 1900
170
+ },
171
+ {
172
+ "epoch": 3.64,
173
+ "grad_norm": 14.023577690124512,
174
+ "learning_rate": 7.077075065009433e-05,
175
+ "loss": 0.4643,
176
+ "step": 2000
177
+ },
178
+ {
179
+ "epoch": 3.82,
180
+ "grad_norm": 13.74384880065918,
181
+ "learning_rate": 6.814038526753205e-05,
182
+ "loss": 0.444,
183
+ "step": 2100
184
+ },
185
+ {
186
+ "epoch": 4.0,
187
+ "grad_norm": 16.15762710571289,
188
+ "learning_rate": 6.545084971874738e-05,
189
+ "loss": 0.3892,
190
+ "step": 2200
191
+ },
192
+ {
193
+ "epoch": 4.0,
194
+ "eval_accuracy": 0.9280318091451292,
195
+ "eval_loss": 0.27286219596862793,
196
+ "eval_runtime": 76.623,
197
+ "eval_samples_per_second": 32.823,
198
+ "eval_steps_per_second": 1.031,
199
+ "step": 2200
200
+ },
201
+ {
202
+ "epoch": 4.18,
203
+ "grad_norm": 16.897714614868164,
204
+ "learning_rate": 6.271091670967436e-05,
205
+ "loss": 0.3776,
206
+ "step": 2300
207
+ },
208
+ {
209
+ "epoch": 4.36,
210
+ "grad_norm": 9.421957015991211,
211
+ "learning_rate": 5.992952333228728e-05,
212
+ "loss": 0.3594,
213
+ "step": 2400
214
+ },
215
+ {
216
+ "epoch": 4.55,
217
+ "grad_norm": 9.755322456359863,
218
+ "learning_rate": 5.7115741913664264e-05,
219
+ "loss": 0.3712,
220
+ "step": 2500
221
+ },
222
+ {
223
+ "epoch": 4.73,
224
+ "grad_norm": 10.371528625488281,
225
+ "learning_rate": 5.427875042394199e-05,
226
+ "loss": 0.3862,
227
+ "step": 2600
228
+ },
229
+ {
230
+ "epoch": 4.91,
231
+ "grad_norm": 8.203412055969238,
232
+ "learning_rate": 5.142780253968481e-05,
233
+ "loss": 0.3819,
234
+ "step": 2700
235
+ },
236
+ {
237
+ "epoch": 5.0,
238
+ "eval_accuracy": 0.9204771371769384,
239
+ "eval_loss": 0.295335978269577,
240
+ "eval_runtime": 78.1287,
241
+ "eval_samples_per_second": 32.19,
242
+ "eval_steps_per_second": 1.011,
243
+ "step": 2750
244
+ },
245
+ {
246
+ "epoch": 5.09,
247
+ "grad_norm": 7.4753098487854,
248
+ "learning_rate": 4.85721974603152e-05,
249
+ "loss": 0.3352,
250
+ "step": 2800
251
+ },
252
+ {
253
+ "epoch": 5.27,
254
+ "grad_norm": 9.154393196105957,
255
+ "learning_rate": 4.5721249576058027e-05,
256
+ "loss": 0.3283,
257
+ "step": 2900
258
+ },
259
+ {
260
+ "epoch": 5.45,
261
+ "grad_norm": 11.729723930358887,
262
+ "learning_rate": 4.288425808633575e-05,
263
+ "loss": 0.3199,
264
+ "step": 3000
265
+ },
266
+ {
267
+ "epoch": 5.64,
268
+ "grad_norm": 11.630327224731445,
269
+ "learning_rate": 4.007047666771274e-05,
270
+ "loss": 0.3143,
271
+ "step": 3100
272
+ },
273
+ {
274
+ "epoch": 5.82,
275
+ "grad_norm": 4.333253860473633,
276
+ "learning_rate": 3.728908329032567e-05,
277
+ "loss": 0.3231,
278
+ "step": 3200
279
+ },
280
+ {
281
+ "epoch": 6.0,
282
+ "grad_norm": 18.835886001586914,
283
+ "learning_rate": 3.4549150281252636e-05,
284
+ "loss": 0.3315,
285
+ "step": 3300
286
+ },
287
+ {
288
+ "epoch": 6.0,
289
+ "eval_accuracy": 0.9260437375745527,
290
+ "eval_loss": 0.2783779799938202,
291
+ "eval_runtime": 76.4348,
292
+ "eval_samples_per_second": 32.904,
293
+ "eval_steps_per_second": 1.034,
294
+ "step": 3300
295
+ },
296
+ {
297
+ "epoch": 6.18,
298
+ "grad_norm": 5.403916835784912,
299
+ "learning_rate": 3.1859614732467954e-05,
300
+ "loss": 0.2781,
301
+ "step": 3400
302
+ },
303
+ {
304
+ "epoch": 6.36,
305
+ "grad_norm": 11.396268844604492,
306
+ "learning_rate": 2.9229249349905684e-05,
307
+ "loss": 0.2865,
308
+ "step": 3500
309
+ },
310
+ {
311
+ "epoch": 6.55,
312
+ "grad_norm": 15.219473838806152,
313
+ "learning_rate": 2.6666633838716314e-05,
314
+ "loss": 0.2839,
315
+ "step": 3600
316
+ },
317
+ {
318
+ "epoch": 6.73,
319
+ "grad_norm": 11.592494010925293,
320
+ "learning_rate": 2.418012691805191e-05,
321
+ "loss": 0.2549,
322
+ "step": 3700
323
+ },
324
+ {
325
+ "epoch": 6.91,
326
+ "grad_norm": 10.148946762084961,
327
+ "learning_rate": 2.1777839056661554e-05,
328
+ "loss": 0.281,
329
+ "step": 3800
330
+ },
331
+ {
332
+ "epoch": 7.0,
333
+ "eval_accuracy": 0.932803180914513,
334
+ "eval_loss": 0.2535076141357422,
335
+ "eval_runtime": 77.3926,
336
+ "eval_samples_per_second": 32.497,
337
+ "eval_steps_per_second": 1.021,
338
+ "step": 3850
339
+ },
340
+ {
341
+ "epoch": 7.09,
342
+ "grad_norm": 10.556535720825195,
343
+ "learning_rate": 1.946760601822809e-05,
344
+ "loss": 0.2247,
345
+ "step": 3900
346
+ },
347
+ {
348
+ "epoch": 7.27,
349
+ "grad_norm": 12.908746719360352,
350
+ "learning_rate": 1.725696330273575e-05,
351
+ "loss": 0.2302,
352
+ "step": 4000
353
+ },
354
+ {
355
+ "epoch": 7.45,
356
+ "grad_norm": 11.96335506439209,
357
+ "learning_rate": 1.5153121567235335e-05,
358
+ "loss": 0.2479,
359
+ "step": 4100
360
+ },
361
+ {
362
+ "epoch": 7.64,
363
+ "grad_norm": 13.116978645324707,
364
+ "learning_rate": 1.3162943106179749e-05,
365
+ "loss": 0.2423,
366
+ "step": 4200
367
+ },
368
+ {
369
+ "epoch": 7.82,
370
+ "grad_norm": 6.12368631362915,
371
+ "learning_rate": 1.1292919468045877e-05,
372
+ "loss": 0.2317,
373
+ "step": 4300
374
+ },
375
+ {
376
+ "epoch": 8.0,
377
+ "grad_norm": 31.24999237060547,
378
+ "learning_rate": 9.549150281252633e-06,
379
+ "loss": 0.217,
380
+ "step": 4400
381
+ },
382
+ {
383
+ "epoch": 8.0,
384
+ "eval_accuracy": 0.9343936381709742,
385
+ "eval_loss": 0.24977222084999084,
386
+ "eval_runtime": 75.2887,
387
+ "eval_samples_per_second": 33.405,
388
+ "eval_steps_per_second": 1.049,
389
+ "step": 4400
390
+ },
391
+ {
392
+ "epoch": 8.18,
393
+ "grad_norm": 12.625907897949219,
394
+ "learning_rate": 7.937323358440935e-06,
395
+ "loss": 0.208,
396
+ "step": 4500
397
+ },
398
+ {
399
+ "epoch": 8.36,
400
+ "grad_norm": 10.004912376403809,
401
+ "learning_rate": 6.462696144011149e-06,
402
+ "loss": 0.2105,
403
+ "step": 4600
404
+ },
405
+ {
406
+ "epoch": 8.55,
407
+ "grad_norm": 18.372529983520508,
408
+ "learning_rate": 5.13007856543209e-06,
409
+ "loss": 0.2038,
410
+ "step": 4700
411
+ },
412
+ {
413
+ "epoch": 8.73,
414
+ "grad_norm": 13.66908073425293,
415
+ "learning_rate": 3.9438173442575e-06,
416
+ "loss": 0.2216,
417
+ "step": 4800
418
+ },
419
+ {
420
+ "epoch": 8.91,
421
+ "grad_norm": 3.5467135906219482,
422
+ "learning_rate": 2.9077818180237693e-06,
423
+ "loss": 0.2152,
424
+ "step": 4900
425
+ },
426
+ {
427
+ "epoch": 9.0,
428
+ "eval_accuracy": 0.9399602385685885,
429
+ "eval_loss": 0.24052982032299042,
430
+ "eval_runtime": 72.694,
431
+ "eval_samples_per_second": 34.597,
432
+ "eval_steps_per_second": 1.087,
433
+ "step": 4950
434
+ },
435
+ {
436
+ "epoch": 9.09,
437
+ "grad_norm": 7.110690116882324,
438
+ "learning_rate": 2.0253513192751373e-06,
439
+ "loss": 0.1938,
440
+ "step": 5000
441
+ },
442
+ {
443
+ "epoch": 9.27,
444
+ "grad_norm": 8.859036445617676,
445
+ "learning_rate": 1.2994041528833266e-06,
446
+ "loss": 0.1902,
447
+ "step": 5100
448
+ },
449
+ {
450
+ "epoch": 9.45,
451
+ "grad_norm": 8.199995994567871,
452
+ "learning_rate": 7.323082076153509e-07,
453
+ "loss": 0.2104,
454
+ "step": 5200
455
+ },
456
+ {
457
+ "epoch": 9.64,
458
+ "grad_norm": 10.105381965637207,
459
+ "learning_rate": 3.2591323257248893e-07,
460
+ "loss": 0.2379,
461
+ "step": 5300
462
+ },
463
+ {
464
+ "epoch": 9.82,
465
+ "grad_norm": 14.23947525024414,
466
+ "learning_rate": 8.15448036932176e-08,
467
+ "loss": 0.217,
468
+ "step": 5400
469
+ },
470
+ {
471
+ "epoch": 10.0,
472
+ "grad_norm": 20.24100112915039,
473
+ "learning_rate": 0.0,
474
+ "loss": 0.2185,
475
+ "step": 5500
476
+ },
477
+ {
478
+ "epoch": 10.0,
479
+ "eval_accuracy": 0.9399602385685885,
480
+ "eval_loss": 0.23698414862155914,
481
+ "eval_runtime": 71.0392,
482
+ "eval_samples_per_second": 35.403,
483
+ "eval_steps_per_second": 1.112,
484
+ "step": 5500
485
+ },
486
+ {
487
+ "epoch": 10.0,
488
+ "step": 5500,
489
+ "total_flos": 1.301428412334932e+19,
490
+ "train_loss": 0.4333542574102228,
491
+ "train_runtime": 11179.012,
492
+ "train_samples_per_second": 15.727,
493
+ "train_steps_per_second": 0.492
494
+ }
495
+ ],
496
+ "logging_steps": 100,
497
+ "max_steps": 5500,
498
+ "num_input_tokens_seen": 0,
499
+ "num_train_epochs": 10,
500
+ "save_steps": 500,
501
+ "total_flos": 1.301428412334932e+19,
502
+ "train_batch_size": 32,
503
+ "trial_name": null,
504
+ "trial_params": null
505
+ }