Shawon16 commited on
Commit
a01e498
·
verified ·
1 Parent(s): 2773d5b

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.0218
22
- - Accuracy: 0.9950
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.5166
22
+ - Accuracy: 0.8934
23
 
24
  ## Model description
25
 
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.099501661129569,
3
+ "eval_accuracy": 0.8934169278996865,
4
+ "eval_loss": 0.5165748596191406,
5
+ "eval_runtime": 353.997,
6
+ "eval_samples_per_second": 3.605,
7
+ "eval_steps_per_second": 1.802
8
+ }
confusion_matrix_kfold_fold_9.png ADDED
confusion_matrix_test_fold_9.png ADDED
confusion_matrix_trainfold_fold_9.png ADDED
test_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.099501661129569,
3
+ "eval_accuracy": 0.8934169278996865,
4
+ "eval_loss": 0.5165748596191406,
5
+ "eval_runtime": 353.997,
6
+ "eval_samples_per_second": 3.605,
7
+ "eval_steps_per_second": 1.802
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,771 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.9950186799501868,
3
+ "best_model_checkpoint": "/media/cse/HDD/Shawon/shawon/10 fold timesformer/VideoMAE_default_fold__8__10_epoch_Aug_batch_1_4_BdSLW60/checkpoint-9030",
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": 32.334835052490234,
14
+ "learning_rate": 5.537098560354375e-06,
15
+ "loss": 16.5679,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.0221483942414175,
20
+ "grad_norm": 32.082523345947266,
21
+ "learning_rate": 1.107419712070875e-05,
22
+ "loss": 16.4682,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.03322259136212625,
27
+ "grad_norm": 29.525415420532227,
28
+ "learning_rate": 1.6611295681063124e-05,
29
+ "loss": 16.414,
30
+ "step": 300
31
+ },
32
+ {
33
+ "epoch": 0.044296788482835,
34
+ "grad_norm": 27.141206741333008,
35
+ "learning_rate": 2.21483942414175e-05,
36
+ "loss": 16.2142,
37
+ "step": 400
38
+ },
39
+ {
40
+ "epoch": 0.05537098560354374,
41
+ "grad_norm": 33.727516174316406,
42
+ "learning_rate": 2.7685492801771873e-05,
43
+ "loss": 14.8236,
44
+ "step": 500
45
+ },
46
+ {
47
+ "epoch": 0.0664451827242525,
48
+ "grad_norm": 48.193450927734375,
49
+ "learning_rate": 3.322259136212625e-05,
50
+ "loss": 12.3716,
51
+ "step": 600
52
+ },
53
+ {
54
+ "epoch": 0.07751937984496124,
55
+ "grad_norm": 44.83864974975586,
56
+ "learning_rate": 3.875968992248062e-05,
57
+ "loss": 10.2729,
58
+ "step": 700
59
+ },
60
+ {
61
+ "epoch": 0.08859357696567,
62
+ "grad_norm": 50.79475021362305,
63
+ "learning_rate": 4.4296788482835e-05,
64
+ "loss": 8.5405,
65
+ "step": 800
66
+ },
67
+ {
68
+ "epoch": 0.09966777408637874,
69
+ "grad_norm": 42.04021072387695,
70
+ "learning_rate": 4.983388704318937e-05,
71
+ "loss": 6.2832,
72
+ "step": 900
73
+ },
74
+ {
75
+ "epoch": 0.1,
76
+ "eval_accuracy": 0.663760896637609,
77
+ "eval_loss": 1.3872833251953125,
78
+ "eval_runtime": 243.25,
79
+ "eval_samples_per_second": 3.301,
80
+ "eval_steps_per_second": 1.653,
81
+ "step": 903
82
+ },
83
+ {
84
+ "epoch": 1.010686600221484,
85
+ "grad_norm": 41.05415344238281,
86
+ "learning_rate": 4.940322382182848e-05,
87
+ "loss": 5.3027,
88
+ "step": 1000
89
+ },
90
+ {
91
+ "epoch": 1.0217607973421927,
92
+ "grad_norm": 43.8326416015625,
93
+ "learning_rate": 4.878799064845577e-05,
94
+ "loss": 3.7644,
95
+ "step": 1100
96
+ },
97
+ {
98
+ "epoch": 1.0328349944629014,
99
+ "grad_norm": 58.01288604736328,
100
+ "learning_rate": 4.8172757475083056e-05,
101
+ "loss": 3.042,
102
+ "step": 1200
103
+ },
104
+ {
105
+ "epoch": 1.0439091915836103,
106
+ "grad_norm": 70.05992126464844,
107
+ "learning_rate": 4.755752430171035e-05,
108
+ "loss": 2.6035,
109
+ "step": 1300
110
+ },
111
+ {
112
+ "epoch": 1.054983388704319,
113
+ "grad_norm": 77.99320983886719,
114
+ "learning_rate": 4.694229112833764e-05,
115
+ "loss": 2.1544,
116
+ "step": 1400
117
+ },
118
+ {
119
+ "epoch": 1.0660575858250276,
120
+ "grad_norm": 82.49065399169922,
121
+ "learning_rate": 4.6327057954964936e-05,
122
+ "loss": 1.8055,
123
+ "step": 1500
124
+ },
125
+ {
126
+ "epoch": 1.0771317829457365,
127
+ "grad_norm": 7.742678165435791,
128
+ "learning_rate": 4.571182478159223e-05,
129
+ "loss": 1.5851,
130
+ "step": 1600
131
+ },
132
+ {
133
+ "epoch": 1.0882059800664452,
134
+ "grad_norm": 31.726713180541992,
135
+ "learning_rate": 4.5096591608219516e-05,
136
+ "loss": 1.2787,
137
+ "step": 1700
138
+ },
139
+ {
140
+ "epoch": 1.0992801771871539,
141
+ "grad_norm": 19.243579864501953,
142
+ "learning_rate": 4.448135843484681e-05,
143
+ "loss": 1.3796,
144
+ "step": 1800
145
+ },
146
+ {
147
+ "epoch": 1.1000553709856036,
148
+ "eval_accuracy": 0.9215442092154421,
149
+ "eval_loss": 0.31225696206092834,
150
+ "eval_runtime": 241.7159,
151
+ "eval_samples_per_second": 3.322,
152
+ "eval_steps_per_second": 1.663,
153
+ "step": 1807
154
+ },
155
+ {
156
+ "epoch": 2.010299003322259,
157
+ "grad_norm": 20.470407485961914,
158
+ "learning_rate": 4.38661252614741e-05,
159
+ "loss": 0.8224,
160
+ "step": 1900
161
+ },
162
+ {
163
+ "epoch": 2.021373200442968,
164
+ "grad_norm": 6.891818523406982,
165
+ "learning_rate": 4.325089208810139e-05,
166
+ "loss": 0.8476,
167
+ "step": 2000
168
+ },
169
+ {
170
+ "epoch": 2.0324473975636765,
171
+ "grad_norm": 62.1899528503418,
172
+ "learning_rate": 4.263565891472868e-05,
173
+ "loss": 0.88,
174
+ "step": 2100
175
+ },
176
+ {
177
+ "epoch": 2.0435215946843854,
178
+ "grad_norm": 31.87169647216797,
179
+ "learning_rate": 4.2020425741355975e-05,
180
+ "loss": 0.6384,
181
+ "step": 2200
182
+ },
183
+ {
184
+ "epoch": 2.0545957918050943,
185
+ "grad_norm": 5.144892692565918,
186
+ "learning_rate": 4.140519256798327e-05,
187
+ "loss": 0.6451,
188
+ "step": 2300
189
+ },
190
+ {
191
+ "epoch": 2.065669988925803,
192
+ "grad_norm": 61.46706008911133,
193
+ "learning_rate": 4.078995939461056e-05,
194
+ "loss": 0.4655,
195
+ "step": 2400
196
+ },
197
+ {
198
+ "epoch": 2.0767441860465117,
199
+ "grad_norm": 2.7628982067108154,
200
+ "learning_rate": 4.0174726221237855e-05,
201
+ "loss": 0.5972,
202
+ "step": 2500
203
+ },
204
+ {
205
+ "epoch": 2.0878183831672206,
206
+ "grad_norm": 16.27060317993164,
207
+ "learning_rate": 3.955949304786514e-05,
208
+ "loss": 0.5972,
209
+ "step": 2600
210
+ },
211
+ {
212
+ "epoch": 2.098892580287929,
213
+ "grad_norm": 7.138740062713623,
214
+ "learning_rate": 3.8944259874492434e-05,
215
+ "loss": 0.311,
216
+ "step": 2700
217
+ },
218
+ {
219
+ "epoch": 2.1,
220
+ "eval_accuracy": 0.9526774595267746,
221
+ "eval_loss": 0.16346044838428497,
222
+ "eval_runtime": 243.2026,
223
+ "eval_samples_per_second": 3.302,
224
+ "eval_steps_per_second": 1.653,
225
+ "step": 2710
226
+ },
227
+ {
228
+ "epoch": 3.009911406423034,
229
+ "grad_norm": 32.57714080810547,
230
+ "learning_rate": 3.832902670111973e-05,
231
+ "loss": 0.2697,
232
+ "step": 2800
233
+ },
234
+ {
235
+ "epoch": 3.020985603543743,
236
+ "grad_norm": 2.181366443634033,
237
+ "learning_rate": 3.7713793527747014e-05,
238
+ "loss": 0.3771,
239
+ "step": 2900
240
+ },
241
+ {
242
+ "epoch": 3.032059800664452,
243
+ "grad_norm": 0.17715582251548767,
244
+ "learning_rate": 3.7098560354374314e-05,
245
+ "loss": 0.364,
246
+ "step": 3000
247
+ },
248
+ {
249
+ "epoch": 3.0431339977851604,
250
+ "grad_norm": 74.82624053955078,
251
+ "learning_rate": 3.64833271810016e-05,
252
+ "loss": 0.2841,
253
+ "step": 3100
254
+ },
255
+ {
256
+ "epoch": 3.0542081949058693,
257
+ "grad_norm": 0.15879639983177185,
258
+ "learning_rate": 3.5868094007628894e-05,
259
+ "loss": 0.3189,
260
+ "step": 3200
261
+ },
262
+ {
263
+ "epoch": 3.065282392026578,
264
+ "grad_norm": 0.40148216485977173,
265
+ "learning_rate": 3.525286083425619e-05,
266
+ "loss": 0.2459,
267
+ "step": 3300
268
+ },
269
+ {
270
+ "epoch": 3.076356589147287,
271
+ "grad_norm": 4.050868034362793,
272
+ "learning_rate": 3.463762766088347e-05,
273
+ "loss": 0.2754,
274
+ "step": 3400
275
+ },
276
+ {
277
+ "epoch": 3.0874307862679955,
278
+ "grad_norm": 0.07379398494958878,
279
+ "learning_rate": 3.4022394487510767e-05,
280
+ "loss": 0.0931,
281
+ "step": 3500
282
+ },
283
+ {
284
+ "epoch": 3.0985049833887044,
285
+ "grad_norm": 1.5800195932388306,
286
+ "learning_rate": 3.340716131413806e-05,
287
+ "loss": 0.1077,
288
+ "step": 3600
289
+ },
290
+ {
291
+ "epoch": 3.1000553709856034,
292
+ "eval_accuracy": 0.9738480697384807,
293
+ "eval_loss": 0.10133728384971619,
294
+ "eval_runtime": 240.4329,
295
+ "eval_samples_per_second": 3.34,
296
+ "eval_steps_per_second": 1.672,
297
+ "step": 3614
298
+ },
299
+ {
300
+ "epoch": 4.0095238095238095,
301
+ "grad_norm": 0.2016027569770813,
302
+ "learning_rate": 3.2791928140765346e-05,
303
+ "loss": 0.0955,
304
+ "step": 3700
305
+ },
306
+ {
307
+ "epoch": 4.020598006644518,
308
+ "grad_norm": 0.13384506106376648,
309
+ "learning_rate": 3.2176694967392646e-05,
310
+ "loss": 0.2288,
311
+ "step": 3800
312
+ },
313
+ {
314
+ "epoch": 4.031672203765227,
315
+ "grad_norm": 0.3776722848415375,
316
+ "learning_rate": 3.156146179401994e-05,
317
+ "loss": 0.1254,
318
+ "step": 3900
319
+ },
320
+ {
321
+ "epoch": 4.042746400885936,
322
+ "grad_norm": 0.28055495023727417,
323
+ "learning_rate": 3.0946228620647226e-05,
324
+ "loss": 0.1256,
325
+ "step": 4000
326
+ },
327
+ {
328
+ "epoch": 4.053820598006644,
329
+ "grad_norm": 0.07850215584039688,
330
+ "learning_rate": 3.033099544727452e-05,
331
+ "loss": 0.0795,
332
+ "step": 4100
333
+ },
334
+ {
335
+ "epoch": 4.064894795127353,
336
+ "grad_norm": 170.31246948242188,
337
+ "learning_rate": 2.971576227390181e-05,
338
+ "loss": 0.0509,
339
+ "step": 4200
340
+ },
341
+ {
342
+ "epoch": 4.075968992248062,
343
+ "grad_norm": 1.1608444452285767,
344
+ "learning_rate": 2.91005291005291e-05,
345
+ "loss": 0.1694,
346
+ "step": 4300
347
+ },
348
+ {
349
+ "epoch": 4.087043189368771,
350
+ "grad_norm": 0.06297096610069275,
351
+ "learning_rate": 2.8485295927156392e-05,
352
+ "loss": 0.1654,
353
+ "step": 4400
354
+ },
355
+ {
356
+ "epoch": 4.09811738648948,
357
+ "grad_norm": 0.1834813803434372,
358
+ "learning_rate": 2.787006275378369e-05,
359
+ "loss": 0.125,
360
+ "step": 4500
361
+ },
362
+ {
363
+ "epoch": 4.1,
364
+ "eval_accuracy": 0.9676214196762142,
365
+ "eval_loss": 0.15175354480743408,
366
+ "eval_runtime": 241.1728,
367
+ "eval_samples_per_second": 3.33,
368
+ "eval_steps_per_second": 1.667,
369
+ "step": 4517
370
+ },
371
+ {
372
+ "epoch": 5.009136212624584,
373
+ "grad_norm": 1.12752103805542,
374
+ "learning_rate": 2.725482958041098e-05,
375
+ "loss": 0.1936,
376
+ "step": 4600
377
+ },
378
+ {
379
+ "epoch": 5.020210409745293,
380
+ "grad_norm": 0.0848553478717804,
381
+ "learning_rate": 2.6639596407038268e-05,
382
+ "loss": 0.034,
383
+ "step": 4700
384
+ },
385
+ {
386
+ "epoch": 5.031284606866002,
387
+ "grad_norm": 0.013141270726919174,
388
+ "learning_rate": 2.602436323366556e-05,
389
+ "loss": 0.0777,
390
+ "step": 4800
391
+ },
392
+ {
393
+ "epoch": 5.042358803986711,
394
+ "grad_norm": 0.02095056138932705,
395
+ "learning_rate": 2.540913006029285e-05,
396
+ "loss": 0.081,
397
+ "step": 4900
398
+ },
399
+ {
400
+ "epoch": 5.05343300110742,
401
+ "grad_norm": 17.830732345581055,
402
+ "learning_rate": 2.479389688692014e-05,
403
+ "loss": 0.0778,
404
+ "step": 5000
405
+ },
406
+ {
407
+ "epoch": 5.064507198228129,
408
+ "grad_norm": 0.08031286299228668,
409
+ "learning_rate": 2.4178663713547438e-05,
410
+ "loss": 0.0259,
411
+ "step": 5100
412
+ },
413
+ {
414
+ "epoch": 5.075581395348837,
415
+ "grad_norm": 0.30868440866470337,
416
+ "learning_rate": 2.3563430540174727e-05,
417
+ "loss": 0.123,
418
+ "step": 5200
419
+ },
420
+ {
421
+ "epoch": 5.086655592469546,
422
+ "grad_norm": 0.023195987567305565,
423
+ "learning_rate": 2.2948197366802017e-05,
424
+ "loss": 0.0355,
425
+ "step": 5300
426
+ },
427
+ {
428
+ "epoch": 5.097729789590255,
429
+ "grad_norm": 0.010711547918617725,
430
+ "learning_rate": 2.233296419342931e-05,
431
+ "loss": 0.1031,
432
+ "step": 5400
433
+ },
434
+ {
435
+ "epoch": 5.100055370985603,
436
+ "eval_accuracy": 0.9863013698630136,
437
+ "eval_loss": 0.046052780002355576,
438
+ "eval_runtime": 239.3825,
439
+ "eval_samples_per_second": 3.354,
440
+ "eval_steps_per_second": 1.679,
441
+ "step": 5421
442
+ },
443
+ {
444
+ "epoch": 6.00874861572536,
445
+ "grad_norm": 0.013949241489171982,
446
+ "learning_rate": 2.1717731020056604e-05,
447
+ "loss": 0.0928,
448
+ "step": 5500
449
+ },
450
+ {
451
+ "epoch": 6.019822812846068,
452
+ "grad_norm": 3.2788469791412354,
453
+ "learning_rate": 2.1102497846683894e-05,
454
+ "loss": 0.1632,
455
+ "step": 5600
456
+ },
457
+ {
458
+ "epoch": 6.030897009966777,
459
+ "grad_norm": 0.0950816348195076,
460
+ "learning_rate": 2.0487264673311183e-05,
461
+ "loss": 0.0995,
462
+ "step": 5700
463
+ },
464
+ {
465
+ "epoch": 6.041971207087486,
466
+ "grad_norm": 0.009910553693771362,
467
+ "learning_rate": 1.987203149993848e-05,
468
+ "loss": 0.1005,
469
+ "step": 5800
470
+ },
471
+ {
472
+ "epoch": 6.053045404208195,
473
+ "grad_norm": 0.14191927015781403,
474
+ "learning_rate": 1.925679832656577e-05,
475
+ "loss": 0.0874,
476
+ "step": 5900
477
+ },
478
+ {
479
+ "epoch": 6.064119601328904,
480
+ "grad_norm": 0.010242072865366936,
481
+ "learning_rate": 1.864156515319306e-05,
482
+ "loss": 0.0226,
483
+ "step": 6000
484
+ },
485
+ {
486
+ "epoch": 6.075193798449613,
487
+ "grad_norm": 0.02359813079237938,
488
+ "learning_rate": 1.8026331979820353e-05,
489
+ "loss": 0.0881,
490
+ "step": 6100
491
+ },
492
+ {
493
+ "epoch": 6.086267995570321,
494
+ "grad_norm": 0.014143938198685646,
495
+ "learning_rate": 1.7411098806447646e-05,
496
+ "loss": 0.0015,
497
+ "step": 6200
498
+ },
499
+ {
500
+ "epoch": 6.09734219269103,
501
+ "grad_norm": 0.007690469268709421,
502
+ "learning_rate": 1.6795865633074936e-05,
503
+ "loss": 0.0235,
504
+ "step": 6300
505
+ },
506
+ {
507
+ "epoch": 6.1,
508
+ "eval_accuracy": 0.9850560398505604,
509
+ "eval_loss": 0.07657956331968307,
510
+ "eval_runtime": 232.0013,
511
+ "eval_samples_per_second": 3.461,
512
+ "eval_steps_per_second": 1.733,
513
+ "step": 6324
514
+ },
515
+ {
516
+ "epoch": 7.008361018826135,
517
+ "grad_norm": 0.029869280755519867,
518
+ "learning_rate": 1.6180632459702226e-05,
519
+ "loss": 0.0093,
520
+ "step": 6400
521
+ },
522
+ {
523
+ "epoch": 7.019435215946844,
524
+ "grad_norm": 0.12701760232448578,
525
+ "learning_rate": 1.556539928632952e-05,
526
+ "loss": 0.0858,
527
+ "step": 6500
528
+ },
529
+ {
530
+ "epoch": 7.030509413067553,
531
+ "grad_norm": 0.00914853997528553,
532
+ "learning_rate": 1.4950166112956812e-05,
533
+ "loss": 0.0149,
534
+ "step": 6600
535
+ },
536
+ {
537
+ "epoch": 7.041583610188261,
538
+ "grad_norm": 0.006422719918191433,
539
+ "learning_rate": 1.4334932939584104e-05,
540
+ "loss": 0.0553,
541
+ "step": 6700
542
+ },
543
+ {
544
+ "epoch": 7.05265780730897,
545
+ "grad_norm": 0.006200818344950676,
546
+ "learning_rate": 1.3719699766211393e-05,
547
+ "loss": 0.0015,
548
+ "step": 6800
549
+ },
550
+ {
551
+ "epoch": 7.063732004429679,
552
+ "grad_norm": 0.006208031438291073,
553
+ "learning_rate": 1.3104466592838688e-05,
554
+ "loss": 0.0662,
555
+ "step": 6900
556
+ },
557
+ {
558
+ "epoch": 7.074806201550388,
559
+ "grad_norm": 0.07536833733320236,
560
+ "learning_rate": 1.2489233419465978e-05,
561
+ "loss": 0.0497,
562
+ "step": 7000
563
+ },
564
+ {
565
+ "epoch": 7.0858803986710965,
566
+ "grad_norm": 0.005990630015730858,
567
+ "learning_rate": 1.187400024609327e-05,
568
+ "loss": 0.02,
569
+ "step": 7100
570
+ },
571
+ {
572
+ "epoch": 7.096954595791805,
573
+ "grad_norm": 0.005243655759841204,
574
+ "learning_rate": 1.1258767072720563e-05,
575
+ "loss": 0.025,
576
+ "step": 7200
577
+ },
578
+ {
579
+ "epoch": 7.100055370985603,
580
+ "eval_accuracy": 0.9937733499377335,
581
+ "eval_loss": 0.030318038538098335,
582
+ "eval_runtime": 234.3879,
583
+ "eval_samples_per_second": 3.426,
584
+ "eval_steps_per_second": 1.715,
585
+ "step": 7228
586
+ },
587
+ {
588
+ "epoch": 8.007973421926911,
589
+ "grad_norm": 0.0038999952375888824,
590
+ "learning_rate": 1.0643533899347853e-05,
591
+ "loss": 0.0314,
592
+ "step": 7300
593
+ },
594
+ {
595
+ "epoch": 8.019047619047619,
596
+ "grad_norm": 0.010756886564195156,
597
+ "learning_rate": 1.0028300725975146e-05,
598
+ "loss": 0.003,
599
+ "step": 7400
600
+ },
601
+ {
602
+ "epoch": 8.030121816168327,
603
+ "grad_norm": 0.022609569132328033,
604
+ "learning_rate": 9.413067552602436e-06,
605
+ "loss": 0.0127,
606
+ "step": 7500
607
+ },
608
+ {
609
+ "epoch": 8.041196013289037,
610
+ "grad_norm": 0.006152728106826544,
611
+ "learning_rate": 8.797834379229729e-06,
612
+ "loss": 0.0078,
613
+ "step": 7600
614
+ },
615
+ {
616
+ "epoch": 8.052270210409745,
617
+ "grad_norm": 0.026683639734983444,
618
+ "learning_rate": 8.18260120585702e-06,
619
+ "loss": 0.0365,
620
+ "step": 7700
621
+ },
622
+ {
623
+ "epoch": 8.063344407530455,
624
+ "grad_norm": 0.0047752223908901215,
625
+ "learning_rate": 7.567368032484312e-06,
626
+ "loss": 0.0482,
627
+ "step": 7800
628
+ },
629
+ {
630
+ "epoch": 8.074418604651163,
631
+ "grad_norm": 0.00478146830573678,
632
+ "learning_rate": 6.952134859111603e-06,
633
+ "loss": 0.0013,
634
+ "step": 7900
635
+ },
636
+ {
637
+ "epoch": 8.085492801771872,
638
+ "grad_norm": 0.004836659412831068,
639
+ "learning_rate": 6.336901685738895e-06,
640
+ "loss": 0.0005,
641
+ "step": 8000
642
+ },
643
+ {
644
+ "epoch": 8.09656699889258,
645
+ "grad_norm": 304.2967224121094,
646
+ "learning_rate": 5.7216685123661875e-06,
647
+ "loss": 0.0611,
648
+ "step": 8100
649
+ },
650
+ {
651
+ "epoch": 8.1,
652
+ "eval_accuracy": 0.9925280199252802,
653
+ "eval_loss": 0.03501025214791298,
654
+ "eval_runtime": 230.6453,
655
+ "eval_samples_per_second": 3.482,
656
+ "eval_steps_per_second": 1.743,
657
+ "step": 8131
658
+ },
659
+ {
660
+ "epoch": 9.007585825027686,
661
+ "grad_norm": 0.019594158977270126,
662
+ "learning_rate": 5.106435338993479e-06,
663
+ "loss": 0.0018,
664
+ "step": 8200
665
+ },
666
+ {
667
+ "epoch": 9.018660022148394,
668
+ "grad_norm": 0.004379997495561838,
669
+ "learning_rate": 4.4912021656207705e-06,
670
+ "loss": 0.0278,
671
+ "step": 8300
672
+ },
673
+ {
674
+ "epoch": 9.029734219269104,
675
+ "grad_norm": 0.004317181650549173,
676
+ "learning_rate": 3.875968992248062e-06,
677
+ "loss": 0.0005,
678
+ "step": 8400
679
+ },
680
+ {
681
+ "epoch": 9.040808416389812,
682
+ "grad_norm": 0.003784640459343791,
683
+ "learning_rate": 3.260735818875354e-06,
684
+ "loss": 0.021,
685
+ "step": 8500
686
+ },
687
+ {
688
+ "epoch": 9.05188261351052,
689
+ "grad_norm": 0.004987145774066448,
690
+ "learning_rate": 2.6455026455026455e-06,
691
+ "loss": 0.0006,
692
+ "step": 8600
693
+ },
694
+ {
695
+ "epoch": 9.06295681063123,
696
+ "grad_norm": 0.018100284039974213,
697
+ "learning_rate": 2.0302694721299375e-06,
698
+ "loss": 0.0005,
699
+ "step": 8700
700
+ },
701
+ {
702
+ "epoch": 9.074031007751937,
703
+ "grad_norm": 0.006015971302986145,
704
+ "learning_rate": 1.415036298757229e-06,
705
+ "loss": 0.0005,
706
+ "step": 8800
707
+ },
708
+ {
709
+ "epoch": 9.085105204872647,
710
+ "grad_norm": 0.01569472998380661,
711
+ "learning_rate": 7.998031253845208e-07,
712
+ "loss": 0.0908,
713
+ "step": 8900
714
+ },
715
+ {
716
+ "epoch": 9.096179401993355,
717
+ "grad_norm": 0.0031440581660717726,
718
+ "learning_rate": 1.8456995201181249e-07,
719
+ "loss": 0.0071,
720
+ "step": 9000
721
+ },
722
+ {
723
+ "epoch": 9.099501661129569,
724
+ "eval_accuracy": 0.9950186799501868,
725
+ "eval_loss": 0.021772464737296104,
726
+ "eval_runtime": 237.3089,
727
+ "eval_samples_per_second": 3.384,
728
+ "eval_steps_per_second": 1.694,
729
+ "step": 9030
730
+ },
731
+ {
732
+ "epoch": 9.099501661129569,
733
+ "step": 9030,
734
+ "total_flos": 9.006226676074414e+19,
735
+ "train_loss": 1.6864825968290262,
736
+ "train_runtime": 29784.8769,
737
+ "train_samples_per_second": 2.425,
738
+ "train_steps_per_second": 0.303
739
+ },
740
+ {
741
+ "epoch": 9.099501661129569,
742
+ "eval_accuracy": 0.8934169278996865,
743
+ "eval_loss": 0.5165748596191406,
744
+ "eval_runtime": 353.997,
745
+ "eval_samples_per_second": 3.605,
746
+ "eval_steps_per_second": 1.802,
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
+ }