selmamalak commited on
Commit
162e0be
1 Parent(s): d270617

End of training

Browse files
Files changed (5) hide show
  1. README.md +5 -5
  2. all_results.json +16 -0
  3. eval_results.json +11 -0
  4. train_results.json +8 -0
  5. trainer_state.json +913 -0
README.md CHANGED
@@ -23,11 +23,11 @@ should probably proofread and complete it, then remove this comment. -->
23
 
24
  This model is a fine-tuned version of [facebook/deit-base-patch16-224](https://huggingface.co/facebook/deit-base-patch16-224) on the medmnist-v2 dataset.
25
  It achieves the following results on the evaluation set:
26
- - Loss: 0.5544
27
- - Accuracy: 0.7956
28
- - Precision: 0.6273
29
- - Recall: 0.5810
30
- - F1: 0.5998
31
 
32
  ## Model description
33
 
 
23
 
24
  This model is a fine-tuned version of [facebook/deit-base-patch16-224](https://huggingface.co/facebook/deit-base-patch16-224) on the medmnist-v2 dataset.
25
  It achieves the following results on the evaluation set:
26
+ - Loss: 0.5476
27
+ - Accuracy: 0.7960
28
+ - Precision: 0.6643
29
+ - Recall: 0.5891
30
+ - F1: 0.6164
31
 
32
  ## Model description
33
 
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.95,
3
+ "eval_accuracy": 0.7960099750623442,
4
+ "eval_f1": 0.6164338119893483,
5
+ "eval_loss": 0.54756760597229,
6
+ "eval_precision": 0.6642856199761363,
7
+ "eval_recall": 0.5891152176750621,
8
+ "eval_runtime": 11.0422,
9
+ "eval_samples_per_second": 181.576,
10
+ "eval_steps_per_second": 11.411,
11
+ "total_flos": 5.442882169274339e+18,
12
+ "train_loss": 0.7150778630457887,
13
+ "train_runtime": 818.9501,
14
+ "train_samples_per_second": 85.561,
15
+ "train_steps_per_second": 1.331
16
+ }
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.95,
3
+ "eval_accuracy": 0.7960099750623442,
4
+ "eval_f1": 0.6164338119893483,
5
+ "eval_loss": 0.54756760597229,
6
+ "eval_precision": 0.6642856199761363,
7
+ "eval_recall": 0.5891152176750621,
8
+ "eval_runtime": 11.0422,
9
+ "eval_samples_per_second": 181.576,
10
+ "eval_steps_per_second": 11.411
11
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.95,
3
+ "total_flos": 5.442882169274339e+18,
4
+ "train_loss": 0.7150778630457887,
5
+ "train_runtime": 818.9501,
6
+ "train_samples_per_second": 85.561,
7
+ "train_steps_per_second": 1.331
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,913 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.7956131605184447,
3
+ "best_model_checkpoint": "deit-base-patch16-224-finetuned-lora-medmnistv2/checkpoint-1090",
4
+ "epoch": 9.954337899543379,
5
+ "eval_steps": 500,
6
+ "global_step": 1090,
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.09,
13
+ "grad_norm": 3.768280267715454,
14
+ "learning_rate": 0.004954128440366973,
15
+ "loss": 1.2551,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 0.18,
20
+ "grad_norm": 1.6224374771118164,
21
+ "learning_rate": 0.004908256880733945,
22
+ "loss": 1.1194,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 0.27,
27
+ "grad_norm": 0.8902679085731506,
28
+ "learning_rate": 0.004862385321100918,
29
+ "loss": 0.9226,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 0.37,
34
+ "grad_norm": 1.336371898651123,
35
+ "learning_rate": 0.00481651376146789,
36
+ "loss": 0.9343,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 0.46,
41
+ "grad_norm": 1.9396265745162964,
42
+ "learning_rate": 0.0047706422018348625,
43
+ "loss": 0.849,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.55,
48
+ "grad_norm": 1.6278876066207886,
49
+ "learning_rate": 0.004724770642201835,
50
+ "loss": 0.8533,
51
+ "step": 60
52
+ },
53
+ {
54
+ "epoch": 0.64,
55
+ "grad_norm": 1.5454286336898804,
56
+ "learning_rate": 0.004678899082568808,
57
+ "loss": 0.8229,
58
+ "step": 70
59
+ },
60
+ {
61
+ "epoch": 0.73,
62
+ "grad_norm": 1.064020037651062,
63
+ "learning_rate": 0.00463302752293578,
64
+ "loss": 0.8622,
65
+ "step": 80
66
+ },
67
+ {
68
+ "epoch": 0.82,
69
+ "grad_norm": 1.5100964307785034,
70
+ "learning_rate": 0.0045871559633027525,
71
+ "loss": 0.8633,
72
+ "step": 90
73
+ },
74
+ {
75
+ "epoch": 0.91,
76
+ "grad_norm": 1.3673182725906372,
77
+ "learning_rate": 0.004541284403669725,
78
+ "loss": 0.9012,
79
+ "step": 100
80
+ },
81
+ {
82
+ "epoch": 1.0,
83
+ "eval_accuracy": 0.7228315054835494,
84
+ "eval_f1": 0.34621811450823214,
85
+ "eval_loss": 0.7629963159561157,
86
+ "eval_precision": 0.42629140706004787,
87
+ "eval_recall": 0.32692877938779574,
88
+ "eval_runtime": 6.2499,
89
+ "eval_samples_per_second": 160.484,
90
+ "eval_steps_per_second": 10.08,
91
+ "step": 109
92
+ },
93
+ {
94
+ "epoch": 1.0,
95
+ "grad_norm": 0.9423499703407288,
96
+ "learning_rate": 0.004495412844036698,
97
+ "loss": 0.8922,
98
+ "step": 110
99
+ },
100
+ {
101
+ "epoch": 1.1,
102
+ "grad_norm": 2.50795841217041,
103
+ "learning_rate": 0.0044495412844036695,
104
+ "loss": 0.8791,
105
+ "step": 120
106
+ },
107
+ {
108
+ "epoch": 1.19,
109
+ "grad_norm": 1.4502021074295044,
110
+ "learning_rate": 0.004403669724770643,
111
+ "loss": 0.8973,
112
+ "step": 130
113
+ },
114
+ {
115
+ "epoch": 1.28,
116
+ "grad_norm": 1.703600287437439,
117
+ "learning_rate": 0.004357798165137615,
118
+ "loss": 0.8246,
119
+ "step": 140
120
+ },
121
+ {
122
+ "epoch": 1.37,
123
+ "grad_norm": 0.7422580122947693,
124
+ "learning_rate": 0.004311926605504587,
125
+ "loss": 0.7913,
126
+ "step": 150
127
+ },
128
+ {
129
+ "epoch": 1.46,
130
+ "grad_norm": 1.146699070930481,
131
+ "learning_rate": 0.0042660550458715595,
132
+ "loss": 0.8028,
133
+ "step": 160
134
+ },
135
+ {
136
+ "epoch": 1.55,
137
+ "grad_norm": 2.6000311374664307,
138
+ "learning_rate": 0.004220183486238533,
139
+ "loss": 0.9162,
140
+ "step": 170
141
+ },
142
+ {
143
+ "epoch": 1.64,
144
+ "grad_norm": 0.8894189596176147,
145
+ "learning_rate": 0.004174311926605505,
146
+ "loss": 0.9022,
147
+ "step": 180
148
+ },
149
+ {
150
+ "epoch": 1.74,
151
+ "grad_norm": 0.9741529226303101,
152
+ "learning_rate": 0.004128440366972477,
153
+ "loss": 0.8879,
154
+ "step": 190
155
+ },
156
+ {
157
+ "epoch": 1.83,
158
+ "grad_norm": 2.898531198501587,
159
+ "learning_rate": 0.00408256880733945,
160
+ "loss": 0.7974,
161
+ "step": 200
162
+ },
163
+ {
164
+ "epoch": 1.92,
165
+ "grad_norm": 1.5416871309280396,
166
+ "learning_rate": 0.004036697247706422,
167
+ "loss": 0.7636,
168
+ "step": 210
169
+ },
170
+ {
171
+ "epoch": 2.0,
172
+ "eval_accuracy": 0.7288135593220338,
173
+ "eval_f1": 0.3788829143385214,
174
+ "eval_loss": 0.7212122082710266,
175
+ "eval_precision": 0.591156357572359,
176
+ "eval_recall": 0.3630725573348524,
177
+ "eval_runtime": 5.4131,
178
+ "eval_samples_per_second": 185.292,
179
+ "eval_steps_per_second": 11.638,
180
+ "step": 219
181
+ },
182
+ {
183
+ "epoch": 2.01,
184
+ "grad_norm": 0.9239124059677124,
185
+ "learning_rate": 0.003990825688073394,
186
+ "loss": 0.8466,
187
+ "step": 220
188
+ },
189
+ {
190
+ "epoch": 2.1,
191
+ "grad_norm": 0.9725480675697327,
192
+ "learning_rate": 0.003944954128440367,
193
+ "loss": 0.8851,
194
+ "step": 230
195
+ },
196
+ {
197
+ "epoch": 2.19,
198
+ "grad_norm": 1.1037930250167847,
199
+ "learning_rate": 0.0038990825688073397,
200
+ "loss": 0.7814,
201
+ "step": 240
202
+ },
203
+ {
204
+ "epoch": 2.28,
205
+ "grad_norm": 1.5679117441177368,
206
+ "learning_rate": 0.0038532110091743124,
207
+ "loss": 0.7781,
208
+ "step": 250
209
+ },
210
+ {
211
+ "epoch": 2.37,
212
+ "grad_norm": 0.7649789452552795,
213
+ "learning_rate": 0.0038073394495412843,
214
+ "loss": 0.8029,
215
+ "step": 260
216
+ },
217
+ {
218
+ "epoch": 2.47,
219
+ "grad_norm": 1.4153032302856445,
220
+ "learning_rate": 0.003761467889908257,
221
+ "loss": 0.7439,
222
+ "step": 270
223
+ },
224
+ {
225
+ "epoch": 2.56,
226
+ "grad_norm": 1.177064299583435,
227
+ "learning_rate": 0.0037155963302752293,
228
+ "loss": 0.8781,
229
+ "step": 280
230
+ },
231
+ {
232
+ "epoch": 2.65,
233
+ "grad_norm": 0.9871839284896851,
234
+ "learning_rate": 0.003669724770642202,
235
+ "loss": 0.7439,
236
+ "step": 290
237
+ },
238
+ {
239
+ "epoch": 2.74,
240
+ "grad_norm": 0.7878830432891846,
241
+ "learning_rate": 0.0036238532110091743,
242
+ "loss": 0.8001,
243
+ "step": 300
244
+ },
245
+ {
246
+ "epoch": 2.83,
247
+ "grad_norm": 1.0698702335357666,
248
+ "learning_rate": 0.003577981651376147,
249
+ "loss": 0.8568,
250
+ "step": 310
251
+ },
252
+ {
253
+ "epoch": 2.92,
254
+ "grad_norm": 0.9107087850570679,
255
+ "learning_rate": 0.0035321100917431194,
256
+ "loss": 0.7189,
257
+ "step": 320
258
+ },
259
+ {
260
+ "epoch": 3.0,
261
+ "eval_accuracy": 0.7258225324027916,
262
+ "eval_f1": 0.393460298491453,
263
+ "eval_loss": 0.7621892690658569,
264
+ "eval_precision": 0.4464810481413866,
265
+ "eval_recall": 0.422982191132074,
266
+ "eval_runtime": 5.2671,
267
+ "eval_samples_per_second": 190.428,
268
+ "eval_steps_per_second": 11.961,
269
+ "step": 328
270
+ },
271
+ {
272
+ "epoch": 3.01,
273
+ "grad_norm": 1.1653732061386108,
274
+ "learning_rate": 0.003486238532110092,
275
+ "loss": 0.7245,
276
+ "step": 330
277
+ },
278
+ {
279
+ "epoch": 3.11,
280
+ "grad_norm": 0.693824291229248,
281
+ "learning_rate": 0.0034403669724770644,
282
+ "loss": 0.7883,
283
+ "step": 340
284
+ },
285
+ {
286
+ "epoch": 3.2,
287
+ "grad_norm": 0.7583943009376526,
288
+ "learning_rate": 0.003394495412844037,
289
+ "loss": 0.7618,
290
+ "step": 350
291
+ },
292
+ {
293
+ "epoch": 3.29,
294
+ "grad_norm": 1.0829322338104248,
295
+ "learning_rate": 0.003348623853211009,
296
+ "loss": 0.7534,
297
+ "step": 360
298
+ },
299
+ {
300
+ "epoch": 3.38,
301
+ "grad_norm": 1.2232468128204346,
302
+ "learning_rate": 0.0033027522935779817,
303
+ "loss": 0.6827,
304
+ "step": 370
305
+ },
306
+ {
307
+ "epoch": 3.47,
308
+ "grad_norm": 1.0095463991165161,
309
+ "learning_rate": 0.003256880733944954,
310
+ "loss": 0.7615,
311
+ "step": 380
312
+ },
313
+ {
314
+ "epoch": 3.56,
315
+ "grad_norm": 1.059535026550293,
316
+ "learning_rate": 0.003211009174311927,
317
+ "loss": 0.8094,
318
+ "step": 390
319
+ },
320
+ {
321
+ "epoch": 3.65,
322
+ "grad_norm": 0.9983842372894287,
323
+ "learning_rate": 0.003165137614678899,
324
+ "loss": 0.7649,
325
+ "step": 400
326
+ },
327
+ {
328
+ "epoch": 3.74,
329
+ "grad_norm": 0.6007803082466125,
330
+ "learning_rate": 0.003119266055045872,
331
+ "loss": 0.7825,
332
+ "step": 410
333
+ },
334
+ {
335
+ "epoch": 3.84,
336
+ "grad_norm": 1.1800183057785034,
337
+ "learning_rate": 0.003073394495412844,
338
+ "loss": 0.7296,
339
+ "step": 420
340
+ },
341
+ {
342
+ "epoch": 3.93,
343
+ "grad_norm": 1.049116611480713,
344
+ "learning_rate": 0.003027522935779817,
345
+ "loss": 0.6904,
346
+ "step": 430
347
+ },
348
+ {
349
+ "epoch": 4.0,
350
+ "eval_accuracy": 0.7437686939182453,
351
+ "eval_f1": 0.4115031629499951,
352
+ "eval_loss": 0.7280949950218201,
353
+ "eval_precision": 0.48879535404583313,
354
+ "eval_recall": 0.4484100461852218,
355
+ "eval_runtime": 5.2319,
356
+ "eval_samples_per_second": 191.708,
357
+ "eval_steps_per_second": 12.041,
358
+ "step": 438
359
+ },
360
+ {
361
+ "epoch": 4.02,
362
+ "grad_norm": 0.9003276824951172,
363
+ "learning_rate": 0.002981651376146789,
364
+ "loss": 0.6584,
365
+ "step": 440
366
+ },
367
+ {
368
+ "epoch": 4.11,
369
+ "grad_norm": 1.3586938381195068,
370
+ "learning_rate": 0.002935779816513762,
371
+ "loss": 0.7175,
372
+ "step": 450
373
+ },
374
+ {
375
+ "epoch": 4.2,
376
+ "grad_norm": 0.5924133062362671,
377
+ "learning_rate": 0.0028899082568807338,
378
+ "loss": 0.67,
379
+ "step": 460
380
+ },
381
+ {
382
+ "epoch": 4.29,
383
+ "grad_norm": 0.9187817573547363,
384
+ "learning_rate": 0.0028440366972477065,
385
+ "loss": 0.697,
386
+ "step": 470
387
+ },
388
+ {
389
+ "epoch": 4.38,
390
+ "grad_norm": 0.7464513778686523,
391
+ "learning_rate": 0.002798165137614679,
392
+ "loss": 0.6691,
393
+ "step": 480
394
+ },
395
+ {
396
+ "epoch": 4.47,
397
+ "grad_norm": 0.6532416343688965,
398
+ "learning_rate": 0.0027522935779816515,
399
+ "loss": 0.7217,
400
+ "step": 490
401
+ },
402
+ {
403
+ "epoch": 4.57,
404
+ "grad_norm": 0.7422506809234619,
405
+ "learning_rate": 0.002706422018348624,
406
+ "loss": 0.7619,
407
+ "step": 500
408
+ },
409
+ {
410
+ "epoch": 4.66,
411
+ "grad_norm": 0.8745353817939758,
412
+ "learning_rate": 0.0026605504587155966,
413
+ "loss": 0.7264,
414
+ "step": 510
415
+ },
416
+ {
417
+ "epoch": 4.75,
418
+ "grad_norm": 0.7916358113288879,
419
+ "learning_rate": 0.002614678899082569,
420
+ "loss": 0.7592,
421
+ "step": 520
422
+ },
423
+ {
424
+ "epoch": 4.84,
425
+ "grad_norm": 1.0833995342254639,
426
+ "learning_rate": 0.0025688073394495416,
427
+ "loss": 0.7263,
428
+ "step": 530
429
+ },
430
+ {
431
+ "epoch": 4.93,
432
+ "grad_norm": 1.0176126956939697,
433
+ "learning_rate": 0.0025229357798165135,
434
+ "loss": 0.7658,
435
+ "step": 540
436
+ },
437
+ {
438
+ "epoch": 5.0,
439
+ "eval_accuracy": 0.7397806580259222,
440
+ "eval_f1": 0.3752685826577452,
441
+ "eval_loss": 0.7215412259101868,
442
+ "eval_precision": 0.4855492779862528,
443
+ "eval_recall": 0.42520631993699903,
444
+ "eval_runtime": 5.5429,
445
+ "eval_samples_per_second": 180.953,
446
+ "eval_steps_per_second": 11.366,
447
+ "step": 547
448
+ },
449
+ {
450
+ "epoch": 5.02,
451
+ "grad_norm": 1.45496666431427,
452
+ "learning_rate": 0.0024770642201834866,
453
+ "loss": 0.7024,
454
+ "step": 550
455
+ },
456
+ {
457
+ "epoch": 5.11,
458
+ "grad_norm": 0.8751833438873291,
459
+ "learning_rate": 0.002431192660550459,
460
+ "loss": 0.741,
461
+ "step": 560
462
+ },
463
+ {
464
+ "epoch": 5.21,
465
+ "grad_norm": 0.7062868475914001,
466
+ "learning_rate": 0.0023853211009174312,
467
+ "loss": 0.6733,
468
+ "step": 570
469
+ },
470
+ {
471
+ "epoch": 5.3,
472
+ "grad_norm": 0.9199981689453125,
473
+ "learning_rate": 0.002339449541284404,
474
+ "loss": 0.6988,
475
+ "step": 580
476
+ },
477
+ {
478
+ "epoch": 5.39,
479
+ "grad_norm": 0.8544884324073792,
480
+ "learning_rate": 0.0022935779816513763,
481
+ "loss": 0.7073,
482
+ "step": 590
483
+ },
484
+ {
485
+ "epoch": 5.48,
486
+ "grad_norm": 1.0523695945739746,
487
+ "learning_rate": 0.002247706422018349,
488
+ "loss": 0.6822,
489
+ "step": 600
490
+ },
491
+ {
492
+ "epoch": 5.57,
493
+ "grad_norm": 0.866678774356842,
494
+ "learning_rate": 0.0022018348623853213,
495
+ "loss": 0.6784,
496
+ "step": 610
497
+ },
498
+ {
499
+ "epoch": 5.66,
500
+ "grad_norm": 0.6989203095436096,
501
+ "learning_rate": 0.0021559633027522936,
502
+ "loss": 0.6796,
503
+ "step": 620
504
+ },
505
+ {
506
+ "epoch": 5.75,
507
+ "grad_norm": 0.4832663834095001,
508
+ "learning_rate": 0.0021100917431192663,
509
+ "loss": 0.6472,
510
+ "step": 630
511
+ },
512
+ {
513
+ "epoch": 5.84,
514
+ "grad_norm": 1.1230007410049438,
515
+ "learning_rate": 0.0020642201834862386,
516
+ "loss": 0.6693,
517
+ "step": 640
518
+ },
519
+ {
520
+ "epoch": 5.94,
521
+ "grad_norm": 0.6341413259506226,
522
+ "learning_rate": 0.002018348623853211,
523
+ "loss": 0.6363,
524
+ "step": 650
525
+ },
526
+ {
527
+ "epoch": 6.0,
528
+ "eval_accuracy": 0.7676969092721835,
529
+ "eval_f1": 0.5121263559819323,
530
+ "eval_loss": 0.6328557133674622,
531
+ "eval_precision": 0.6350277126119822,
532
+ "eval_recall": 0.49284577902844884,
533
+ "eval_runtime": 5.4702,
534
+ "eval_samples_per_second": 183.357,
535
+ "eval_steps_per_second": 11.517,
536
+ "step": 657
537
+ },
538
+ {
539
+ "epoch": 6.03,
540
+ "grad_norm": 0.7254393696784973,
541
+ "learning_rate": 0.0019724770642201837,
542
+ "loss": 0.6398,
543
+ "step": 660
544
+ },
545
+ {
546
+ "epoch": 6.12,
547
+ "grad_norm": 0.6462991237640381,
548
+ "learning_rate": 0.0019266055045871562,
549
+ "loss": 0.6703,
550
+ "step": 670
551
+ },
552
+ {
553
+ "epoch": 6.21,
554
+ "grad_norm": 0.7446946501731873,
555
+ "learning_rate": 0.0018807339449541285,
556
+ "loss": 0.6184,
557
+ "step": 680
558
+ },
559
+ {
560
+ "epoch": 6.3,
561
+ "grad_norm": 0.5196910500526428,
562
+ "learning_rate": 0.001834862385321101,
563
+ "loss": 0.6308,
564
+ "step": 690
565
+ },
566
+ {
567
+ "epoch": 6.39,
568
+ "grad_norm": 1.019028902053833,
569
+ "learning_rate": 0.0017889908256880735,
570
+ "loss": 0.664,
571
+ "step": 700
572
+ },
573
+ {
574
+ "epoch": 6.48,
575
+ "grad_norm": 1.3352869749069214,
576
+ "learning_rate": 0.001743119266055046,
577
+ "loss": 0.638,
578
+ "step": 710
579
+ },
580
+ {
581
+ "epoch": 6.58,
582
+ "grad_norm": 0.8105664253234863,
583
+ "learning_rate": 0.0016972477064220186,
584
+ "loss": 0.5801,
585
+ "step": 720
586
+ },
587
+ {
588
+ "epoch": 6.67,
589
+ "grad_norm": 1.0205323696136475,
590
+ "learning_rate": 0.0016513761467889909,
591
+ "loss": 0.6923,
592
+ "step": 730
593
+ },
594
+ {
595
+ "epoch": 6.76,
596
+ "grad_norm": 0.48183074593544006,
597
+ "learning_rate": 0.0016055045871559634,
598
+ "loss": 0.6583,
599
+ "step": 740
600
+ },
601
+ {
602
+ "epoch": 6.85,
603
+ "grad_norm": 0.7392696738243103,
604
+ "learning_rate": 0.001559633027522936,
605
+ "loss": 0.6784,
606
+ "step": 750
607
+ },
608
+ {
609
+ "epoch": 6.94,
610
+ "grad_norm": 0.7750418782234192,
611
+ "learning_rate": 0.0015137614678899084,
612
+ "loss": 0.6299,
613
+ "step": 760
614
+ },
615
+ {
616
+ "epoch": 7.0,
617
+ "eval_accuracy": 0.7716849451645065,
618
+ "eval_f1": 0.5713473581488847,
619
+ "eval_loss": 0.6116508841514587,
620
+ "eval_precision": 0.5962341473849692,
621
+ "eval_recall": 0.5781145727281558,
622
+ "eval_runtime": 5.4791,
623
+ "eval_samples_per_second": 183.06,
624
+ "eval_steps_per_second": 11.498,
625
+ "step": 766
626
+ },
627
+ {
628
+ "epoch": 7.03,
629
+ "grad_norm": 1.6914126873016357,
630
+ "learning_rate": 0.001467889908256881,
631
+ "loss": 0.7058,
632
+ "step": 770
633
+ },
634
+ {
635
+ "epoch": 7.12,
636
+ "grad_norm": 1.1266179084777832,
637
+ "learning_rate": 0.0014220183486238532,
638
+ "loss": 0.5969,
639
+ "step": 780
640
+ },
641
+ {
642
+ "epoch": 7.21,
643
+ "grad_norm": 0.8198122978210449,
644
+ "learning_rate": 0.0013761467889908258,
645
+ "loss": 0.6296,
646
+ "step": 790
647
+ },
648
+ {
649
+ "epoch": 7.31,
650
+ "grad_norm": 0.8668613433837891,
651
+ "learning_rate": 0.0013302752293577983,
652
+ "loss": 0.6779,
653
+ "step": 800
654
+ },
655
+ {
656
+ "epoch": 7.4,
657
+ "grad_norm": 0.757351279258728,
658
+ "learning_rate": 0.0012844036697247708,
659
+ "loss": 0.6589,
660
+ "step": 810
661
+ },
662
+ {
663
+ "epoch": 7.49,
664
+ "grad_norm": 1.231268286705017,
665
+ "learning_rate": 0.0012385321100917433,
666
+ "loss": 0.5909,
667
+ "step": 820
668
+ },
669
+ {
670
+ "epoch": 7.58,
671
+ "grad_norm": 0.6386240720748901,
672
+ "learning_rate": 0.0011926605504587156,
673
+ "loss": 0.6166,
674
+ "step": 830
675
+ },
676
+ {
677
+ "epoch": 7.67,
678
+ "grad_norm": 0.7646284103393555,
679
+ "learning_rate": 0.0011467889908256881,
680
+ "loss": 0.5815,
681
+ "step": 840
682
+ },
683
+ {
684
+ "epoch": 7.76,
685
+ "grad_norm": 1.1417447328567505,
686
+ "learning_rate": 0.0011009174311926607,
687
+ "loss": 0.5825,
688
+ "step": 850
689
+ },
690
+ {
691
+ "epoch": 7.85,
692
+ "grad_norm": 0.8213087916374207,
693
+ "learning_rate": 0.0010550458715596332,
694
+ "loss": 0.652,
695
+ "step": 860
696
+ },
697
+ {
698
+ "epoch": 7.95,
699
+ "grad_norm": 0.8747499585151672,
700
+ "learning_rate": 0.0010091743119266055,
701
+ "loss": 0.6011,
702
+ "step": 870
703
+ },
704
+ {
705
+ "epoch": 8.0,
706
+ "eval_accuracy": 0.7796610169491526,
707
+ "eval_f1": 0.5902452724888688,
708
+ "eval_loss": 0.5919255018234253,
709
+ "eval_precision": 0.6162323714867404,
710
+ "eval_recall": 0.575685794889542,
711
+ "eval_runtime": 5.3294,
712
+ "eval_samples_per_second": 188.202,
713
+ "eval_steps_per_second": 11.821,
714
+ "step": 876
715
+ },
716
+ {
717
+ "epoch": 8.04,
718
+ "grad_norm": 0.9114809632301331,
719
+ "learning_rate": 0.0009633027522935781,
720
+ "loss": 0.6365,
721
+ "step": 880
722
+ },
723
+ {
724
+ "epoch": 8.13,
725
+ "grad_norm": 0.7782912254333496,
726
+ "learning_rate": 0.0009174311926605505,
727
+ "loss": 0.5558,
728
+ "step": 890
729
+ },
730
+ {
731
+ "epoch": 8.22,
732
+ "grad_norm": 1.2907453775405884,
733
+ "learning_rate": 0.000871559633027523,
734
+ "loss": 0.5756,
735
+ "step": 900
736
+ },
737
+ {
738
+ "epoch": 8.31,
739
+ "grad_norm": 1.1318057775497437,
740
+ "learning_rate": 0.0008256880733944954,
741
+ "loss": 0.6574,
742
+ "step": 910
743
+ },
744
+ {
745
+ "epoch": 8.4,
746
+ "grad_norm": 0.5756754875183105,
747
+ "learning_rate": 0.000779816513761468,
748
+ "loss": 0.5494,
749
+ "step": 920
750
+ },
751
+ {
752
+ "epoch": 8.49,
753
+ "grad_norm": 0.7023201584815979,
754
+ "learning_rate": 0.0007339449541284405,
755
+ "loss": 0.6101,
756
+ "step": 930
757
+ },
758
+ {
759
+ "epoch": 8.58,
760
+ "grad_norm": 0.9913588166236877,
761
+ "learning_rate": 0.0006880733944954129,
762
+ "loss": 0.6034,
763
+ "step": 940
764
+ },
765
+ {
766
+ "epoch": 8.68,
767
+ "grad_norm": 0.982480525970459,
768
+ "learning_rate": 0.0006422018348623854,
769
+ "loss": 0.5981,
770
+ "step": 950
771
+ },
772
+ {
773
+ "epoch": 8.77,
774
+ "grad_norm": 1.1374191045761108,
775
+ "learning_rate": 0.0005963302752293578,
776
+ "loss": 0.6134,
777
+ "step": 960
778
+ },
779
+ {
780
+ "epoch": 8.86,
781
+ "grad_norm": 0.7602503299713135,
782
+ "learning_rate": 0.0005504587155963303,
783
+ "loss": 0.5561,
784
+ "step": 970
785
+ },
786
+ {
787
+ "epoch": 8.95,
788
+ "grad_norm": 0.8967719674110413,
789
+ "learning_rate": 0.0005045871559633027,
790
+ "loss": 0.6043,
791
+ "step": 980
792
+ },
793
+ {
794
+ "epoch": 9.0,
795
+ "eval_accuracy": 0.7946161515453639,
796
+ "eval_f1": 0.5983005873522603,
797
+ "eval_loss": 0.5475942492485046,
798
+ "eval_precision": 0.6295002331514908,
799
+ "eval_recall": 0.5812943349243114,
800
+ "eval_runtime": 5.2749,
801
+ "eval_samples_per_second": 190.145,
802
+ "eval_steps_per_second": 11.943,
803
+ "step": 985
804
+ },
805
+ {
806
+ "epoch": 9.04,
807
+ "grad_norm": 0.9292692542076111,
808
+ "learning_rate": 0.00045871559633027525,
809
+ "loss": 0.5415,
810
+ "step": 990
811
+ },
812
+ {
813
+ "epoch": 9.13,
814
+ "grad_norm": 0.8102772235870361,
815
+ "learning_rate": 0.0004128440366972477,
816
+ "loss": 0.5611,
817
+ "step": 1000
818
+ },
819
+ {
820
+ "epoch": 9.22,
821
+ "grad_norm": 1.1832762956619263,
822
+ "learning_rate": 0.00036697247706422024,
823
+ "loss": 0.5876,
824
+ "step": 1010
825
+ },
826
+ {
827
+ "epoch": 9.32,
828
+ "grad_norm": 1.1258927583694458,
829
+ "learning_rate": 0.0003211009174311927,
830
+ "loss": 0.4946,
831
+ "step": 1020
832
+ },
833
+ {
834
+ "epoch": 9.41,
835
+ "grad_norm": 0.8979453444480896,
836
+ "learning_rate": 0.00027522935779816516,
837
+ "loss": 0.5668,
838
+ "step": 1030
839
+ },
840
+ {
841
+ "epoch": 9.5,
842
+ "grad_norm": 0.9506226778030396,
843
+ "learning_rate": 0.00022935779816513763,
844
+ "loss": 0.5384,
845
+ "step": 1040
846
+ },
847
+ {
848
+ "epoch": 9.59,
849
+ "grad_norm": 1.1283026933670044,
850
+ "learning_rate": 0.00018348623853211012,
851
+ "loss": 0.5772,
852
+ "step": 1050
853
+ },
854
+ {
855
+ "epoch": 9.68,
856
+ "grad_norm": 0.9244889616966248,
857
+ "learning_rate": 0.00013761467889908258,
858
+ "loss": 0.5662,
859
+ "step": 1060
860
+ },
861
+ {
862
+ "epoch": 9.77,
863
+ "grad_norm": 0.8923238515853882,
864
+ "learning_rate": 9.174311926605506e-05,
865
+ "loss": 0.5921,
866
+ "step": 1070
867
+ },
868
+ {
869
+ "epoch": 9.86,
870
+ "grad_norm": 1.1653807163238525,
871
+ "learning_rate": 4.587155963302753e-05,
872
+ "loss": 0.5846,
873
+ "step": 1080
874
+ },
875
+ {
876
+ "epoch": 9.95,
877
+ "grad_norm": 0.8477634191513062,
878
+ "learning_rate": 0.0,
879
+ "loss": 0.5671,
880
+ "step": 1090
881
+ },
882
+ {
883
+ "epoch": 9.95,
884
+ "eval_accuracy": 0.7956131605184447,
885
+ "eval_f1": 0.5997799904348683,
886
+ "eval_loss": 0.5543830990791321,
887
+ "eval_precision": 0.627341257239154,
888
+ "eval_recall": 0.5810367281093276,
889
+ "eval_runtime": 5.4717,
890
+ "eval_samples_per_second": 183.308,
891
+ "eval_steps_per_second": 11.514,
892
+ "step": 1090
893
+ },
894
+ {
895
+ "epoch": 9.95,
896
+ "step": 1090,
897
+ "total_flos": 5.442882169274339e+18,
898
+ "train_loss": 0.7150778630457887,
899
+ "train_runtime": 818.9501,
900
+ "train_samples_per_second": 85.561,
901
+ "train_steps_per_second": 1.331
902
+ }
903
+ ],
904
+ "logging_steps": 10,
905
+ "max_steps": 1090,
906
+ "num_input_tokens_seen": 0,
907
+ "num_train_epochs": 10,
908
+ "save_steps": 500,
909
+ "total_flos": 5.442882169274339e+18,
910
+ "train_batch_size": 16,
911
+ "trial_name": null,
912
+ "trial_params": null
913
+ }