c14kevincardenas commited on
Commit
f031120
·
verified ·
1 Parent(s): c0ad7dc

End of training

Browse files
Files changed (5) hide show
  1. README.md +7 -5
  2. all_results.json +13 -0
  3. eval_results.json +8 -0
  4. train_results.json +8 -0
  5. trainer_state.json +726 -0
README.md CHANGED
@@ -1,8 +1,10 @@
1
  ---
2
  library_name: transformers
3
- license: other
4
- base_model: apple/mobilevit-small
5
  tags:
 
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
@@ -16,10 +18,10 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # mobilevit-small_alpha0.7_temp5.0
18
 
19
- This model is a fine-tuned version of [apple/mobilevit-small](https://huggingface.co/apple/mobilevit-small) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.8711
22
- - Accuracy: 0.6403
23
 
24
  ## Model description
25
 
 
1
  ---
2
  library_name: transformers
3
+ license: apache-2.0
4
+ base_model: c14kevincardenas/ClimBEiTv2
5
  tags:
6
+ - knowledge_distillation
7
+ - vision
8
  - generated_from_trainer
9
  metrics:
10
  - accuracy
 
18
 
19
  # mobilevit-small_alpha0.7_temp5.0
20
 
21
+ This model is a fine-tuned version of [c14kevincardenas/ClimBEiTv2](https://huggingface.co/c14kevincardenas/ClimBEiTv2) on the c14kevincardenas/beta_caller_284_person_crop_seq_withlimb_2 dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.8550
24
+ - Accuracy: 0.6680
25
 
26
  ## Model description
27
 
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "eval_accuracy": 0.6679841897233202,
4
+ "eval_loss": 0.8549718856811523,
5
+ "eval_runtime": 30.7112,
6
+ "eval_samples_per_second": 32.952,
7
+ "eval_steps_per_second": 0.521,
8
+ "total_flos": 0.0,
9
+ "train_loss": 0.6163041236665514,
10
+ "train_runtime": 10385.8394,
11
+ "train_samples_per_second": 11.034,
12
+ "train_steps_per_second": 0.173
13
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "eval_accuracy": 0.6679841897233202,
4
+ "eval_loss": 0.8549718856811523,
5
+ "eval_runtime": 30.7112,
6
+ "eval_samples_per_second": 32.952,
7
+ "eval_steps_per_second": 0.521
8
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "total_flos": 0.0,
4
+ "train_loss": 0.6163041236665514,
5
+ "train_runtime": 10385.8394,
6
+ "train_samples_per_second": 11.034,
7
+ "train_steps_per_second": 0.173
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,726 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.8549718856811523,
3
+ "best_model_checkpoint": "./kd_results/apple/mobilevit-small_alpha0.7_temp5.0/checkpoint-1710",
4
+ "epoch": 20.0,
5
+ "eval_steps": 500,
6
+ "global_step": 1800,
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.2777777777777778,
13
+ "grad_norm": 5.067017078399658,
14
+ "learning_rate": 6.944444444444445e-06,
15
+ "loss": 1.3328,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.5555555555555556,
20
+ "grad_norm": 1.9458526372909546,
21
+ "learning_rate": 1.388888888888889e-05,
22
+ "loss": 1.136,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.8333333333333334,
27
+ "grad_norm": 1.8754066228866577,
28
+ "learning_rate": 2.0833333333333336e-05,
29
+ "loss": 1.1311,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 1.0,
34
+ "eval_accuracy": 0.24703557312252963,
35
+ "eval_loss": 1.4279208183288574,
36
+ "eval_runtime": 28.5592,
37
+ "eval_samples_per_second": 35.435,
38
+ "eval_steps_per_second": 0.56,
39
+ "step": 90
40
+ },
41
+ {
42
+ "epoch": 1.1111111111111112,
43
+ "grad_norm": 2.692765951156616,
44
+ "learning_rate": 2.777777777777778e-05,
45
+ "loss": 1.1094,
46
+ "step": 100
47
+ },
48
+ {
49
+ "epoch": 1.3888888888888888,
50
+ "grad_norm": 3.097252607345581,
51
+ "learning_rate": 3.472222222222222e-05,
52
+ "loss": 1.1009,
53
+ "step": 125
54
+ },
55
+ {
56
+ "epoch": 1.6666666666666665,
57
+ "grad_norm": 1.8249784708023071,
58
+ "learning_rate": 4.166666666666667e-05,
59
+ "loss": 1.1182,
60
+ "step": 150
61
+ },
62
+ {
63
+ "epoch": 1.9444444444444444,
64
+ "grad_norm": 3.1942126750946045,
65
+ "learning_rate": 4.8611111111111115e-05,
66
+ "loss": 1.0789,
67
+ "step": 175
68
+ },
69
+ {
70
+ "epoch": 2.0,
71
+ "eval_accuracy": 0.3191699604743083,
72
+ "eval_loss": 1.333102822303772,
73
+ "eval_runtime": 28.5466,
74
+ "eval_samples_per_second": 35.451,
75
+ "eval_steps_per_second": 0.56,
76
+ "step": 180
77
+ },
78
+ {
79
+ "epoch": 2.2222222222222223,
80
+ "grad_norm": 5.066985607147217,
81
+ "learning_rate": 4.938271604938271e-05,
82
+ "loss": 1.0393,
83
+ "step": 200
84
+ },
85
+ {
86
+ "epoch": 2.5,
87
+ "grad_norm": 4.512152671813965,
88
+ "learning_rate": 4.8611111111111115e-05,
89
+ "loss": 1.0193,
90
+ "step": 225
91
+ },
92
+ {
93
+ "epoch": 2.7777777777777777,
94
+ "grad_norm": 4.158876895904541,
95
+ "learning_rate": 4.783950617283951e-05,
96
+ "loss": 0.9867,
97
+ "step": 250
98
+ },
99
+ {
100
+ "epoch": 3.0,
101
+ "eval_accuracy": 0.38636363636363635,
102
+ "eval_loss": 1.2525354623794556,
103
+ "eval_runtime": 28.5695,
104
+ "eval_samples_per_second": 35.422,
105
+ "eval_steps_per_second": 0.56,
106
+ "step": 270
107
+ },
108
+ {
109
+ "epoch": 3.0555555555555554,
110
+ "grad_norm": 4.744362831115723,
111
+ "learning_rate": 4.70679012345679e-05,
112
+ "loss": 0.961,
113
+ "step": 275
114
+ },
115
+ {
116
+ "epoch": 3.3333333333333335,
117
+ "grad_norm": 4.1431050300598145,
118
+ "learning_rate": 4.62962962962963e-05,
119
+ "loss": 0.8868,
120
+ "step": 300
121
+ },
122
+ {
123
+ "epoch": 3.611111111111111,
124
+ "grad_norm": 3.916766405105591,
125
+ "learning_rate": 4.5524691358024696e-05,
126
+ "loss": 0.8619,
127
+ "step": 325
128
+ },
129
+ {
130
+ "epoch": 3.888888888888889,
131
+ "grad_norm": 3.961512804031372,
132
+ "learning_rate": 4.4753086419753084e-05,
133
+ "loss": 0.8433,
134
+ "step": 350
135
+ },
136
+ {
137
+ "epoch": 4.0,
138
+ "eval_accuracy": 0.49703557312252966,
139
+ "eval_loss": 1.067132830619812,
140
+ "eval_runtime": 28.8584,
141
+ "eval_samples_per_second": 35.068,
142
+ "eval_steps_per_second": 0.554,
143
+ "step": 360
144
+ },
145
+ {
146
+ "epoch": 4.166666666666667,
147
+ "grad_norm": 4.347306728363037,
148
+ "learning_rate": 4.3981481481481486e-05,
149
+ "loss": 0.7999,
150
+ "step": 375
151
+ },
152
+ {
153
+ "epoch": 4.444444444444445,
154
+ "grad_norm": 5.401865005493164,
155
+ "learning_rate": 4.3209876543209875e-05,
156
+ "loss": 0.7392,
157
+ "step": 400
158
+ },
159
+ {
160
+ "epoch": 4.722222222222222,
161
+ "grad_norm": 5.163962364196777,
162
+ "learning_rate": 4.243827160493827e-05,
163
+ "loss": 0.725,
164
+ "step": 425
165
+ },
166
+ {
167
+ "epoch": 5.0,
168
+ "grad_norm": 6.064295291900635,
169
+ "learning_rate": 4.166666666666667e-05,
170
+ "loss": 0.7496,
171
+ "step": 450
172
+ },
173
+ {
174
+ "epoch": 5.0,
175
+ "eval_accuracy": 0.5602766798418972,
176
+ "eval_loss": 0.9999355673789978,
177
+ "eval_runtime": 28.6185,
178
+ "eval_samples_per_second": 35.362,
179
+ "eval_steps_per_second": 0.559,
180
+ "step": 450
181
+ },
182
+ {
183
+ "epoch": 5.277777777777778,
184
+ "grad_norm": 3.596616744995117,
185
+ "learning_rate": 4.089506172839506e-05,
186
+ "loss": 0.7087,
187
+ "step": 475
188
+ },
189
+ {
190
+ "epoch": 5.555555555555555,
191
+ "grad_norm": 8.382354736328125,
192
+ "learning_rate": 4.012345679012346e-05,
193
+ "loss": 0.7061,
194
+ "step": 500
195
+ },
196
+ {
197
+ "epoch": 5.833333333333333,
198
+ "grad_norm": 5.777190685272217,
199
+ "learning_rate": 3.935185185185186e-05,
200
+ "loss": 0.6948,
201
+ "step": 525
202
+ },
203
+ {
204
+ "epoch": 6.0,
205
+ "eval_accuracy": 0.5988142292490118,
206
+ "eval_loss": 0.937413215637207,
207
+ "eval_runtime": 28.9208,
208
+ "eval_samples_per_second": 34.992,
209
+ "eval_steps_per_second": 0.553,
210
+ "step": 540
211
+ },
212
+ {
213
+ "epoch": 6.111111111111111,
214
+ "grad_norm": 4.378117561340332,
215
+ "learning_rate": 3.8580246913580246e-05,
216
+ "loss": 0.6473,
217
+ "step": 550
218
+ },
219
+ {
220
+ "epoch": 6.388888888888889,
221
+ "grad_norm": 6.292396545410156,
222
+ "learning_rate": 3.780864197530865e-05,
223
+ "loss": 0.6451,
224
+ "step": 575
225
+ },
226
+ {
227
+ "epoch": 6.666666666666667,
228
+ "grad_norm": 10.105191230773926,
229
+ "learning_rate": 3.7037037037037037e-05,
230
+ "loss": 0.6372,
231
+ "step": 600
232
+ },
233
+ {
234
+ "epoch": 6.944444444444445,
235
+ "grad_norm": 4.2663350105285645,
236
+ "learning_rate": 3.626543209876543e-05,
237
+ "loss": 0.6495,
238
+ "step": 625
239
+ },
240
+ {
241
+ "epoch": 7.0,
242
+ "eval_accuracy": 0.6146245059288538,
243
+ "eval_loss": 0.9065309166908264,
244
+ "eval_runtime": 28.1886,
245
+ "eval_samples_per_second": 35.901,
246
+ "eval_steps_per_second": 0.568,
247
+ "step": 630
248
+ },
249
+ {
250
+ "epoch": 7.222222222222222,
251
+ "grad_norm": 6.072882175445557,
252
+ "learning_rate": 3.5493827160493834e-05,
253
+ "loss": 0.5943,
254
+ "step": 650
255
+ },
256
+ {
257
+ "epoch": 7.5,
258
+ "grad_norm": 7.997155666351318,
259
+ "learning_rate": 3.472222222222222e-05,
260
+ "loss": 0.592,
261
+ "step": 675
262
+ },
263
+ {
264
+ "epoch": 7.777777777777778,
265
+ "grad_norm": 5.652843952178955,
266
+ "learning_rate": 3.395061728395062e-05,
267
+ "loss": 0.6101,
268
+ "step": 700
269
+ },
270
+ {
271
+ "epoch": 8.0,
272
+ "eval_accuracy": 0.6007905138339921,
273
+ "eval_loss": 0.9330638647079468,
274
+ "eval_runtime": 28.9448,
275
+ "eval_samples_per_second": 34.963,
276
+ "eval_steps_per_second": 0.553,
277
+ "step": 720
278
+ },
279
+ {
280
+ "epoch": 8.055555555555555,
281
+ "grad_norm": 5.346714019775391,
282
+ "learning_rate": 3.317901234567901e-05,
283
+ "loss": 0.5961,
284
+ "step": 725
285
+ },
286
+ {
287
+ "epoch": 8.333333333333334,
288
+ "grad_norm": 5.7756123542785645,
289
+ "learning_rate": 3.240740740740741e-05,
290
+ "loss": 0.5641,
291
+ "step": 750
292
+ },
293
+ {
294
+ "epoch": 8.61111111111111,
295
+ "grad_norm": 4.24644136428833,
296
+ "learning_rate": 3.16358024691358e-05,
297
+ "loss": 0.5661,
298
+ "step": 775
299
+ },
300
+ {
301
+ "epoch": 8.88888888888889,
302
+ "grad_norm": 7.723337173461914,
303
+ "learning_rate": 3.08641975308642e-05,
304
+ "loss": 0.5591,
305
+ "step": 800
306
+ },
307
+ {
308
+ "epoch": 9.0,
309
+ "eval_accuracy": 0.6571146245059288,
310
+ "eval_loss": 0.8673222064971924,
311
+ "eval_runtime": 29.0659,
312
+ "eval_samples_per_second": 34.817,
313
+ "eval_steps_per_second": 0.55,
314
+ "step": 810
315
+ },
316
+ {
317
+ "epoch": 9.166666666666666,
318
+ "grad_norm": 6.067270755767822,
319
+ "learning_rate": 3.0092592592592593e-05,
320
+ "loss": 0.5474,
321
+ "step": 825
322
+ },
323
+ {
324
+ "epoch": 9.444444444444445,
325
+ "grad_norm": 4.425726413726807,
326
+ "learning_rate": 2.9320987654320992e-05,
327
+ "loss": 0.5453,
328
+ "step": 850
329
+ },
330
+ {
331
+ "epoch": 9.722222222222221,
332
+ "grad_norm": 10.190550804138184,
333
+ "learning_rate": 2.8549382716049384e-05,
334
+ "loss": 0.5452,
335
+ "step": 875
336
+ },
337
+ {
338
+ "epoch": 10.0,
339
+ "grad_norm": 8.709136009216309,
340
+ "learning_rate": 2.777777777777778e-05,
341
+ "loss": 0.5406,
342
+ "step": 900
343
+ },
344
+ {
345
+ "epoch": 10.0,
346
+ "eval_accuracy": 0.6235177865612648,
347
+ "eval_loss": 0.8704849481582642,
348
+ "eval_runtime": 28.8308,
349
+ "eval_samples_per_second": 35.101,
350
+ "eval_steps_per_second": 0.555,
351
+ "step": 900
352
+ },
353
+ {
354
+ "epoch": 10.277777777777779,
355
+ "grad_norm": 2.9625649452209473,
356
+ "learning_rate": 2.700617283950617e-05,
357
+ "loss": 0.5191,
358
+ "step": 925
359
+ },
360
+ {
361
+ "epoch": 10.555555555555555,
362
+ "grad_norm": 5.884112358093262,
363
+ "learning_rate": 2.623456790123457e-05,
364
+ "loss": 0.5156,
365
+ "step": 950
366
+ },
367
+ {
368
+ "epoch": 10.833333333333334,
369
+ "grad_norm": 9.00991153717041,
370
+ "learning_rate": 2.5462962962962965e-05,
371
+ "loss": 0.5188,
372
+ "step": 975
373
+ },
374
+ {
375
+ "epoch": 11.0,
376
+ "eval_accuracy": 0.6630434782608695,
377
+ "eval_loss": 0.8578750491142273,
378
+ "eval_runtime": 28.2136,
379
+ "eval_samples_per_second": 35.869,
380
+ "eval_steps_per_second": 0.567,
381
+ "step": 990
382
+ },
383
+ {
384
+ "epoch": 11.11111111111111,
385
+ "grad_norm": 6.289473533630371,
386
+ "learning_rate": 2.4691358024691357e-05,
387
+ "loss": 0.5064,
388
+ "step": 1000
389
+ },
390
+ {
391
+ "epoch": 11.38888888888889,
392
+ "grad_norm": 5.463348388671875,
393
+ "learning_rate": 2.3919753086419755e-05,
394
+ "loss": 0.4854,
395
+ "step": 1025
396
+ },
397
+ {
398
+ "epoch": 11.666666666666666,
399
+ "grad_norm": 5.968869686126709,
400
+ "learning_rate": 2.314814814814815e-05,
401
+ "loss": 0.4956,
402
+ "step": 1050
403
+ },
404
+ {
405
+ "epoch": 11.944444444444445,
406
+ "grad_norm": 3.9279537200927734,
407
+ "learning_rate": 2.2376543209876542e-05,
408
+ "loss": 0.4875,
409
+ "step": 1075
410
+ },
411
+ {
412
+ "epoch": 12.0,
413
+ "eval_accuracy": 0.6324110671936759,
414
+ "eval_loss": 0.8808870315551758,
415
+ "eval_runtime": 28.7749,
416
+ "eval_samples_per_second": 35.17,
417
+ "eval_steps_per_second": 0.556,
418
+ "step": 1080
419
+ },
420
+ {
421
+ "epoch": 12.222222222222221,
422
+ "grad_norm": 2.7190964221954346,
423
+ "learning_rate": 2.1604938271604937e-05,
424
+ "loss": 0.4716,
425
+ "step": 1100
426
+ },
427
+ {
428
+ "epoch": 12.5,
429
+ "grad_norm": 5.235175132751465,
430
+ "learning_rate": 2.0833333333333336e-05,
431
+ "loss": 0.4614,
432
+ "step": 1125
433
+ },
434
+ {
435
+ "epoch": 12.777777777777779,
436
+ "grad_norm": 4.206543445587158,
437
+ "learning_rate": 2.006172839506173e-05,
438
+ "loss": 0.4596,
439
+ "step": 1150
440
+ },
441
+ {
442
+ "epoch": 13.0,
443
+ "eval_accuracy": 0.6324110671936759,
444
+ "eval_loss": 0.8903521299362183,
445
+ "eval_runtime": 28.7353,
446
+ "eval_samples_per_second": 35.218,
447
+ "eval_steps_per_second": 0.557,
448
+ "step": 1170
449
+ },
450
+ {
451
+ "epoch": 13.055555555555555,
452
+ "grad_norm": 5.413076400756836,
453
+ "learning_rate": 1.9290123456790123e-05,
454
+ "loss": 0.4607,
455
+ "step": 1175
456
+ },
457
+ {
458
+ "epoch": 13.333333333333334,
459
+ "grad_norm": 7.035764217376709,
460
+ "learning_rate": 1.8518518518518518e-05,
461
+ "loss": 0.4549,
462
+ "step": 1200
463
+ },
464
+ {
465
+ "epoch": 13.61111111111111,
466
+ "grad_norm": 7.080498218536377,
467
+ "learning_rate": 1.7746913580246917e-05,
468
+ "loss": 0.445,
469
+ "step": 1225
470
+ },
471
+ {
472
+ "epoch": 13.88888888888889,
473
+ "grad_norm": 5.1540937423706055,
474
+ "learning_rate": 1.697530864197531e-05,
475
+ "loss": 0.4418,
476
+ "step": 1250
477
+ },
478
+ {
479
+ "epoch": 14.0,
480
+ "eval_accuracy": 0.6541501976284585,
481
+ "eval_loss": 0.8596166968345642,
482
+ "eval_runtime": 28.5111,
483
+ "eval_samples_per_second": 35.495,
484
+ "eval_steps_per_second": 0.561,
485
+ "step": 1260
486
+ },
487
+ {
488
+ "epoch": 14.166666666666666,
489
+ "grad_norm": 4.73467493057251,
490
+ "learning_rate": 1.6203703703703704e-05,
491
+ "loss": 0.4534,
492
+ "step": 1275
493
+ },
494
+ {
495
+ "epoch": 14.444444444444445,
496
+ "grad_norm": 3.350318193435669,
497
+ "learning_rate": 1.54320987654321e-05,
498
+ "loss": 0.4293,
499
+ "step": 1300
500
+ },
501
+ {
502
+ "epoch": 14.722222222222221,
503
+ "grad_norm": 5.757190227508545,
504
+ "learning_rate": 1.4660493827160496e-05,
505
+ "loss": 0.437,
506
+ "step": 1325
507
+ },
508
+ {
509
+ "epoch": 15.0,
510
+ "grad_norm": 10.14773178100586,
511
+ "learning_rate": 1.388888888888889e-05,
512
+ "loss": 0.4385,
513
+ "step": 1350
514
+ },
515
+ {
516
+ "epoch": 15.0,
517
+ "eval_accuracy": 0.6304347826086957,
518
+ "eval_loss": 0.8768613338470459,
519
+ "eval_runtime": 28.9356,
520
+ "eval_samples_per_second": 34.974,
521
+ "eval_steps_per_second": 0.553,
522
+ "step": 1350
523
+ },
524
+ {
525
+ "epoch": 15.277777777777779,
526
+ "grad_norm": 4.023919105529785,
527
+ "learning_rate": 1.3117283950617285e-05,
528
+ "loss": 0.418,
529
+ "step": 1375
530
+ },
531
+ {
532
+ "epoch": 15.555555555555555,
533
+ "grad_norm": 5.213397026062012,
534
+ "learning_rate": 1.2345679012345678e-05,
535
+ "loss": 0.4343,
536
+ "step": 1400
537
+ },
538
+ {
539
+ "epoch": 15.833333333333334,
540
+ "grad_norm": 5.7322797775268555,
541
+ "learning_rate": 1.1574074074074075e-05,
542
+ "loss": 0.4175,
543
+ "step": 1425
544
+ },
545
+ {
546
+ "epoch": 16.0,
547
+ "eval_accuracy": 0.6561264822134387,
548
+ "eval_loss": 0.8624676465988159,
549
+ "eval_runtime": 28.9613,
550
+ "eval_samples_per_second": 34.943,
551
+ "eval_steps_per_second": 0.552,
552
+ "step": 1440
553
+ },
554
+ {
555
+ "epoch": 16.11111111111111,
556
+ "grad_norm": 2.6554200649261475,
557
+ "learning_rate": 1.0802469135802469e-05,
558
+ "loss": 0.4027,
559
+ "step": 1450
560
+ },
561
+ {
562
+ "epoch": 16.38888888888889,
563
+ "grad_norm": 4.859523773193359,
564
+ "learning_rate": 1.0030864197530866e-05,
565
+ "loss": 0.4188,
566
+ "step": 1475
567
+ },
568
+ {
569
+ "epoch": 16.666666666666668,
570
+ "grad_norm": 6.244664669036865,
571
+ "learning_rate": 9.259259259259259e-06,
572
+ "loss": 0.4194,
573
+ "step": 1500
574
+ },
575
+ {
576
+ "epoch": 16.944444444444443,
577
+ "grad_norm": 4.917417049407959,
578
+ "learning_rate": 8.487654320987654e-06,
579
+ "loss": 0.413,
580
+ "step": 1525
581
+ },
582
+ {
583
+ "epoch": 17.0,
584
+ "eval_accuracy": 0.6571146245059288,
585
+ "eval_loss": 0.8577585220336914,
586
+ "eval_runtime": 28.9271,
587
+ "eval_samples_per_second": 34.985,
588
+ "eval_steps_per_second": 0.553,
589
+ "step": 1530
590
+ },
591
+ {
592
+ "epoch": 17.22222222222222,
593
+ "grad_norm": 5.908167839050293,
594
+ "learning_rate": 7.71604938271605e-06,
595
+ "loss": 0.4077,
596
+ "step": 1550
597
+ },
598
+ {
599
+ "epoch": 17.5,
600
+ "grad_norm": 4.492807388305664,
601
+ "learning_rate": 6.944444444444445e-06,
602
+ "loss": 0.3998,
603
+ "step": 1575
604
+ },
605
+ {
606
+ "epoch": 17.77777777777778,
607
+ "grad_norm": 3.352818489074707,
608
+ "learning_rate": 6.172839506172839e-06,
609
+ "loss": 0.3973,
610
+ "step": 1600
611
+ },
612
+ {
613
+ "epoch": 18.0,
614
+ "eval_accuracy": 0.658102766798419,
615
+ "eval_loss": 0.857342541217804,
616
+ "eval_runtime": 29.3279,
617
+ "eval_samples_per_second": 34.506,
618
+ "eval_steps_per_second": 0.546,
619
+ "step": 1620
620
+ },
621
+ {
622
+ "epoch": 18.055555555555557,
623
+ "grad_norm": 3.0056979656219482,
624
+ "learning_rate": 5.401234567901234e-06,
625
+ "loss": 0.405,
626
+ "step": 1625
627
+ },
628
+ {
629
+ "epoch": 18.333333333333332,
630
+ "grad_norm": 3.8960962295532227,
631
+ "learning_rate": 4.6296296296296296e-06,
632
+ "loss": 0.4028,
633
+ "step": 1650
634
+ },
635
+ {
636
+ "epoch": 18.61111111111111,
637
+ "grad_norm": 3.9841091632843018,
638
+ "learning_rate": 3.858024691358025e-06,
639
+ "loss": 0.3954,
640
+ "step": 1675
641
+ },
642
+ {
643
+ "epoch": 18.88888888888889,
644
+ "grad_norm": 3.473726749420166,
645
+ "learning_rate": 3.0864197530864196e-06,
646
+ "loss": 0.3964,
647
+ "step": 1700
648
+ },
649
+ {
650
+ "epoch": 19.0,
651
+ "eval_accuracy": 0.6679841897233202,
652
+ "eval_loss": 0.8549718856811523,
653
+ "eval_runtime": 28.7243,
654
+ "eval_samples_per_second": 35.232,
655
+ "eval_steps_per_second": 0.557,
656
+ "step": 1710
657
+ },
658
+ {
659
+ "epoch": 19.166666666666668,
660
+ "grad_norm": 3.783536672592163,
661
+ "learning_rate": 2.3148148148148148e-06,
662
+ "loss": 0.401,
663
+ "step": 1725
664
+ },
665
+ {
666
+ "epoch": 19.444444444444443,
667
+ "grad_norm": 2.4962282180786133,
668
+ "learning_rate": 1.5432098765432098e-06,
669
+ "loss": 0.3949,
670
+ "step": 1750
671
+ },
672
+ {
673
+ "epoch": 19.72222222222222,
674
+ "grad_norm": 5.234297752380371,
675
+ "learning_rate": 7.716049382716049e-07,
676
+ "loss": 0.3994,
677
+ "step": 1775
678
+ },
679
+ {
680
+ "epoch": 20.0,
681
+ "grad_norm": 10.562817573547363,
682
+ "learning_rate": 0.0,
683
+ "loss": 0.4004,
684
+ "step": 1800
685
+ },
686
+ {
687
+ "epoch": 20.0,
688
+ "eval_accuracy": 0.6403162055335968,
689
+ "eval_loss": 0.8710843324661255,
690
+ "eval_runtime": 28.7285,
691
+ "eval_samples_per_second": 35.226,
692
+ "eval_steps_per_second": 0.557,
693
+ "step": 1800
694
+ },
695
+ {
696
+ "epoch": 20.0,
697
+ "step": 1800,
698
+ "total_flos": 0.0,
699
+ "train_loss": 0.6163041236665514,
700
+ "train_runtime": 10385.8394,
701
+ "train_samples_per_second": 11.034,
702
+ "train_steps_per_second": 0.173
703
+ }
704
+ ],
705
+ "logging_steps": 25,
706
+ "max_steps": 1800,
707
+ "num_input_tokens_seen": 0,
708
+ "num_train_epochs": 20,
709
+ "save_steps": 500,
710
+ "stateful_callbacks": {
711
+ "TrainerControl": {
712
+ "args": {
713
+ "should_epoch_stop": false,
714
+ "should_evaluate": false,
715
+ "should_log": false,
716
+ "should_save": true,
717
+ "should_training_stop": true
718
+ },
719
+ "attributes": {}
720
+ }
721
+ },
722
+ "total_flos": 0.0,
723
+ "train_batch_size": 64,
724
+ "trial_name": null,
725
+ "trial_params": null
726
+ }