sbaner24 commited on
Commit
a6d9bb0
1 Parent(s): 8d90e4d

End of training

Browse files
Files changed (4) hide show
  1. all_results.json +9 -9
  2. eval_results.json +5 -5
  3. train_results.json +5 -5
  4. trainer_state.json +317 -371
all_results.json CHANGED
@@ -1,12 +1,12 @@
1
  {
2
- "epoch": 40.0,
3
  "eval_accuracy": 1.0,
4
- "eval_loss": 0.06565026193857193,
5
- "eval_runtime": 0.226,
6
- "eval_samples_per_second": 119.477,
7
- "eval_steps_per_second": 4.425,
8
- "train_loss": 0.33988649606704713,
9
- "train_runtime": 421.7269,
10
- "train_samples_per_second": 28.573,
11
- "train_steps_per_second": 0.119
12
  }
 
1
  {
2
+ "epoch": 33.33,
3
  "eval_accuracy": 1.0,
4
+ "eval_loss": 0.10236643999814987,
5
+ "eval_runtime": 0.1221,
6
+ "eval_samples_per_second": 139.257,
7
+ "eval_steps_per_second": 8.192,
8
+ "train_loss": 0.26825646698474886,
9
+ "train_runtime": 260.0145,
10
+ "train_samples_per_second": 29.037,
11
+ "train_steps_per_second": 0.192
12
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 40.0,
3
  "eval_accuracy": 1.0,
4
- "eval_loss": 0.06565026193857193,
5
- "eval_runtime": 0.226,
6
- "eval_samples_per_second": 119.477,
7
- "eval_steps_per_second": 4.425
8
  }
 
1
  {
2
+ "epoch": 33.33,
3
  "eval_accuracy": 1.0,
4
+ "eval_loss": 0.10236643999814987,
5
+ "eval_runtime": 0.1221,
6
+ "eval_samples_per_second": 139.257,
7
+ "eval_steps_per_second": 8.192
8
  }
train_results.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "epoch": 40.0,
3
- "train_loss": 0.33988649606704713,
4
- "train_runtime": 421.7269,
5
- "train_samples_per_second": 28.573,
6
- "train_steps_per_second": 0.119
7
  }
 
1
  {
2
+ "epoch": 33.33,
3
+ "train_loss": 0.26825646698474886,
4
+ "train_runtime": 260.0145,
5
+ "train_samples_per_second": 29.037,
6
+ "train_steps_per_second": 0.192
7
  }
