Shawon16 commited on
Commit
479aa60
·
verified ·
1 Parent(s): dcc015b

End of training

Browse files
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0526
22
- - Accuracy: 0.9875
23
 
24
  ## Model description
25
 
 
18
 
19
  This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.3845
22
+ - Accuracy: 0.9287
23
 
24
  ## Model description
25
 
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.099501661129569,
3
+ "eval_accuracy": 0.9286833855799373,
4
+ "eval_loss": 0.38447505235671997,
5
+ "eval_runtime": 537.4022,
6
+ "eval_samples_per_second": 2.374,
7
+ "eval_steps_per_second": 1.187
8
+ }
confusion_matrix_kfold_fold_5.png ADDED
confusion_matrix_test_fold_5.png ADDED
confusion_matrix_trainfold_fold_5.png ADDED
test_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.099501661129569,
3
+ "eval_accuracy": 0.9286833855799373,
4
+ "eval_loss": 0.38447505235671997,
5
+ "eval_runtime": 537.4022,
6
+ "eval_samples_per_second": 2.374,
7
+ "eval_steps_per_second": 1.187
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,771 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.9925280199252802,
3
+ "best_model_checkpoint": "/media/cse/HDD/Shawon/shawon/10 fold timesformer/VideoMAE_default_fold__4__10_epoch_Aug_batch_1_4_BdSLW60/checkpoint-8131",
4
+ "epoch": 9.099501661129569,
5
+ "eval_steps": 500,
6
+ "global_step": 9030,
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.01107419712070875,
13
+ "grad_norm": 35.217525482177734,
14
+ "learning_rate": 5.537098560354375e-06,
15
+ "loss": 16.5715,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.0221483942414175,
20
+ "grad_norm": 31.530065536499023,
21
+ "learning_rate": 1.107419712070875e-05,
22
+ "loss": 16.4548,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.03322259136212625,
27
+ "grad_norm": 32.5143928527832,
28
+ "learning_rate": 1.6611295681063124e-05,
29
+ "loss": 16.3944,
30
+ "step": 300
31
+ },
32
+ {
33
+ "epoch": 0.044296788482835,
34
+ "grad_norm": 27.03118896484375,
35
+ "learning_rate": 2.21483942414175e-05,
36
+ "loss": 16.2123,
37
+ "step": 400
38
+ },
39
+ {
40
+ "epoch": 0.05537098560354374,
41
+ "grad_norm": 34.12999725341797,
42
+ "learning_rate": 2.7685492801771873e-05,
43
+ "loss": 15.2606,
44
+ "step": 500
45
+ },
46
+ {
47
+ "epoch": 0.0664451827242525,
48
+ "grad_norm": 42.32147216796875,
49
+ "learning_rate": 3.322259136212625e-05,
50
+ "loss": 13.53,
51
+ "step": 600
52
+ },
53
+ {
54
+ "epoch": 0.07751937984496124,
55
+ "grad_norm": 44.13176727294922,
56
+ "learning_rate": 3.875968992248062e-05,
57
+ "loss": 11.9277,
58
+ "step": 700
59
+ },
60
+ {
61
+ "epoch": 0.08859357696567,
62
+ "grad_norm": 45.3419189453125,
63
+ "learning_rate": 4.4296788482835e-05,
64
+ "loss": 9.4988,
65
+ "step": 800
66
+ },
67
+ {
68
+ "epoch": 0.09966777408637874,
69
+ "grad_norm": 55.63376235961914,
70
+ "learning_rate": 4.983388704318937e-05,
71
+ "loss": 7.1534,
72
+ "step": 900
73
+ },
74
+ {
75
+ "epoch": 0.1,
76
+ "eval_accuracy": 0.5990037359900373,
77
+ "eval_loss": 1.634743094444275,
78
+ "eval_runtime": 254.712,
79
+ "eval_samples_per_second": 3.153,
80
+ "eval_steps_per_second": 1.578,
81
+ "step": 903
82
+ },
83
+ {
84
+ "epoch": 1.010686600221484,
85
+ "grad_norm": 37.874595642089844,
86
+ "learning_rate": 4.940322382182848e-05,
87
+ "loss": 5.7449,
88
+ "step": 1000
89
+ },
90
+ {
91
+ "epoch": 1.0217607973421927,
92
+ "grad_norm": 30.460966110229492,
93
+ "learning_rate": 4.878799064845577e-05,
94
+ "loss": 4.1019,
95
+ "step": 1100
96
+ },
97
+ {
98
+ "epoch": 1.0328349944629014,
99
+ "grad_norm": 44.03368377685547,
100
+ "learning_rate": 4.8172757475083056e-05,
101
+ "loss": 3.1676,
102
+ "step": 1200
103
+ },
104
+ {
105
+ "epoch": 1.0439091915836103,
106
+ "grad_norm": 44.42673873901367,
107
+ "learning_rate": 4.755752430171035e-05,
108
+ "loss": 2.8025,
109
+ "step": 1300
110
+ },
111
+ {
112
+ "epoch": 1.054983388704319,
113
+ "grad_norm": 72.93189239501953,
114
+ "learning_rate": 4.694229112833764e-05,
115
+ "loss": 2.4045,
116
+ "step": 1400
117
+ },
118
+ {
119
+ "epoch": 1.0660575858250276,
120
+ "grad_norm": 64.83969116210938,
121
+ "learning_rate": 4.6327057954964936e-05,
122
+ "loss": 1.8242,
123
+ "step": 1500
124
+ },
125
+ {
126
+ "epoch": 1.0771317829457365,
127
+ "grad_norm": 10.798648834228516,
128
+ "learning_rate": 4.571182478159223e-05,
129
+ "loss": 1.5073,
130
+ "step": 1600
131
+ },
132
+ {
133
+ "epoch": 1.0882059800664452,
134
+ "grad_norm": 68.70486450195312,
135
+ "learning_rate": 4.5096591608219516e-05,
136
+ "loss": 1.2643,
137
+ "step": 1700
138
+ },
139
+ {
140
+ "epoch": 1.0992801771871539,
141
+ "grad_norm": 39.017173767089844,
142
+ "learning_rate": 4.448135843484681e-05,
143
+ "loss": 1.2018,
144
+ "step": 1800
145
+ },
146
+ {
147
+ "epoch": 1.1000553709856036,
148
+ "eval_accuracy": 0.9028642590286425,
149
+ "eval_loss": 0.33687880635261536,
150
+ "eval_runtime": 246.5833,
151
+ "eval_samples_per_second": 3.257,
152
+ "eval_steps_per_second": 1.63,
153
+ "step": 1807
154
+ },
155
+ {
156
+ "epoch": 2.010299003322259,
157
+ "grad_norm": 2.68186354637146,
158
+ "learning_rate": 4.38661252614741e-05,
159
+ "loss": 0.9702,
160
+ "step": 1900
161
+ },
162
+ {
163
+ "epoch": 2.021373200442968,
164
+ "grad_norm": 0.49348923563957214,
165
+ "learning_rate": 4.325089208810139e-05,
166
+ "loss": 0.6578,
167
+ "step": 2000
168
+ },
169
+ {
170
+ "epoch": 2.0324473975636765,
171
+ "grad_norm": 25.38459014892578,
172
+ "learning_rate": 4.263565891472868e-05,
173
+ "loss": 0.8361,
174
+ "step": 2100
175
+ },
176
+ {
177
+ "epoch": 2.0435215946843854,
178
+ "grad_norm": 25.555809020996094,
179
+ "learning_rate": 4.2020425741355975e-05,
180
+ "loss": 0.6547,
181
+ "step": 2200
182
+ },
183
+ {
184
+ "epoch": 2.0545957918050943,
185
+ "grad_norm": 1.4637202024459839,
186
+ "learning_rate": 4.140519256798327e-05,
187
+ "loss": 0.623,
188
+ "step": 2300
189
+ },
190
+ {
191
+ "epoch": 2.065669988925803,
192
+ "grad_norm": 0.4001486003398895,
193
+ "learning_rate": 4.078995939461056e-05,
194
+ "loss": 0.524,
195
+ "step": 2400
196
+ },
197
+ {
198
+ "epoch": 2.0767441860465117,
199
+ "grad_norm": 10.103757858276367,
200
+ "learning_rate": 4.0174726221237855e-05,
201
+ "loss": 0.4509,
202
+ "step": 2500
203
+ },
204
+ {
205
+ "epoch": 2.0878183831672206,
206
+ "grad_norm": 33.20508575439453,
207
+ "learning_rate": 3.955949304786514e-05,
208
+ "loss": 0.5262,
209
+ "step": 2600
210
+ },
211
+ {
212
+ "epoch": 2.098892580287929,
213
+ "grad_norm": 6.72857666015625,
214
+ "learning_rate": 3.8944259874492434e-05,
215
+ "loss": 0.3895,
216
+ "step": 2700
217
+ },
218
+ {
219
+ "epoch": 2.1,
220
+ "eval_accuracy": 0.958904109589041,
221
+ "eval_loss": 0.15578480064868927,
222
+ "eval_runtime": 244.3662,
223
+ "eval_samples_per_second": 3.286,
224
+ "eval_steps_per_second": 1.645,
225
+ "step": 2710
226
+ },
227
+ {
228
+ "epoch": 3.009911406423034,
229
+ "grad_norm": 0.40964722633361816,
230
+ "learning_rate": 3.832902670111973e-05,
231
+ "loss": 0.2033,
232
+ "step": 2800
233
+ },
234
+ {
235
+ "epoch": 3.020985603543743,
236
+ "grad_norm": 0.33143919706344604,
237
+ "learning_rate": 3.7713793527747014e-05,
238
+ "loss": 0.2356,
239
+ "step": 2900
240
+ },
241
+ {
242
+ "epoch": 3.032059800664452,
243
+ "grad_norm": 1.2071921825408936,
244
+ "learning_rate": 3.7098560354374314e-05,
245
+ "loss": 0.3155,
246
+ "step": 3000
247
+ },
248
+ {
249
+ "epoch": 3.0431339977851604,
250
+ "grad_norm": 0.3948413133621216,
251
+ "learning_rate": 3.64833271810016e-05,
252
+ "loss": 0.168,
253
+ "step": 3100
254
+ },
255
+ {
256
+ "epoch": 3.0542081949058693,
257
+ "grad_norm": 0.20336775481700897,
258
+ "learning_rate": 3.5868094007628894e-05,
259
+ "loss": 0.2734,
260
+ "step": 3200
261
+ },
262
+ {
263
+ "epoch": 3.065282392026578,
264
+ "grad_norm": 9.776823997497559,
265
+ "learning_rate": 3.525286083425619e-05,
266
+ "loss": 0.2658,
267
+ "step": 3300
268
+ },
269
+ {
270
+ "epoch": 3.076356589147287,
271
+ "grad_norm": 9.96722412109375,
272
+ "learning_rate": 3.463762766088347e-05,
273
+ "loss": 0.2547,
274
+ "step": 3400
275
+ },
276
+ {
277
+ "epoch": 3.0874307862679955,
278
+ "grad_norm": 0.9340003132820129,
279
+ "learning_rate": 3.4022394487510767e-05,
280
+ "loss": 0.1343,
281
+ "step": 3500
282
+ },
283
+ {
284
+ "epoch": 3.0985049833887044,
285
+ "grad_norm": 0.0304119773209095,
286
+ "learning_rate": 3.340716131413806e-05,
287
+ "loss": 0.1654,
288
+ "step": 3600
289
+ },
290
+ {
291
+ "epoch": 3.1000553709856034,
292
+ "eval_accuracy": 0.9788293897882939,
293
+ "eval_loss": 0.09220333397388458,
294
+ "eval_runtime": 247.2233,
295
+ "eval_samples_per_second": 3.248,
296
+ "eval_steps_per_second": 1.626,
297
+ "step": 3614
298
+ },
299
+ {
300
+ "epoch": 4.0095238095238095,
301
+ "grad_norm": 0.05322951823472977,
302
+ "learning_rate": 3.2791928140765346e-05,
303
+ "loss": 0.0955,
304
+ "step": 3700
305
+ },
306
+ {
307
+ "epoch": 4.020598006644518,
308
+ "grad_norm": 0.14431843161582947,
309
+ "learning_rate": 3.2176694967392646e-05,
310
+ "loss": 0.181,
311
+ "step": 3800
312
+ },
313
+ {
314
+ "epoch": 4.031672203765227,
315
+ "grad_norm": 0.14612068235874176,
316
+ "learning_rate": 3.156146179401994e-05,
317
+ "loss": 0.2801,
318
+ "step": 3900
319
+ },
320
+ {
321
+ "epoch": 4.042746400885936,
322
+ "grad_norm": 0.030130434781312943,
323
+ "learning_rate": 3.0946228620647226e-05,
324
+ "loss": 0.11,
325
+ "step": 4000
326
+ },
327
+ {
328
+ "epoch": 4.053820598006644,
329
+ "grad_norm": 84.40876770019531,
330
+ "learning_rate": 3.033099544727452e-05,
331
+ "loss": 0.1632,
332
+ "step": 4100
333
+ },
334
+ {
335
+ "epoch": 4.064894795127353,
336
+ "grad_norm": 247.53504943847656,
337
+ "learning_rate": 2.971576227390181e-05,
338
+ "loss": 0.1784,
339
+ "step": 4200
340
+ },
341
+ {
342
+ "epoch": 4.075968992248062,
343
+ "grad_norm": 0.022293824702501297,
344
+ "learning_rate": 2.91005291005291e-05,
345
+ "loss": 0.1768,
346
+ "step": 4300
347
+ },
348
+ {
349
+ "epoch": 4.087043189368771,
350
+ "grad_norm": 0.04350820183753967,
351
+ "learning_rate": 2.8485295927156392e-05,
352
+ "loss": 0.2005,
353
+ "step": 4400
354
+ },
355
+ {
356
+ "epoch": 4.09811738648948,
357
+ "grad_norm": 0.03336182236671448,
358
+ "learning_rate": 2.787006275378369e-05,
359
+ "loss": 0.0988,
360
+ "step": 4500
361
+ },
362
+ {
363
+ "epoch": 4.1,
364
+ "eval_accuracy": 0.9813200498132005,
365
+ "eval_loss": 0.07138894498348236,
366
+ "eval_runtime": 242.9311,
367
+ "eval_samples_per_second": 3.305,
368
+ "eval_steps_per_second": 1.655,
369
+ "step": 4517
370
+ },
371
+ {
372
+ "epoch": 5.009136212624584,
373
+ "grad_norm": 0.015874436125159264,
374
+ "learning_rate": 2.725482958041098e-05,
375
+ "loss": 0.066,
376
+ "step": 4600
377
+ },
378
+ {
379
+ "epoch": 5.020210409745293,
380
+ "grad_norm": 0.04679744318127632,
381
+ "learning_rate": 2.6639596407038268e-05,
382
+ "loss": 0.1048,
383
+ "step": 4700
384
+ },
385
+ {
386
+ "epoch": 5.031284606866002,
387
+ "grad_norm": 46.831634521484375,
388
+ "learning_rate": 2.602436323366556e-05,
389
+ "loss": 0.1694,
390
+ "step": 4800
391
+ },
392
+ {
393
+ "epoch": 5.042358803986711,
394
+ "grad_norm": 0.02305000275373459,
395
+ "learning_rate": 2.540913006029285e-05,
396
+ "loss": 0.0276,
397
+ "step": 4900
398
+ },
399
+ {
400
+ "epoch": 5.05343300110742,
401
+ "grad_norm": 0.024845095351338387,
402
+ "learning_rate": 2.479389688692014e-05,
403
+ "loss": 0.0752,
404
+ "step": 5000
405
+ },
406
+ {
407
+ "epoch": 5.064507198228129,
408
+ "grad_norm": 0.07386521995067596,
409
+ "learning_rate": 2.4178663713547438e-05,
410
+ "loss": 0.1106,
411
+ "step": 5100
412
+ },
413
+ {
414
+ "epoch": 5.075581395348837,
415
+ "grad_norm": 0.011052440851926804,
416
+ "learning_rate": 2.3563430540174727e-05,
417
+ "loss": 0.1798,
418
+ "step": 5200
419
+ },
420
+ {
421
+ "epoch": 5.086655592469546,
422
+ "grad_norm": 0.03108247183263302,
423
+ "learning_rate": 2.2948197366802017e-05,
424
+ "loss": 0.0989,
425
+ "step": 5300
426
+ },
427
+ {
428
+ "epoch": 5.097729789590255,
429
+ "grad_norm": 0.026010781526565552,
430
+ "learning_rate": 2.233296419342931e-05,
431
+ "loss": 0.1344,
432
+ "step": 5400
433
+ },
434
+ {
435
+ "epoch": 5.100055370985603,
436
+ "eval_accuracy": 0.9900373599003736,
437
+ "eval_loss": 0.05479121580719948,
438
+ "eval_runtime": 245.2178,
439
+ "eval_samples_per_second": 3.275,
440
+ "eval_steps_per_second": 1.639,
441
+ "step": 5421
442
+ },
443
+ {
444
+ "epoch": 6.00874861572536,
445
+ "grad_norm": 0.017900297418236732,
446
+ "learning_rate": 2.1717731020056604e-05,
447
+ "loss": 0.056,
448
+ "step": 5500
449
+ },
450
+ {
451
+ "epoch": 6.019822812846068,
452
+ "grad_norm": 0.014143782667815685,
453
+ "learning_rate": 2.1102497846683894e-05,
454
+ "loss": 0.0514,
455
+ "step": 5600
456
+ },
457
+ {
458
+ "epoch": 6.030897009966777,
459
+ "grad_norm": 1.0569688081741333,
460
+ "learning_rate": 2.0487264673311183e-05,
461
+ "loss": 0.0268,
462
+ "step": 5700
463
+ },
464
+ {
465
+ "epoch": 6.041971207087486,
466
+ "grad_norm": 0.01698586530983448,
467
+ "learning_rate": 1.987203149993848e-05,
468
+ "loss": 0.1061,
469
+ "step": 5800
470
+ },
471
+ {
472
+ "epoch": 6.053045404208195,
473
+ "grad_norm": 0.00996414851397276,
474
+ "learning_rate": 1.925679832656577e-05,
475
+ "loss": 0.011,
476
+ "step": 5900
477
+ },
478
+ {
479
+ "epoch": 6.064119601328904,
480
+ "grad_norm": 0.038351599127054214,
481
+ "learning_rate": 1.864156515319306e-05,
482
+ "loss": 0.0827,
483
+ "step": 6000
484
+ },
485
+ {
486
+ "epoch": 6.075193798449613,
487
+ "grad_norm": 0.08223330229520798,
488
+ "learning_rate": 1.8026331979820353e-05,
489
+ "loss": 0.0402,
490
+ "step": 6100
491
+ },
492
+ {
493
+ "epoch": 6.086267995570321,
494
+ "grad_norm": 0.011590097099542618,
495
+ "learning_rate": 1.7411098806447646e-05,
496
+ "loss": 0.0894,
497
+ "step": 6200
498
+ },
499
+ {
500
+ "epoch": 6.09734219269103,
501
+ "grad_norm": 0.01950395107269287,
502
+ "learning_rate": 1.6795865633074936e-05,
503
+ "loss": 0.1058,
504
+ "step": 6300
505
+ },
506
+ {
507
+ "epoch": 6.1,
508
+ "eval_accuracy": 0.987546699875467,
509
+ "eval_loss": 0.055294327437877655,
510
+ "eval_runtime": 242.6622,
511
+ "eval_samples_per_second": 3.309,
512
+ "eval_steps_per_second": 1.657,
513
+ "step": 6324
514
+ },
515
+ {
516
+ "epoch": 7.008361018826135,
517
+ "grad_norm": 0.0069948905147612095,
518
+ "learning_rate": 1.6180632459702226e-05,
519
+ "loss": 0.0577,
520
+ "step": 6400
521
+ },
522
+ {
523
+ "epoch": 7.019435215946844,
524
+ "grad_norm": 1.0322211980819702,
525
+ "learning_rate": 1.556539928632952e-05,
526
+ "loss": 0.0197,
527
+ "step": 6500
528
+ },
529
+ {
530
+ "epoch": 7.030509413067553,
531
+ "grad_norm": 0.008567863143980503,
532
+ "learning_rate": 1.4950166112956812e-05,
533
+ "loss": 0.1023,
534
+ "step": 6600
535
+ },
536
+ {
537
+ "epoch": 7.041583610188261,
538
+ "grad_norm": 0.026618080213665962,
539
+ "learning_rate": 1.4334932939584104e-05,
540
+ "loss": 0.0176,
541
+ "step": 6700
542
+ },
543
+ {
544
+ "epoch": 7.05265780730897,
545
+ "grad_norm": 0.12838630378246307,
546
+ "learning_rate": 1.3719699766211393e-05,
547
+ "loss": 0.0046,
548
+ "step": 6800
549
+ },
550
+ {
551
+ "epoch": 7.063732004429679,
552
+ "grad_norm": 0.0924014300107956,
553
+ "learning_rate": 1.3104466592838688e-05,
554
+ "loss": 0.0248,
555
+ "step": 6900
556
+ },
557
+ {
558
+ "epoch": 7.074806201550388,
559
+ "grad_norm": 0.005219871178269386,
560
+ "learning_rate": 1.2489233419465978e-05,
561
+ "loss": 0.0611,
562
+ "step": 7000
563
+ },
564
+ {
565
+ "epoch": 7.0858803986710965,
566
+ "grad_norm": 0.013417864218354225,
567
+ "learning_rate": 1.187400024609327e-05,
568
+ "loss": 0.0089,
569
+ "step": 7100
570
+ },
571
+ {
572
+ "epoch": 7.096954595791805,
573
+ "grad_norm": 0.005858865100890398,
574
+ "learning_rate": 1.1258767072720563e-05,
575
+ "loss": 0.0246,
576
+ "step": 7200
577
+ },
578
+ {
579
+ "epoch": 7.100055370985603,
580
+ "eval_accuracy": 0.9912826899128269,
581
+ "eval_loss": 0.0487552247941494,
582
+ "eval_runtime": 351.5655,
583
+ "eval_samples_per_second": 2.284,
584
+ "eval_steps_per_second": 1.143,
585
+ "step": 7228
586
+ },
587
+ {
588
+ "epoch": 8.007973421926911,
589
+ "grad_norm": 0.0037075302097946405,
590
+ "learning_rate": 1.0643533899347853e-05,
591
+ "loss": 0.0388,
592
+ "step": 7300
593
+ },
594
+ {
595
+ "epoch": 8.019047619047619,
596
+ "grad_norm": 0.004365282598882914,
597
+ "learning_rate": 1.0028300725975146e-05,
598
+ "loss": 0.0097,
599
+ "step": 7400
600
+ },
601
+ {
602
+ "epoch": 8.030121816168327,
603
+ "grad_norm": 0.006515075918287039,
604
+ "learning_rate": 9.413067552602436e-06,
605
+ "loss": 0.0278,
606
+ "step": 7500
607
+ },
608
+ {
609
+ "epoch": 8.041196013289037,
610
+ "grad_norm": 0.004097383469343185,
611
+ "learning_rate": 8.797834379229729e-06,
612
+ "loss": 0.0281,
613
+ "step": 7600
614
+ },
615
+ {
616
+ "epoch": 8.052270210409745,
617
+ "grad_norm": 0.031094931066036224,
618
+ "learning_rate": 8.18260120585702e-06,
619
+ "loss": 0.0221,
620
+ "step": 7700
621
+ },
622
+ {
623
+ "epoch": 8.063344407530455,
624
+ "grad_norm": 0.005156899336725473,
625
+ "learning_rate": 7.567368032484312e-06,
626
+ "loss": 0.0014,
627
+ "step": 7800
628
+ },
629
+ {
630
+ "epoch": 8.074418604651163,
631
+ "grad_norm": 0.007017307449132204,
632
+ "learning_rate": 6.952134859111603e-06,
633
+ "loss": 0.0049,
634
+ "step": 7900
635
+ },
636
+ {
637
+ "epoch": 8.085492801771872,
638
+ "grad_norm": 0.023146001622080803,
639
+ "learning_rate": 6.336901685738895e-06,
640
+ "loss": 0.029,
641
+ "step": 8000
642
+ },
643
+ {
644
+ "epoch": 8.09656699889258,
645
+ "grad_norm": 0.007268809247761965,
646
+ "learning_rate": 5.7216685123661875e-06,
647
+ "loss": 0.0006,
648
+ "step": 8100
649
+ },
650
+ {
651
+ "epoch": 8.1,
652
+ "eval_accuracy": 0.9925280199252802,
653
+ "eval_loss": 0.03840411454439163,
654
+ "eval_runtime": 359.8229,
655
+ "eval_samples_per_second": 2.232,
656
+ "eval_steps_per_second": 1.117,
657
+ "step": 8131
658
+ },
659
+ {
660
+ "epoch": 9.007585825027686,
661
+ "grad_norm": 2.1333770751953125,
662
+ "learning_rate": 5.106435338993479e-06,
663
+ "loss": 0.0007,
664
+ "step": 8200
665
+ },
666
+ {
667
+ "epoch": 9.018660022148394,
668
+ "grad_norm": 0.0032328683882951736,
669
+ "learning_rate": 4.4912021656207705e-06,
670
+ "loss": 0.0127,
671
+ "step": 8300
672
+ },
673
+ {
674
+ "epoch": 9.029734219269104,
675
+ "grad_norm": 0.12912631034851074,
676
+ "learning_rate": 3.875968992248062e-06,
677
+ "loss": 0.0084,
678
+ "step": 8400
679
+ },
680
+ {
681
+ "epoch": 9.040808416389812,
682
+ "grad_norm": 0.004084580112248659,
683
+ "learning_rate": 3.260735818875354e-06,
684
+ "loss": 0.0005,
685
+ "step": 8500
686
+ },
687
+ {
688
+ "epoch": 9.05188261351052,
689
+ "grad_norm": 0.0032446524128317833,
690
+ "learning_rate": 2.6455026455026455e-06,
691
+ "loss": 0.025,
692
+ "step": 8600
693
+ },
694
+ {
695
+ "epoch": 9.06295681063123,
696
+ "grad_norm": 0.010469436645507812,
697
+ "learning_rate": 2.0302694721299375e-06,
698
+ "loss": 0.0007,
699
+ "step": 8700
700
+ },
701
+ {
702
+ "epoch": 9.074031007751937,
703
+ "grad_norm": 0.004183362703770399,
704
+ "learning_rate": 1.415036298757229e-06,
705
+ "loss": 0.0005,
706
+ "step": 8800
707
+ },
708
+ {
709
+ "epoch": 9.085105204872647,
710
+ "grad_norm": 0.003930332604795694,
711
+ "learning_rate": 7.998031253845208e-07,
712
+ "loss": 0.0005,
713
+ "step": 8900
714
+ },
715
+ {
716
+ "epoch": 9.096179401993355,
717
+ "grad_norm": 0.004005803726613522,
718
+ "learning_rate": 1.8456995201181249e-07,
719
+ "loss": 0.0005,
720
+ "step": 9000
721
+ },
722
+ {
723
+ "epoch": 9.099501661129569,
724
+ "eval_accuracy": 0.987546699875467,
725
+ "eval_loss": 0.052565645426511765,
726
+ "eval_runtime": 362.3648,
727
+ "eval_samples_per_second": 2.216,
728
+ "eval_steps_per_second": 1.109,
729
+ "step": 9030
730
+ },
731
+ {
732
+ "epoch": 9.099501661129569,
733
+ "step": 9030,
734
+ "total_flos": 9.006226676074414e+19,
735
+ "train_loss": 1.7522039692807667,
736
+ "train_runtime": 33181.045,
737
+ "train_samples_per_second": 2.177,
738
+ "train_steps_per_second": 0.272
739
+ },
740
+ {
741
+ "epoch": 9.099501661129569,
742
+ "eval_accuracy": 0.9286833855799373,
743
+ "eval_loss": 0.38447505235671997,
744
+ "eval_runtime": 537.4022,
745
+ "eval_samples_per_second": 2.374,
746
+ "eval_steps_per_second": 1.187,
747
+ "step": 9030
748
+ }
749
+ ],
750
+ "logging_steps": 100,
751
+ "max_steps": 9030,
752
+ "num_input_tokens_seen": 0,
753
+ "num_train_epochs": 9223372036854775807,
754
+ "save_steps": 500,
755
+ "stateful_callbacks": {
756
+ "TrainerControl": {
757
+ "args": {
758
+ "should_epoch_stop": false,
759
+ "should_evaluate": false,
760
+ "should_log": false,
761
+ "should_save": true,
762
+ "should_training_stop": true
763
+ },
764
+ "attributes": {}
765
+ }
766
+ },
767
+ "total_flos": 9.006226676074414e+19,
768
+ "train_batch_size": 2,
769
+ "trial_name": null,
770
+ "trial_params": null
771
+ }