trainer_state.json CHANGED
@@ -1,685 +1,631 @@
1
  {
2
  "best_metric": 1.0,
3
- "best_model_checkpoint": "vit-base-patch16-224-Trial008-YEL_STEM2/checkpoint-25",
4
- "epoch": 40.0,
5
  "global_step": 50,
6
  "is_hyper_param_search": false,
7
  "is_local_process_zero": true,
8
  "is_world_process_zero": true,
9
  "log_history": [
10
  {
11
- "epoch": 0.8,
12
  "learning_rate": 1e-05,
13
- "loss": 0.711,
14
  "step": 1
15
  },
16
  {
17
- "epoch": 0.8,
18
- "eval_accuracy": 0.6296296296296297,
19
- "eval_loss": 0.6244667172431946,
20
- "eval_runtime": 0.1884,
21
- "eval_samples_per_second": 143.327,
22
- "eval_steps_per_second": 5.308,
23
  "step": 1
24
  },
25
  {
26
- "epoch": 1.6,
27
  "learning_rate": 2e-05,
28
- "loss": 0.6994,
29
  "step": 2
30
  },
31
  {
32
- "epoch": 1.6,
33
- "eval_accuracy": 0.7777777777777778,
34
- "eval_loss": 0.5949051976203918,
35
- "eval_runtime": 0.1691,
36
- "eval_samples_per_second": 159.667,
37
- "eval_steps_per_second": 5.914,
38
- "step": 2
39
- },
40
- {
41
- "epoch": 2.4,
42
  "learning_rate": 3e-05,
43
- "loss": 0.7097,
44
  "step": 3
45
  },
46
  {
47
- "epoch": 2.4,
48
- "eval_accuracy": 0.7777777777777778,
49
- "eval_loss": 0.5389389395713806,
50
- "eval_runtime": 0.1697,
51
- "eval_samples_per_second": 159.129,
52
- "eval_steps_per_second": 5.894,
53
  "step": 3
54
  },
55
  {
56
- "epoch": 3.2,
57
  "learning_rate": 4e-05,
58
- "loss": 0.6501,
59
  "step": 4
60
  },
61
  {
62
- "epoch": 4.0,
63
- "learning_rate": 5e-05,
64
- "loss": 0.5437,
65
- "step": 5
 
 
 
66
  },
67
  {
68
- "epoch": 4.0,
69
- "eval_accuracy": 0.7407407407407407,
70
- "eval_loss": 0.48060357570648193,
71
- "eval_runtime": 0.1681,
72
- "eval_samples_per_second": 160.574,
73
- "eval_steps_per_second": 5.947,
74
  "step": 5
75
  },
76
  {
77
- "epoch": 4.8,
78
  "learning_rate": 4.888888888888889e-05,
79
- "loss": 0.562,
80
  "step": 6
81
  },
82
  {
83
- "epoch": 4.8,
84
- "eval_accuracy": 0.8518518518518519,
85
- "eval_loss": 0.3975556194782257,
86
- "eval_runtime": 0.2988,
87
- "eval_samples_per_second": 90.348,
88
- "eval_steps_per_second": 3.346,
89
  "step": 6
90
  },
91
  {
92
- "epoch": 5.6,
93
  "learning_rate": 4.7777777777777784e-05,
94
- "loss": 0.431,
95
  "step": 7
96
  },
97
  {
98
- "epoch": 5.6,
99
- "eval_accuracy": 0.9259259259259259,
100
- "eval_loss": 0.3175361752510071,
101
- "eval_runtime": 0.1659,
102
- "eval_samples_per_second": 162.763,
103
- "eval_steps_per_second": 6.028,
104
  "step": 7
105
  },
106
  {
107
- "epoch": 6.4,
108
  "learning_rate": 4.666666666666667e-05,
109
- "loss": 0.4314,
110
  "step": 8
111
  },
112
  {
113
- "epoch": 6.4,
114
- "eval_accuracy": 0.9259259259259259,
115
- "eval_loss": 0.2999328076839447,
116
- "eval_runtime": 0.167,
117
- "eval_samples_per_second": 161.655,
118
- "eval_steps_per_second": 5.987,
119
- "step": 8
120
  },
121
  {
122
- "epoch": 7.2,
123
- "learning_rate": 4.555555555555556e-05,
124
- "loss": 0.3984,
 
 
 
125
  "step": 9
126
  },
127
  {
128
- "epoch": 8.0,
129
  "learning_rate": 4.4444444444444447e-05,
130
- "loss": 0.5044,
131
  "step": 10
132
  },
133
  {
134
- "epoch": 8.0,
135
- "eval_accuracy": 0.7777777777777778,
136
- "eval_loss": 0.42318493127822876,
137
- "eval_runtime": 0.1707,
138
- "eval_samples_per_second": 158.131,
139
- "eval_steps_per_second": 5.857,
140
  "step": 10
141
  },
142
  {
143
- "epoch": 8.8,
144
  "learning_rate": 4.3333333333333334e-05,
145
- "loss": 0.4276,
146
- "step": 11
147
- },
148
- {
149
- "epoch": 8.8,
150
- "eval_accuracy": 0.7777777777777778,
151
- "eval_loss": 0.37759944796562195,
152
- "eval_runtime": 0.1672,
153
- "eval_samples_per_second": 161.516,
154
- "eval_steps_per_second": 5.982,
155
  "step": 11
156
  },
157
  {
158
- "epoch": 9.6,
159
  "learning_rate": 4.222222222222222e-05,
160
- "loss": 0.7983,
161
  "step": 12
162
  },
163
  {
164
- "epoch": 9.6,
165
- "eval_accuracy": 0.8148148148148148,
166
- "eval_loss": 0.3195733428001404,
167
- "eval_runtime": 0.1664,
168
- "eval_samples_per_second": 162.293,
169
- "eval_steps_per_second": 6.011,
170
  "step": 12
171
  },
172
  {
173
- "epoch": 10.4,
174
  "learning_rate": 4.111111111111111e-05,
175
- "loss": 0.6141,
176
  "step": 13
177
  },
178
  {
179
- "epoch": 10.4,
180
- "eval_accuracy": 0.8888888888888888,
181
- "eval_loss": 0.2574685513973236,
182
- "eval_runtime": 0.1759,
183
- "eval_samples_per_second": 153.499,
184
- "eval_steps_per_second": 5.685,
185
  "step": 13
186
  },
187
  {
188
- "epoch": 11.2,
189
  "learning_rate": 4e-05,
190
- "loss": 0.3103,
191
  "step": 14
192
  },
193
  {
194
- "epoch": 12.0,
195
  "learning_rate": 3.888888888888889e-05,
196
- "loss": 0.3859,
197
  "step": 15
198
  },
199
  {
200
- "epoch": 12.0,
201
- "eval_accuracy": 0.8148148148148148,
202
- "eval_loss": 0.3282352089881897,
203
- "eval_runtime": 0.1708,
204
- "eval_samples_per_second": 158.12,
205
- "eval_steps_per_second": 5.856,
206
  "step": 15
207
  },
208
  {
209
- "epoch": 12.8,
210
  "learning_rate": 3.777777777777778e-05,
211
- "loss": 0.36,
212
  "step": 16
213
  },
214
  {
215
- "epoch": 12.8,
216
- "eval_accuracy": 0.8148148148148148,
217
- "eval_loss": 0.32574817538261414,
218
- "eval_runtime": 0.1663,
219
- "eval_samples_per_second": 162.347,
220
- "eval_steps_per_second": 6.013,
221
  "step": 16
222
  },
223
  {
224
- "epoch": 13.6,
225
  "learning_rate": 3.6666666666666666e-05,
226
- "loss": 0.2369,
227
  "step": 17
228
  },
229
  {
230
- "epoch": 13.6,
231
- "eval_accuracy": 0.8888888888888888,
232
- "eval_loss": 0.20562799274921417,
233
- "eval_runtime": 0.1749,
234
- "eval_samples_per_second": 154.356,
235
- "eval_steps_per_second": 5.717,
236
- "step": 17
237
- },
238
- {
239
- "epoch": 14.4,
240
  "learning_rate": 3.555555555555556e-05,
241
- "loss": 0.2539,
242
  "step": 18
243
  },
244
  {
245
- "epoch": 14.4,
246
- "eval_accuracy": 0.9629629629629629,
247
- "eval_loss": 0.13304640352725983,
248
- "eval_runtime": 0.1693,
249
- "eval_samples_per_second": 159.465,
250
- "eval_steps_per_second": 5.906,
251
  "step": 18
252
  },
253
  {
254
- "epoch": 15.2,
255
  "learning_rate": 3.444444444444445e-05,
256
- "loss": 0.2318,
257
  "step": 19
258
  },
259
  {
260
- "epoch": 16.0,
261
- "learning_rate": 3.3333333333333335e-05,
262
- "loss": 0.2351,
263
- "step": 20
 
 
 
264
  },
265
  {
266
- "epoch": 16.0,
267
- "eval_accuracy": 0.9259259259259259,
268
- "eval_loss": 0.1432814747095108,
269
- "eval_runtime": 0.174,
270
- "eval_samples_per_second": 155.172,
271
- "eval_steps_per_second": 5.747,
272
  "step": 20
273
  },
274
  {
275
- "epoch": 16.8,
276
  "learning_rate": 3.222222222222223e-05,
277
- "loss": 0.2446,
278
  "step": 21
279
  },
280
  {
281
- "epoch": 16.8,
282
- "eval_accuracy": 0.8888888888888888,
283
- "eval_loss": 0.17627812922000885,
284
- "eval_runtime": 0.1795,
285
- "eval_samples_per_second": 150.378,
286
- "eval_steps_per_second": 5.57,
287
  "step": 21
288
  },
289
  {
290
- "epoch": 17.6,
291
  "learning_rate": 3.111111111111111e-05,
292
- "loss": 0.2202,
293
  "step": 22
294
  },
295
  {
296
- "epoch": 17.6,
297
- "eval_accuracy": 0.8888888888888888,
298
- "eval_loss": 0.17611810564994812,
299
- "eval_runtime": 0.1748,
300
- "eval_samples_per_second": 154.478,
301
- "eval_steps_per_second": 5.721,
302
  "step": 22
303
  },
304
  {
305
- "epoch": 18.4,
306
  "learning_rate": 3e-05,
307
- "loss": 0.2028,
308
  "step": 23
309
  },
310
  {
311
- "epoch": 18.4,
312
- "eval_accuracy": 0.9629629629629629,
313
- "eval_loss": 0.12490027397871017,
314
- "eval_runtime": 0.1747,
315
- "eval_samples_per_second": 154.56,
316
- "eval_steps_per_second": 5.724,
317
- "step": 23
318
  },
319
  {
320
- "epoch": 19.2,
321
- "learning_rate": 2.8888888888888888e-05,
322
- "loss": 0.2794,
 
 
 
323
  "step": 24
324
  },
325
  {
326
- "epoch": 20.0,
327
  "learning_rate": 2.777777777777778e-05,
328
- "loss": 0.1888,
329
  "step": 25
330
  },
331
  {
332
- "epoch": 20.0,
333
  "eval_accuracy": 1.0,
334
- "eval_loss": 0.06565026193857193,
335
- "eval_runtime": 0.1681,
336
- "eval_samples_per_second": 160.654,
337
- "eval_steps_per_second": 5.95,
338
  "step": 25
339
  },
340
  {
341
- "epoch": 20.8,
342
  "learning_rate": 2.6666666666666667e-05,
343
- "loss": 0.2796,
344
  "step": 26
345
  },
346
  {
347
- "epoch": 20.8,
348
- "eval_accuracy": 1.0,
349
- "eval_loss": 0.05862481892108917,
350
- "eval_runtime": 0.1702,
351
- "eval_samples_per_second": 158.644,
352
- "eval_steps_per_second": 5.876,
353
- "step": 26
354
- },
355
- {
356
- "epoch": 21.6,
357
  "learning_rate": 2.5555555555555554e-05,
358
- "loss": 0.1939,
359
  "step": 27
360
  },
361
  {
362
- "epoch": 21.6,
363
  "eval_accuracy": 1.0,
364
- "eval_loss": 0.04629632458090782,
365
- "eval_runtime": 0.1688,
366
- "eval_samples_per_second": 159.947,
367
- "eval_steps_per_second": 5.924,
368
  "step": 27
369
  },
370
  {
371
- "epoch": 22.4,
372
  "learning_rate": 2.4444444444444445e-05,
373
- "loss": 0.1829,
374
  "step": 28
375
  },
376
  {
377
- "epoch": 22.4,
378
- "eval_accuracy": 0.9629629629629629,
379
- "eval_loss": 0.05591176077723503,
380
- "eval_runtime": 0.1979,
381
- "eval_samples_per_second": 136.426,
382
- "eval_steps_per_second": 5.053,
383
  "step": 28
384
  },
385
  {
386
- "epoch": 23.2,
387
  "learning_rate": 2.3333333333333336e-05,
388
- "loss": 0.6172,
389
  "step": 29
390
  },
391
  {
392
- "epoch": 24.0,
393
  "learning_rate": 2.2222222222222223e-05,
394
- "loss": 0.2094,
395
  "step": 30
396
  },
397
  {
398
- "epoch": 24.0,
399
- "eval_accuracy": 0.9259259259259259,
400
- "eval_loss": 0.1417052000761032,
401
- "eval_runtime": 0.1738,
402
- "eval_samples_per_second": 155.368,
403
- "eval_steps_per_second": 5.754,
404
  "step": 30
405
  },
406
  {
407
- "epoch": 24.8,
408
  "learning_rate": 2.111111111111111e-05,
409
- "loss": 0.2454,
410
  "step": 31
411
  },
412
  {
413
- "epoch": 24.8,
414
- "eval_accuracy": 0.9259259259259259,
415
- "eval_loss": 0.14769193530082703,
416
- "eval_runtime": 0.1664,
417
- "eval_samples_per_second": 162.215,
418
- "eval_steps_per_second": 6.008,
419
  "step": 31
420
  },
421
  {
422
- "epoch": 25.6,
423
  "learning_rate": 2e-05,
424
- "loss": 0.4863,
425
  "step": 32
426
  },
427
  {
428
- "epoch": 25.6,
429
- "eval_accuracy": 0.9259259259259259,
430
- "eval_loss": 0.13248541951179504,
431
- "eval_runtime": 0.1763,
432
- "eval_samples_per_second": 153.175,
433
- "eval_steps_per_second": 5.673,
434
- "step": 32
435
- },
436
- {
437
- "epoch": 26.4,
438
  "learning_rate": 1.888888888888889e-05,
439
- "loss": 0.398,
440
  "step": 33
441
  },
442
  {
443
- "epoch": 26.4,
444
- "eval_accuracy": 0.9259259259259259,
445
- "eval_loss": 0.09621105343103409,
446
- "eval_runtime": 0.1741,
447
- "eval_samples_per_second": 155.06,
448
- "eval_steps_per_second": 5.743,
449
  "step": 33
450
  },
451
  {
452
- "epoch": 27.2,
453
  "learning_rate": 1.777777777777778e-05,
454
- "loss": 0.2793,
455
  "step": 34
456
  },
457
  {
458
- "epoch": 28.0,
459
- "learning_rate": 1.6666666666666667e-05,
460
- "loss": 0.1589,
461
- "step": 35
 
 
 
462
  },
463
  {
464
- "epoch": 28.0,
465
- "eval_accuracy": 1.0,
466
- "eval_loss": 0.05301348865032196,
467
- "eval_runtime": 0.1697,
468
- "eval_samples_per_second": 159.125,
469
- "eval_steps_per_second": 5.894,
470
  "step": 35
471
  },
472
  {
473
- "epoch": 28.8,
474
  "learning_rate": 1.5555555555555555e-05,
475
- "loss": 0.1964,
476
  "step": 36
477
  },
478
  {
479
- "epoch": 28.8,
480
  "eval_accuracy": 1.0,
481
- "eval_loss": 0.04289659485220909,
482
- "eval_runtime": 0.167,
483
- "eval_samples_per_second": 161.679,
484
- "eval_steps_per_second": 5.988,
485
  "step": 36
486
  },
487
  {
488
- "epoch": 29.6,
489
  "learning_rate": 1.4444444444444444e-05,
490
- "loss": 0.1684,
491
  "step": 37
492
  },
493
  {
494
- "epoch": 29.6,
495
  "eval_accuracy": 1.0,
496
- "eval_loss": 0.03966989740729332,
497
- "eval_runtime": 0.1681,
498
- "eval_samples_per_second": 160.571,
499
- "eval_steps_per_second": 5.947,
500
  "step": 37
501
  },
502
  {
503
- "epoch": 30.4,
504
  "learning_rate": 1.3333333333333333e-05,
505
- "loss": 0.1793,
506
  "step": 38
507
  },
508
  {
509
- "epoch": 30.4,
510
- "eval_accuracy": 1.0,
511
- "eval_loss": 0.036163847893476486,
512
- "eval_runtime": 0.1711,
513
- "eval_samples_per_second": 157.765,
514
- "eval_steps_per_second": 5.843,
515
- "step": 38
516
  },
517
  {
518
- "epoch": 31.2,
519
- "learning_rate": 1.2222222222222222e-05,
520
- "loss": 0.2068,
 
 
 
521
  "step": 39
522
  },
523
  {
524
- "epoch": 32.0,
525
  "learning_rate": 1.1111111111111112e-05,
526
- "loss": 0.1572,
527
  "step": 40
528
  },
529
  {
530
- "epoch": 32.0,
531
  "eval_accuracy": 1.0,
532
- "eval_loss": 0.03249576315283775,
533
- "eval_runtime": 0.1743,
534
- "eval_samples_per_second": 154.882,
535
- "eval_steps_per_second": 5.736,
536
  "step": 40
537
  },
538
  {
539
- "epoch": 32.8,
540
  "learning_rate": 1e-05,
541
- "loss": 0.2235,
542
- "step": 41
543
- },
544
- {
545
- "epoch": 32.8,
546
- "eval_accuracy": 1.0,
547
- "eval_loss": 0.030854476615786552,
548
- "eval_runtime": 0.1698,
549
- "eval_samples_per_second": 158.993,
550
- "eval_steps_per_second": 5.889,
551
  "step": 41
552
  },
553
  {
554
- "epoch": 33.6,
555
  "learning_rate": 8.88888888888889e-06,
556
- "loss": 0.203,
557
  "step": 42
558
  },
559
  {
560
- "epoch": 33.6,
561
  "eval_accuracy": 1.0,
562
- "eval_loss": 0.028395000845193863,
563
- "eval_runtime": 0.1749,
564
- "eval_samples_per_second": 154.363,
565
- "eval_steps_per_second": 5.717,
566
  "step": 42
567
  },
568
  {
569
- "epoch": 34.4,
570
  "learning_rate": 7.777777777777777e-06,
571
- "loss": 0.4193,
572
  "step": 43
573
  },
574
  {
575
- "epoch": 34.4,
576
  "eval_accuracy": 1.0,
577
- "eval_loss": 0.02703581191599369,
578
- "eval_runtime": 0.167,
579
- "eval_samples_per_second": 161.692,
580
- "eval_steps_per_second": 5.989,
581
  "step": 43
582
  },
583
  {
584
- "epoch": 35.2,
585
  "learning_rate": 6.666666666666667e-06,
586
- "loss": 0.1791,
587
  "step": 44
588
  },
589
  {
590
- "epoch": 36.0,
591
  "learning_rate": 5.555555555555556e-06,
592
- "loss": 0.1704,
593
  "step": 45
594
  },
595
  {
596
- "epoch": 36.0,
597
  "eval_accuracy": 1.0,
598
- "eval_loss": 0.024205761030316353,
599
- "eval_runtime": 0.1736,
600
- "eval_samples_per_second": 155.486,
601
- "eval_steps_per_second": 5.759,
602
  "step": 45
603
  },
604
  {
605
- "epoch": 36.8,
606
  "learning_rate": 4.444444444444445e-06,
607
- "loss": 0.2349,
608
  "step": 46
609
  },
610
  {
611
- "epoch": 36.8,
612
  "eval_accuracy": 1.0,
613
- "eval_loss": 0.02339680679142475,
614
- "eval_runtime": 0.1662,
615
- "eval_samples_per_second": 162.408,
616
- "eval_steps_per_second": 6.015,
617
  "step": 46
618
  },
619
  {
620
- "epoch": 37.6,
621
  "learning_rate": 3.3333333333333333e-06,
622
- "loss": 0.1324,
623
- "step": 47
624
- },
625
- {
626
- "epoch": 37.6,
627
- "eval_accuracy": 1.0,
628
- "eval_loss": 0.02463681250810623,
629
- "eval_runtime": 0.1666,
630
- "eval_samples_per_second": 162.113,
631
- "eval_steps_per_second": 6.004,
632
  "step": 47
633
  },
634
  {
635
- "epoch": 38.4,
636
  "learning_rate": 2.2222222222222225e-06,
637
- "loss": 0.3921,
638
  "step": 48
639
  },
640
  {
641
- "epoch": 38.4,
642
  "eval_accuracy": 1.0,
643
- "eval_loss": 0.025700703263282776,
644
- "eval_runtime": 0.1717,
645
- "eval_samples_per_second": 157.29,
646
- "eval_steps_per_second": 5.826,
647
  "step": 48
648
  },
649
  {
650
- "epoch": 39.2,
651
  "learning_rate": 1.1111111111111112e-06,
652
- "loss": 0.2879,
 
 
 
 
 
 
 
 
 
653
  "step": 49
654
  },
655
  {
656
- "epoch": 40.0,
657
  "learning_rate": 0.0,
658
- "loss": 0.1619,
659
  "step": 50
660
  },
661
  {
662
- "epoch": 40.0,
663
  "eval_accuracy": 1.0,
664
- "eval_loss": 0.026837240904569626,
665
- "eval_runtime": 0.1752,
666
- "eval_samples_per_second": 154.138,
667
- "eval_steps_per_second": 5.709,
668
  "step": 50
669
  },
670
  {
671
- "epoch": 40.0,
672
  "step": 50,
673
- "total_flos": 7.470227798846669e+17,
674
- "train_loss": 0.33988649606704713,
675
- "train_runtime": 421.7269,
676
- "train_samples_per_second": 28.573,
677
- "train_steps_per_second": 0.119
678
  }
679
  ],
680
  "max_steps": 50,
681
  "num_train_epochs": 50,
682
- "total_flos": 7.470227798846669e+17,
683
  "trial_name": null,
684
  "trial_params": null
685
  }
 
1
  {
2
  "best_metric": 1.0,
3
+ "best_model_checkpoint": "vit-base-patch16-224-Trial008-YEL_STEM2/checkpoint-10",
4
+ "epoch": 33.333333333333336,
5
  "global_step": 50,
6
  "is_hyper_param_search": false,
7
  "is_local_process_zero": true,
8
  "is_world_process_zero": true,
9
  "log_history": [
10
  {
11
+ "epoch": 0.67,
12
  "learning_rate": 1e-05,
13
+ "loss": 0.7598,
14
  "step": 1
15
  },
16
  {
17
+ "epoch": 0.67,
18
+ "eval_accuracy": 0.6470588235294118,
19
+ "eval_loss": 0.6967736482620239,
20
+ "eval_runtime": 0.1155,
21
+ "eval_samples_per_second": 147.167,
22
+ "eval_steps_per_second": 8.657,
23
  "step": 1
24
  },
25
  {
26
+ "epoch": 1.33,
27
  "learning_rate": 2e-05,
28
+ "loss": 0.7926,
29
  "step": 2
30
  },
31
  {
32
+ "epoch": 2.0,
 
 
 
 
 
 
 
 
 
33
  "learning_rate": 3e-05,
34
+ "loss": 0.7294,
35
  "step": 3
36
  },
37
  {
38
+ "epoch": 2.0,
39
+ "eval_accuracy": 0.47058823529411764,
40
+ "eval_loss": 0.766238808631897,
41
+ "eval_runtime": 0.1178,
42
+ "eval_samples_per_second": 144.35,
43
+ "eval_steps_per_second": 8.491,
44
  "step": 3
45
  },
46
  {
47
+ "epoch": 2.67,
48
  "learning_rate": 4e-05,
49
+ "loss": 0.6662,
50
  "step": 4
51
  },
52
  {
53
+ "epoch": 2.67,
54
+ "eval_accuracy": 0.5882352941176471,
55
+ "eval_loss": 0.7196066379547119,
56
+ "eval_runtime": 0.1128,
57
+ "eval_samples_per_second": 150.728,
58
+ "eval_steps_per_second": 8.866,
59
+ "step": 4
60
  },
61
  {
62
+ "epoch": 3.33,
63
+ "learning_rate": 5e-05,
64
+ "loss": 0.6872,
 
 
 
65
  "step": 5
66
  },
67
  {
68
+ "epoch": 4.0,
69
  "learning_rate": 4.888888888888889e-05,
70
+ "loss": 0.5662,
71
  "step": 6
72
  },
73
  {
74
+ "epoch": 4.0,
75
+ "eval_accuracy": 0.8235294117647058,
76
+ "eval_loss": 0.3941333293914795,
77
+ "eval_runtime": 0.1124,
78
+ "eval_samples_per_second": 151.225,
79
+ "eval_steps_per_second": 8.896,
80
  "step": 6
81
  },
82
  {
83
+ "epoch": 4.67,
84
  "learning_rate": 4.7777777777777784e-05,
85
+ "loss": 0.4781,
86
  "step": 7
87
  },
88
  {
89
+ "epoch": 4.67,
90
+ "eval_accuracy": 0.8235294117647058,
91
+ "eval_loss": 0.3458152413368225,
92
+ "eval_runtime": 0.1128,
93
+ "eval_samples_per_second": 150.724,
94
+ "eval_steps_per_second": 8.866,
95
  "step": 7
96
  },
97
  {
98
+ "epoch": 5.33,
99
  "learning_rate": 4.666666666666667e-05,
100
+ "loss": 0.3621,
101
  "step": 8
102
  },
103
  {
104
+ "epoch": 6.0,
105
+ "learning_rate": 4.555555555555556e-05,
106
+ "loss": 0.3259,
107
+ "step": 9
 
 
 
108
  },
109
  {
110
+ "epoch": 6.0,
111
+ "eval_accuracy": 0.9411764705882353,
112
+ "eval_loss": 0.16989070177078247,
113
+ "eval_runtime": 0.1116,
114
+ "eval_samples_per_second": 152.266,
115
+ "eval_steps_per_second": 8.957,
116
  "step": 9
117
  },
118
  {
119
+ "epoch": 6.67,
120
  "learning_rate": 4.4444444444444447e-05,
121
+ "loss": 0.2903,
122
  "step": 10
123
  },
124
  {
125
+ "epoch": 6.67,
126
+ "eval_accuracy": 1.0,
127
+ "eval_loss": 0.10236643999814987,
128
+ "eval_runtime": 0.112,
129
+ "eval_samples_per_second": 151.781,
130
+ "eval_steps_per_second": 8.928,
131
  "step": 10
132
  },
133
  {
134
+ "epoch": 7.33,
135
  "learning_rate": 4.3333333333333334e-05,
136
+ "loss": 0.2539,
 
 
 
 
 
 
 
 
 
137
  "step": 11
138
  },
139
  {
140
+ "epoch": 8.0,
141
  "learning_rate": 4.222222222222222e-05,
142
+ "loss": 0.2206,
143
  "step": 12
144
  },
145
  {
146
+ "epoch": 8.0,
147
+ "eval_accuracy": 1.0,
148
+ "eval_loss": 0.07876813411712646,
149
+ "eval_runtime": 0.1154,
150
+ "eval_samples_per_second": 147.279,
151
+ "eval_steps_per_second": 8.663,
152
  "step": 12
153
  },
154
  {
155
+ "epoch": 8.67,
156
  "learning_rate": 4.111111111111111e-05,
157
+ "loss": 0.3215,
158
  "step": 13
159
  },
160
  {
161
+ "epoch": 8.67,
162
+ "eval_accuracy": 1.0,
163
+ "eval_loss": 0.04137047752737999,
164
+ "eval_runtime": 0.1141,
165
+ "eval_samples_per_second": 148.944,
166
+ "eval_steps_per_second": 8.761,
167
  "step": 13
168
  },
169
  {
170
+ "epoch": 9.33,
171
  "learning_rate": 4e-05,
172
+ "loss": 0.1603,
173
  "step": 14
174
  },
175
  {
176
+ "epoch": 10.0,
177
  "learning_rate": 3.888888888888889e-05,
178
+ "loss": 0.1741,
179
  "step": 15
180
  },
181
  {
182
+ "epoch": 10.0,
183
+ "eval_accuracy": 1.0,
184
+ "eval_loss": 0.021776309236884117,
185
+ "eval_runtime": 0.1108,
186
+ "eval_samples_per_second": 153.382,
187
+ "eval_steps_per_second": 9.022,
188
  "step": 15
189
  },
190
  {
191
+ "epoch": 10.67,
192
  "learning_rate": 3.777777777777778e-05,
193
+ "loss": 0.2222,
194
  "step": 16
195
  },
196
  {
197
+ "epoch": 10.67,
198
+ "eval_accuracy": 1.0,
199
+ "eval_loss": 0.020698698237538338,
200
+ "eval_runtime": 0.1126,
201
+ "eval_samples_per_second": 151.006,
202
+ "eval_steps_per_second": 8.883,
203
  "step": 16
204
  },
205
  {
206
+ "epoch": 11.33,
207
  "learning_rate": 3.6666666666666666e-05,
208
+ "loss": 0.228,
209
  "step": 17
210
  },
211
  {
212
+ "epoch": 12.0,
 
 
 
 
 
 
 
 
 
213
  "learning_rate": 3.555555555555556e-05,
214
+ "loss": 0.1534,
215
  "step": 18
216
  },
217
  {
218
+ "epoch": 12.0,
219
+ "eval_accuracy": 1.0,
220
+ "eval_loss": 0.012763559818267822,
221
+ "eval_runtime": 0.1133,
222
+ "eval_samples_per_second": 150.062,
223
+ "eval_steps_per_second": 8.827,
224
  "step": 18
225
  },
226
  {
227
+ "epoch": 12.67,
228
  "learning_rate": 3.444444444444445e-05,
229
+ "loss": 0.273,
230
  "step": 19
231
  },
232
  {
233
+ "epoch": 12.67,
234
+ "eval_accuracy": 1.0,
235
+ "eval_loss": 0.010324841365218163,
236
+ "eval_runtime": 0.1154,
237
+ "eval_samples_per_second": 147.298,
238
+ "eval_steps_per_second": 8.665,
239
+ "step": 19
240
  },
241
  {
242
+ "epoch": 13.33,
243
+ "learning_rate": 3.3333333333333335e-05,
244
+ "loss": 0.3743,
 
 
 
245
  "step": 20
246
  },
247
  {
248
+ "epoch": 14.0,
249
  "learning_rate": 3.222222222222223e-05,
250
+ "loss": 0.2021,
251
  "step": 21
252
  },
253
  {
254
+ "epoch": 14.0,
255
+ "eval_accuracy": 1.0,
256
+ "eval_loss": 0.007954830303788185,
257
+ "eval_runtime": 0.1144,
258
+ "eval_samples_per_second": 148.646,
259
+ "eval_steps_per_second": 8.744,
260
  "step": 21
261
  },
262
  {
263
+ "epoch": 14.67,
264
  "learning_rate": 3.111111111111111e-05,
265
+ "loss": 0.2193,
266
  "step": 22
267
  },
268
  {
269
+ "epoch": 14.67,
270
+ "eval_accuracy": 1.0,
271
+ "eval_loss": 0.009969166480004787,
272
+ "eval_runtime": 0.113,
273
+ "eval_samples_per_second": 150.48,
274
+ "eval_steps_per_second": 8.852,
275
  "step": 22
276
  },
277
  {
278
+ "epoch": 15.33,
279
  "learning_rate": 3e-05,
280
+ "loss": 0.1586,
281
  "step": 23
282
  },
283
  {
284
+ "epoch": 16.0,
285
+ "learning_rate": 2.8888888888888888e-05,
286
+ "loss": 0.2132,
287
+ "step": 24
 
 
 
288
  },
289
  {
290
+ "epoch": 16.0,
291
+ "eval_accuracy": 1.0,
292
+ "eval_loss": 0.02467266283929348,
293
+ "eval_runtime": 0.114,
294
+ "eval_samples_per_second": 149.139,
295
+ "eval_steps_per_second": 8.773,
296
  "step": 24
297
  },
298
  {
299
+ "epoch": 16.67,
300
  "learning_rate": 2.777777777777778e-05,
301
+ "loss": 0.2163,
302
  "step": 25
303
  },
304
  {
305
+ "epoch": 16.67,
306
  "eval_accuracy": 1.0,
307
+ "eval_loss": 0.02664419636130333,
308
+ "eval_runtime": 0.1146,
309
+ "eval_samples_per_second": 148.38,
310
+ "eval_steps_per_second": 8.728,
311
  "step": 25
312
  },
313
  {
314
+ "epoch": 17.33,
315
  "learning_rate": 2.6666666666666667e-05,
316
+ "loss": 0.2017,
317
  "step": 26
318
  },
319
  {
320
+ "epoch": 18.0,
 
 
 
 
 
 
 
 
 
321
  "learning_rate": 2.5555555555555554e-05,
322
+ "loss": 0.1626,
323
  "step": 27
324
  },
325
  {
326
+ "epoch": 18.0,
327
  "eval_accuracy": 1.0,
328
+ "eval_loss": 0.01014714501798153,
329
+ "eval_runtime": 0.1128,
330
+ "eval_samples_per_second": 150.755,
331
+ "eval_steps_per_second": 8.868,
332
  "step": 27
333
  },
334
  {
335
+ "epoch": 18.67,
336
  "learning_rate": 2.4444444444444445e-05,
337
+ "loss": 0.2492,
338
  "step": 28
339
  },
340
  {
341
+ "epoch": 18.67,
342
+ "eval_accuracy": 1.0,
343
+ "eval_loss": 0.005902642849832773,
344
+ "eval_runtime": 0.1149,
345
+ "eval_samples_per_second": 147.968,
346
+ "eval_steps_per_second": 8.704,
347
  "step": 28
348
  },
349
  {
350
+ "epoch": 19.33,
351
  "learning_rate": 2.3333333333333336e-05,
352
+ "loss": 0.1315,
353
  "step": 29
354
  },
355
  {
356
+ "epoch": 20.0,
357
  "learning_rate": 2.2222222222222223e-05,
358
+ "loss": 0.1308,
359
  "step": 30
360
  },
361
  {
362
+ "epoch": 20.0,
363
+ "eval_accuracy": 1.0,
364
+ "eval_loss": 0.005620644427835941,
365
+ "eval_runtime": 0.1147,
366
+ "eval_samples_per_second": 148.149,
367
+ "eval_steps_per_second": 8.715,
368
  "step": 30
369
  },
370
  {
371
+ "epoch": 20.67,
372
  "learning_rate": 2.111111111111111e-05,
373
+ "loss": 0.2144,
374
  "step": 31
375
  },
376
  {
377
+ "epoch": 20.67,
378
+ "eval_accuracy": 1.0,
379
+ "eval_loss": 0.006035304628312588,
380
+ "eval_runtime": 0.1154,
381
+ "eval_samples_per_second": 147.274,
382
+ "eval_steps_per_second": 8.663,
383
  "step": 31
384
  },
385
  {
386
+ "epoch": 21.33,
387
  "learning_rate": 2e-05,
388
+ "loss": 0.1608,
389
  "step": 32
390
  },
391
  {
392
+ "epoch": 22.0,
 
 
 
 
 
 
 
 
 
393
  "learning_rate": 1.888888888888889e-05,
394
+ "loss": 0.1389,
395
  "step": 33
396
  },
397
  {
398
+ "epoch": 22.0,
399
+ "eval_accuracy": 1.0,
400
+ "eval_loss": 0.004421359859406948,
401
+ "eval_runtime": 0.1131,
402
+ "eval_samples_per_second": 150.363,
403
+ "eval_steps_per_second": 8.845,
404
  "step": 33
405
  },
406
  {
407
+ "epoch": 22.67,
408
  "learning_rate": 1.777777777777778e-05,
409
+ "loss": 0.2548,
410
  "step": 34
411
  },
412
  {
413
+ "epoch": 22.67,
414
+ "eval_accuracy": 1.0,
415
+ "eval_loss": 0.004027603659778833,
416
+ "eval_runtime": 0.113,
417
+ "eval_samples_per_second": 150.382,
418
+ "eval_steps_per_second": 8.846,
419
+ "step": 34
420
  },
421
  {
422
+ "epoch": 23.33,
423
+ "learning_rate": 1.6666666666666667e-05,
424
+ "loss": 0.1916,
 
 
 
425
  "step": 35
426
  },
427
  {
428
+ "epoch": 24.0,
429
  "learning_rate": 1.5555555555555555e-05,
430
+ "loss": 0.1324,
431
  "step": 36
432
  },
433
  {
434
+ "epoch": 24.0,
435
  "eval_accuracy": 1.0,
436
+ "eval_loss": 0.0037342249415814877,
437
+ "eval_runtime": 0.1133,
438
+ "eval_samples_per_second": 150.098,
439
+ "eval_steps_per_second": 8.829,
440
  "step": 36
441
  },
442
  {
443
+ "epoch": 24.67,
444
  "learning_rate": 1.4444444444444444e-05,
445
+ "loss": 0.1958,
446
  "step": 37
447
  },
448
  {
449
+ "epoch": 24.67,
450
  "eval_accuracy": 1.0,
451
+ "eval_loss": 0.0036163795739412308,
452
+ "eval_runtime": 0.1138,
453
+ "eval_samples_per_second": 149.366,
454
+ "eval_steps_per_second": 8.786,
455
  "step": 37
456
  },
457
  {
458
+ "epoch": 25.33,
459
  "learning_rate": 1.3333333333333333e-05,
460
+ "loss": 0.1812,
461
  "step": 38
462
  },
463
  {
464
+ "epoch": 26.0,
465
+ "learning_rate": 1.2222222222222222e-05,
466
+ "loss": 0.2476,
467
+ "step": 39
 
 
 
468
  },
469
  {
470
+ "epoch": 26.0,
471
+ "eval_accuracy": 1.0,
472
+ "eval_loss": 0.0034641919191926718,
473
+ "eval_runtime": 0.1123,
474
+ "eval_samples_per_second": 151.377,
475
+ "eval_steps_per_second": 8.905,
476
  "step": 39
477
  },
478
  {
479
+ "epoch": 26.67,
480
  "learning_rate": 1.1111111111111112e-05,
481
+ "loss": 0.1439,
482
  "step": 40
483
  },
484
  {
485
+ "epoch": 26.67,
486
  "eval_accuracy": 1.0,
487
+ "eval_loss": 0.0033218893222510815,
488
+ "eval_runtime": 0.116,
489
+ "eval_samples_per_second": 146.604,
490
+ "eval_steps_per_second": 8.624,
491
  "step": 40
492
  },
493
  {
494
+ "epoch": 27.33,
495
  "learning_rate": 1e-05,
496
+ "loss": 0.1375,
 
 
 
 
 
 
 
 
 
497
  "step": 41
498
  },
499
  {
500
+ "epoch": 28.0,
501
  "learning_rate": 8.88888888888889e-06,
502
+ "loss": 0.1202,
503
  "step": 42
504
  },
505
  {
506
+ "epoch": 28.0,
507
  "eval_accuracy": 1.0,
508
+ "eval_loss": 0.002967243082821369,
509
+ "eval_runtime": 0.1135,
510
+ "eval_samples_per_second": 149.812,
511
+ "eval_steps_per_second": 8.812,
512
  "step": 42
513
  },
514
  {
515
+ "epoch": 28.67,
516
  "learning_rate": 7.777777777777777e-06,
517
+ "loss": 0.1368,
518
  "step": 43
519
  },
520
  {
521
+ "epoch": 28.67,
522
  "eval_accuracy": 1.0,
523
+ "eval_loss": 0.002836533123627305,
524
+ "eval_runtime": 0.1133,
525
+ "eval_samples_per_second": 150.077,
526
+ "eval_steps_per_second": 8.828,
527
  "step": 43
528
  },
529
  {
530
+ "epoch": 29.33,
531
  "learning_rate": 6.666666666666667e-06,
532
+ "loss": 0.403,
533
  "step": 44
534
  },
535
  {
536
+ "epoch": 30.0,
537
  "learning_rate": 5.555555555555556e-06,
538
+ "loss": 0.1016,
539
  "step": 45
540
  },
541
  {
542
+ "epoch": 30.0,
543
  "eval_accuracy": 1.0,
544
+ "eval_loss": 0.0027249290142208338,
545
+ "eval_runtime": 0.1143,
546
+ "eval_samples_per_second": 148.728,
547
+ "eval_steps_per_second": 8.749,
548
  "step": 45
549
  },
550
  {
551
+ "epoch": 30.67,
552
  "learning_rate": 4.444444444444445e-06,
553
+ "loss": 0.1282,
554
  "step": 46
555
  },
556
  {
557
+ "epoch": 30.67,
558
  "eval_accuracy": 1.0,
559
+ "eval_loss": 0.0026905094273388386,
560
+ "eval_runtime": 0.1127,
561
+ "eval_samples_per_second": 150.9,
562
+ "eval_steps_per_second": 8.876,
563
  "step": 46
564
  },
565
  {
566
+ "epoch": 31.33,
567
  "learning_rate": 3.3333333333333333e-06,
568
+ "loss": 0.0778,
 
 
 
 
 
 
 
 
 
569
  "step": 47
570
  },
571
  {
572
+ "epoch": 32.0,
573
  "learning_rate": 2.2222222222222225e-06,
574
+ "loss": 0.1128,
575
  "step": 48
576
  },
577
  {
578
+ "epoch": 32.0,
579
  "eval_accuracy": 1.0,
580
+ "eval_loss": 0.0026230851653963327,
581
+ "eval_runtime": 0.1139,
582
+ "eval_samples_per_second": 149.284,
583
+ "eval_steps_per_second": 8.781,
584
  "step": 48
585
  },
586
  {
587
+ "epoch": 32.67,
588
  "learning_rate": 1.1111111111111112e-06,
589
+ "loss": 0.2366,
590
+ "step": 49
591
+ },
592
+ {
593
+ "epoch": 32.67,
594
+ "eval_accuracy": 1.0,
595
+ "eval_loss": 0.0026192485820502043,
596
+ "eval_runtime": 0.1122,
597
+ "eval_samples_per_second": 151.562,
598
+ "eval_steps_per_second": 8.915,
599
  "step": 49
600
  },
601
  {
602
+ "epoch": 33.33,
603
  "learning_rate": 0.0,
604
+ "loss": 0.1727,
605
  "step": 50
606
  },
607
  {
608
+ "epoch": 33.33,
609
  "eval_accuracy": 1.0,
610
+ "eval_loss": 0.002621703315526247,
611
+ "eval_runtime": 0.1147,
612
+ "eval_samples_per_second": 148.149,
613
+ "eval_steps_per_second": 8.715,
614
  "step": 50
615
  },
616
  {
617
+ "epoch": 33.33,
618
  "step": 50,
619
+ "total_flos": 3.9079210362638746e+17,
620
+ "train_loss": 0.26825646698474886,
621
+ "train_runtime": 260.0145,
622
+ "train_samples_per_second": 29.037,
623
+ "train_steps_per_second": 0.192
624
  }
625
  ],
626
  "max_steps": 50,
627
  "num_train_epochs": 50,
628
+ "total_flos": 3.9079210362638746e+17,
629
  "trial_name": null,
630
  "trial_params": null
631
  }