mikhail-panzo commited on
Commit
7c21ae5
1 Parent(s): be87573

Training in progress, step 3000, checkpoint

Browse files
checkpoint-3000/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:61ab35843ce36cdb5a57f281c67c01867aff952e256c13dc41ff7d83873527c8
3
  size 577789320
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c61b2dc892339c3acc2fb4e5779d443427022a2e1e45877d33966f006c38064
3
  size 577789320
checkpoint-3000/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9cfd8c65e67e5537db496c7cde5a3da2426080cb8fcc955384ce3bb5205dc826
3
  size 1155772233
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c85b2a5d88f03cd8b051400143032da46c139a64884ba76fb2050ccd7310163
3
  size 1155772233
checkpoint-3000/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed8b64d2b9675b3822d90fbace5be0418e655114989d4633b7304a9e48ef7a07
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfe7d7f3086eb3f8d77ea389748fb4527346ae1ff6c3fa944ba21155aa600e8a
3
  size 14244
checkpoint-3000/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4fffdc59c95629c2f51b5d1ae5c411cab9d0ed1b87aa833d9f835c9d1d4a1025
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b669784ac10692534c390adc0bb33ad380d01e66637c5e74b0b2b5ba96afd67
3
  size 1064
checkpoint-3000/trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "best_metric": 0.3533318042755127,
3
  "best_model_checkpoint": "mikhail_panzo/zlm_b64_le4_s12000/checkpoint-3000",
4
- "epoch": 2.513717277486911,
5
  "eval_steps": 500,
6
  "global_step": 3000,
7
  "is_hyper_param_search": false,
@@ -9,471 +9,471 @@
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 0.041884816753926704,
13
- "grad_norm": 2.697540283203125,
14
- "learning_rate": 2.4000000000000003e-06,
15
- "loss": 1.0526,
16
  "step": 50
17
  },
18
  {
19
- "epoch": 0.08376963350785341,
20
- "grad_norm": 4.0351996421813965,
21
- "learning_rate": 4.9000000000000005e-06,
22
- "loss": 0.8484,
23
  "step": 100
24
  },
25
  {
26
- "epoch": 0.1256544502617801,
27
- "grad_norm": 4.858840465545654,
28
- "learning_rate": 7.4e-06,
29
- "loss": 0.7505,
30
  "step": 150
31
  },
32
  {
33
- "epoch": 0.16753926701570682,
34
- "grad_norm": 2.396157741546631,
35
- "learning_rate": 9.900000000000002e-06,
36
- "loss": 0.6889,
37
  "step": 200
38
  },
39
  {
40
- "epoch": 0.2094240837696335,
41
- "grad_norm": 2.660891056060791,
42
- "learning_rate": 1.24e-05,
43
- "loss": 0.6363,
44
  "step": 250
45
  },
46
  {
47
- "epoch": 0.2513089005235602,
48
- "grad_norm": 2.848365068435669,
49
- "learning_rate": 1.49e-05,
50
- "loss": 0.6107,
51
  "step": 300
52
  },
53
  {
54
- "epoch": 0.2931937172774869,
55
- "grad_norm": 2.0031681060791016,
56
- "learning_rate": 1.74e-05,
57
- "loss": 0.5673,
58
  "step": 350
59
  },
60
  {
61
- "epoch": 0.33507853403141363,
62
- "grad_norm": 4.771584510803223,
63
- "learning_rate": 1.9900000000000003e-05,
64
- "loss": 0.5566,
65
  "step": 400
66
  },
67
  {
68
- "epoch": 0.3769633507853403,
69
- "grad_norm": 9.876974105834961,
70
- "learning_rate": 2.2400000000000002e-05,
71
- "loss": 0.5487,
72
  "step": 450
73
  },
74
  {
75
- "epoch": 0.418848167539267,
76
- "grad_norm": 2.113546848297119,
77
- "learning_rate": 2.4900000000000002e-05,
78
- "loss": 0.5268,
79
  "step": 500
80
  },
81
  {
82
- "epoch": 0.418848167539267,
83
- "eval_loss": 0.46557939052581787,
84
- "eval_runtime": 297.9193,
85
- "eval_samples_per_second": 28.494,
86
- "eval_steps_per_second": 3.565,
87
  "step": 500
88
  },
89
  {
90
- "epoch": 0.4607329842931937,
91
- "grad_norm": 2.7003655433654785,
92
- "learning_rate": 2.7400000000000002e-05,
93
- "loss": 0.5212,
94
  "step": 550
95
  },
96
  {
97
- "epoch": 0.5026178010471204,
98
- "grad_norm": 5.649191379547119,
99
- "learning_rate": 2.9900000000000002e-05,
100
- "loss": 0.5138,
101
  "step": 600
102
  },
103
  {
104
- "epoch": 0.5445026178010471,
105
- "grad_norm": 2.272181749343872,
106
- "learning_rate": 3.24e-05,
107
- "loss": 0.5121,
108
  "step": 650
109
  },
110
  {
111
- "epoch": 0.5863874345549738,
112
- "grad_norm": 2.957671642303467,
113
- "learning_rate": 3.49e-05,
114
- "loss": 0.496,
115
  "step": 700
116
  },
117
  {
118
- "epoch": 0.6282722513089005,
119
- "grad_norm": 3.577693462371826,
120
- "learning_rate": 3.74e-05,
121
- "loss": 0.4886,
122
  "step": 750
123
  },
124
  {
125
- "epoch": 0.6701570680628273,
126
- "grad_norm": 2.466113328933716,
127
- "learning_rate": 3.99e-05,
128
- "loss": 0.4879,
129
  "step": 800
130
  },
131
  {
132
- "epoch": 0.7120418848167539,
133
- "grad_norm": 2.941208600997925,
134
- "learning_rate": 4.24e-05,
135
- "loss": 0.4865,
136
  "step": 850
137
  },
138
  {
139
- "epoch": 0.7539267015706806,
140
- "grad_norm": 2.3843142986297607,
141
- "learning_rate": 4.49e-05,
142
- "loss": 0.4689,
143
  "step": 900
144
  },
145
  {
146
- "epoch": 0.7958115183246073,
147
- "grad_norm": 2.232208728790283,
148
- "learning_rate": 4.74e-05,
149
- "loss": 0.4768,
150
  "step": 950
151
  },
152
  {
153
- "epoch": 0.837696335078534,
154
- "grad_norm": 2.1702969074249268,
155
- "learning_rate": 4.99e-05,
156
- "loss": 0.4583,
157
  "step": 1000
158
  },
159
  {
160
- "epoch": 0.837696335078534,
161
- "eval_loss": 0.4044812321662903,
162
- "eval_runtime": 296.3081,
163
- "eval_samples_per_second": 28.649,
164
- "eval_steps_per_second": 3.584,
165
  "step": 1000
166
  },
167
  {
168
- "epoch": 0.8795811518324608,
169
- "grad_norm": 2.7613942623138428,
170
- "learning_rate": 5.2400000000000007e-05,
171
- "loss": 0.4664,
172
  "step": 1050
173
  },
174
  {
175
- "epoch": 0.9214659685863874,
176
- "grad_norm": 2.4752321243286133,
177
- "learning_rate": 5.4900000000000006e-05,
178
- "loss": 0.442,
179
  "step": 1100
180
  },
181
  {
182
- "epoch": 0.9633507853403142,
183
- "grad_norm": 2.27407169342041,
184
- "learning_rate": 5.74e-05,
185
- "loss": 0.4498,
186
  "step": 1150
187
  },
188
  {
189
- "epoch": 1.0052356020942408,
190
- "grad_norm": 1.8998862504959106,
191
- "learning_rate": 5.99e-05,
192
- "loss": 0.4433,
193
  "step": 1200
194
  },
195
  {
196
- "epoch": 1.0471204188481675,
197
- "grad_norm": 2.8479738235473633,
198
- "learning_rate": 6.24e-05,
199
- "loss": 0.4417,
200
  "step": 1250
201
  },
202
  {
203
- "epoch": 1.0890052356020943,
204
- "grad_norm": 1.9095953702926636,
205
- "learning_rate": 6.49e-05,
206
- "loss": 0.444,
207
  "step": 1300
208
  },
209
  {
210
- "epoch": 1.130890052356021,
211
- "grad_norm": 1.368077278137207,
212
- "learning_rate": 6.740000000000001e-05,
213
- "loss": 0.4519,
214
  "step": 1350
215
  },
216
  {
217
- "epoch": 1.1727748691099475,
218
- "grad_norm": 2.0456721782684326,
219
- "learning_rate": 6.99e-05,
220
- "loss": 0.4411,
221
  "step": 1400
222
  },
223
  {
224
- "epoch": 1.2146596858638743,
225
- "grad_norm": 1.6217619180679321,
226
- "learning_rate": 7.24e-05,
227
- "loss": 0.4324,
228
  "step": 1450
229
  },
230
  {
231
- "epoch": 1.256544502617801,
232
- "grad_norm": 2.961646795272827,
233
- "learning_rate": 7.49e-05,
234
- "loss": 0.4321,
235
  "step": 1500
236
  },
237
  {
238
- "epoch": 1.256544502617801,
239
- "eval_loss": 0.3930450975894928,
240
- "eval_runtime": 301.1428,
241
- "eval_samples_per_second": 28.189,
242
- "eval_steps_per_second": 3.527,
243
  "step": 1500
244
  },
245
  {
246
- "epoch": 1.2984293193717278,
247
- "grad_norm": 2.543201446533203,
248
- "learning_rate": 7.740000000000001e-05,
249
- "loss": 0.4364,
250
  "step": 1550
251
  },
252
  {
253
- "epoch": 1.3403141361256545,
254
- "grad_norm": 1.5807560682296753,
255
- "learning_rate": 7.99e-05,
256
- "loss": 0.433,
257
  "step": 1600
258
  },
259
  {
260
- "epoch": 1.3821989528795813,
261
- "grad_norm": 1.587385892868042,
262
- "learning_rate": 8.24e-05,
263
- "loss": 0.4285,
264
  "step": 1650
265
  },
266
  {
267
- "epoch": 1.4240837696335078,
268
- "grad_norm": 2.1177687644958496,
269
- "learning_rate": 8.49e-05,
270
- "loss": 0.4209,
271
  "step": 1700
272
  },
273
  {
274
- "epoch": 1.4659685863874345,
275
- "grad_norm": 1.6078702211380005,
276
- "learning_rate": 8.740000000000001e-05,
277
- "loss": 0.426,
278
  "step": 1750
279
  },
280
  {
281
- "epoch": 1.5078534031413613,
282
- "grad_norm": 2.10492205619812,
283
- "learning_rate": 8.99e-05,
284
- "loss": 0.4191,
285
  "step": 1800
286
  },
287
  {
288
- "epoch": 1.5497382198952878,
289
- "grad_norm": 2.169309616088867,
290
- "learning_rate": 9.240000000000001e-05,
291
- "loss": 0.4178,
292
  "step": 1850
293
  },
294
  {
295
- "epoch": 1.5916230366492146,
296
- "grad_norm": 1.559006929397583,
297
- "learning_rate": 9.49e-05,
298
- "loss": 0.4133,
299
  "step": 1900
300
  },
301
  {
302
- "epoch": 1.6335078534031413,
303
- "grad_norm": 2.0436346530914307,
304
- "learning_rate": 9.74e-05,
305
- "loss": 0.4234,
306
  "step": 1950
307
  },
308
  {
309
- "epoch": 1.675392670157068,
310
- "grad_norm": 2.089486837387085,
311
- "learning_rate": 9.99e-05,
312
- "loss": 0.4092,
313
  "step": 2000
314
  },
315
  {
316
- "epoch": 1.675392670157068,
317
- "eval_loss": 0.37686464190483093,
318
- "eval_runtime": 310.3113,
319
- "eval_samples_per_second": 27.356,
320
- "eval_steps_per_second": 3.422,
321
  "step": 2000
322
  },
323
  {
324
- "epoch": 1.7179057591623037,
325
- "grad_norm": 2.684563398361206,
326
- "learning_rate": 9.952e-05,
327
- "loss": 0.4099,
328
  "step": 2050
329
  },
330
  {
331
- "epoch": 1.7597905759162304,
332
- "grad_norm": 1.7815254926681519,
333
- "learning_rate": 9.902e-05,
334
- "loss": 0.4093,
335
  "step": 2100
336
  },
337
  {
338
- "epoch": 1.8016753926701572,
339
- "grad_norm": 1.9900892972946167,
340
- "learning_rate": 9.852e-05,
341
- "loss": 0.4116,
342
  "step": 2150
343
  },
344
  {
345
- "epoch": 1.8435602094240837,
346
- "grad_norm": 1.8637460470199585,
347
- "learning_rate": 9.802e-05,
348
- "loss": 0.4089,
349
  "step": 2200
350
  },
351
  {
352
- "epoch": 1.8854450261780105,
353
- "grad_norm": 1.58171546459198,
354
- "learning_rate": 9.752e-05,
355
- "loss": 0.4053,
356
  "step": 2250
357
  },
358
  {
359
- "epoch": 1.9273298429319372,
360
- "grad_norm": 1.4995630979537964,
361
- "learning_rate": 9.702e-05,
362
- "loss": 0.4049,
363
  "step": 2300
364
  },
365
  {
366
- "epoch": 1.9692146596858637,
367
- "grad_norm": 2.447049379348755,
368
- "learning_rate": 9.652e-05,
369
- "loss": 0.4029,
370
  "step": 2350
371
  },
372
  {
373
- "epoch": 2.0110994764397905,
374
- "grad_norm": 1.8776074647903442,
375
- "learning_rate": 9.602e-05,
376
- "loss": 0.3988,
377
  "step": 2400
378
  },
379
  {
380
- "epoch": 2.0529842931937172,
381
- "grad_norm": 3.1785056591033936,
382
- "learning_rate": 9.552000000000001e-05,
383
- "loss": 0.4023,
384
  "step": 2450
385
  },
386
  {
387
- "epoch": 2.094869109947644,
388
- "grad_norm": 1.4416707754135132,
389
- "learning_rate": 9.502000000000001e-05,
390
- "loss": 0.4008,
391
  "step": 2500
392
  },
393
  {
394
- "epoch": 2.094869109947644,
395
- "eval_loss": 0.35975897312164307,
396
- "eval_runtime": 275.2855,
397
- "eval_samples_per_second": 30.837,
398
- "eval_steps_per_second": 3.858,
399
  "step": 2500
400
  },
401
  {
402
- "epoch": 2.1367539267015707,
403
- "grad_norm": 1.3332245349884033,
404
- "learning_rate": 9.452000000000001e-05,
405
- "loss": 0.3985,
406
  "step": 2550
407
  },
408
  {
409
- "epoch": 2.1786387434554975,
410
- "grad_norm": 1.291338324546814,
411
- "learning_rate": 9.402000000000001e-05,
412
- "loss": 0.4021,
413
  "step": 2600
414
  },
415
  {
416
- "epoch": 2.220523560209424,
417
- "grad_norm": 1.6172423362731934,
418
- "learning_rate": 9.352000000000001e-05,
419
- "loss": 0.4006,
420
  "step": 2650
421
  },
422
  {
423
- "epoch": 2.262408376963351,
424
- "grad_norm": 1.8729867935180664,
425
- "learning_rate": 9.302e-05,
426
- "loss": 0.3966,
427
  "step": 2700
428
  },
429
  {
430
- "epoch": 2.3042931937172773,
431
- "grad_norm": 1.623901605606079,
432
- "learning_rate": 9.252e-05,
433
- "loss": 0.3951,
434
  "step": 2750
435
  },
436
  {
437
- "epoch": 2.346178010471204,
438
- "grad_norm": 1.032320261001587,
439
- "learning_rate": 9.202e-05,
440
- "loss": 0.3904,
441
  "step": 2800
442
  },
443
  {
444
- "epoch": 2.3880628272251307,
445
- "grad_norm": 2.0027778148651123,
446
- "learning_rate": 9.152e-05,
447
- "loss": 0.3933,
448
  "step": 2850
449
  },
450
  {
451
- "epoch": 2.4299476439790575,
452
- "grad_norm": 1.5681087970733643,
453
- "learning_rate": 9.102e-05,
454
- "loss": 0.4024,
455
  "step": 2900
456
  },
457
  {
458
- "epoch": 2.4718324607329842,
459
- "grad_norm": 1.2643622159957886,
460
- "learning_rate": 9.052000000000001e-05,
461
- "loss": 0.3939,
462
  "step": 2950
463
  },
464
  {
465
- "epoch": 2.513717277486911,
466
- "grad_norm": 1.4029465913772583,
467
- "learning_rate": 9.002000000000001e-05,
468
- "loss": 0.3941,
469
  "step": 3000
470
  },
471
  {
472
- "epoch": 2.513717277486911,
473
- "eval_loss": 0.3533318042755127,
474
- "eval_runtime": 271.9646,
475
- "eval_samples_per_second": 31.214,
476
- "eval_steps_per_second": 3.905,
477
  "step": 3000
478
  }
479
  ],
@@ -494,8 +494,8 @@
494
  "attributes": {}
495
  }
496
  },
497
- "total_flos": 2.691966274071149e+16,
498
- "train_batch_size": 16,
499
  "trial_name": null,
500
  "trial_params": null
501
  }
 
1
  {
2
+ "best_metric": 0.3613213300704956,
3
  "best_model_checkpoint": "mikhail_panzo/zlm_b64_le4_s12000/checkpoint-3000",
4
+ "epoch": 2.5125628140703515,
5
  "eval_steps": 500,
6
  "global_step": 3000,
7
  "is_hyper_param_search": false,
 
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
+ "epoch": 0.04187604690117253,
13
+ "grad_norm": 3.0184223651885986,
14
+ "learning_rate": 2.35e-06,
15
+ "loss": 1.1206,
16
  "step": 50
17
  },
18
  {
19
+ "epoch": 0.08375209380234507,
20
+ "grad_norm": 3.4442005157470703,
21
+ "learning_rate": 4.85e-06,
22
+ "loss": 0.8185,
23
  "step": 100
24
  },
25
  {
26
+ "epoch": 0.12562814070351758,
27
+ "grad_norm": 2.6812493801116943,
28
+ "learning_rate": 7.35e-06,
29
+ "loss": 0.7576,
30
  "step": 150
31
  },
32
  {
33
+ "epoch": 0.16750418760469013,
34
+ "grad_norm": 4.518482685089111,
35
+ "learning_rate": 9.85e-06,
36
+ "loss": 0.6938,
37
  "step": 200
38
  },
39
  {
40
+ "epoch": 0.20938023450586266,
41
+ "grad_norm": 4.683146953582764,
42
+ "learning_rate": 1.235e-05,
43
+ "loss": 0.6386,
44
  "step": 250
45
  },
46
  {
47
+ "epoch": 0.25125628140703515,
48
+ "grad_norm": 2.7392578125,
49
+ "learning_rate": 1.485e-05,
50
+ "loss": 0.6226,
51
  "step": 300
52
  },
53
  {
54
+ "epoch": 0.2931323283082077,
55
+ "grad_norm": 2.017056465148926,
56
+ "learning_rate": 1.7349999999999998e-05,
57
+ "loss": 0.6128,
58
  "step": 350
59
  },
60
  {
61
+ "epoch": 0.33500837520938026,
62
+ "grad_norm": 3.6669230461120605,
63
+ "learning_rate": 1.985e-05,
64
+ "loss": 0.5675,
65
  "step": 400
66
  },
67
  {
68
+ "epoch": 0.3768844221105528,
69
+ "grad_norm": 4.16896915435791,
70
+ "learning_rate": 2.235e-05,
71
+ "loss": 0.5655,
72
  "step": 450
73
  },
74
  {
75
+ "epoch": 0.4187604690117253,
76
+ "grad_norm": 3.691459894180298,
77
+ "learning_rate": 2.485e-05,
78
+ "loss": 0.5487,
79
  "step": 500
80
  },
81
  {
82
+ "epoch": 0.4187604690117253,
83
+ "eval_loss": 0.47458764910697937,
84
+ "eval_runtime": 214.7196,
85
+ "eval_samples_per_second": 39.535,
86
+ "eval_steps_per_second": 4.946,
87
  "step": 500
88
  },
89
  {
90
+ "epoch": 0.46063651591289784,
91
+ "grad_norm": 3.3792996406555176,
92
+ "learning_rate": 2.7350000000000004e-05,
93
+ "loss": 0.5232,
94
  "step": 550
95
  },
96
  {
97
+ "epoch": 0.5025125628140703,
98
+ "grad_norm": 3.6263363361358643,
99
+ "learning_rate": 2.985e-05,
100
+ "loss": 0.5132,
101
  "step": 600
102
  },
103
  {
104
+ "epoch": 0.5443886097152428,
105
+ "grad_norm": 2.652538776397705,
106
+ "learning_rate": 3.235e-05,
107
+ "loss": 0.5067,
108
  "step": 650
109
  },
110
  {
111
+ "epoch": 0.5862646566164154,
112
+ "grad_norm": 3.5355172157287598,
113
+ "learning_rate": 3.485e-05,
114
+ "loss": 0.4981,
115
  "step": 700
116
  },
117
  {
118
+ "epoch": 0.628140703517588,
119
+ "grad_norm": 2.5035619735717773,
120
+ "learning_rate": 3.735e-05,
121
+ "loss": 0.5094,
122
  "step": 750
123
  },
124
  {
125
+ "epoch": 0.6700167504187605,
126
+ "grad_norm": 2.9585518836975098,
127
+ "learning_rate": 3.9850000000000006e-05,
128
+ "loss": 0.5053,
129
  "step": 800
130
  },
131
  {
132
+ "epoch": 0.711892797319933,
133
+ "grad_norm": 9.997116088867188,
134
+ "learning_rate": 4.235e-05,
135
+ "loss": 0.502,
136
  "step": 850
137
  },
138
  {
139
+ "epoch": 0.7537688442211056,
140
+ "grad_norm": 1.8630377054214478,
141
+ "learning_rate": 4.4850000000000006e-05,
142
+ "loss": 0.483,
143
  "step": 900
144
  },
145
  {
146
+ "epoch": 0.7956448911222781,
147
+ "grad_norm": 1.5534099340438843,
148
+ "learning_rate": 4.735e-05,
149
+ "loss": 0.4743,
150
  "step": 950
151
  },
152
  {
153
+ "epoch": 0.8375209380234506,
154
+ "grad_norm": 2.8155503273010254,
155
+ "learning_rate": 4.9850000000000006e-05,
156
+ "loss": 0.483,
157
  "step": 1000
158
  },
159
  {
160
+ "epoch": 0.8375209380234506,
161
+ "eval_loss": 0.4226582944393158,
162
+ "eval_runtime": 213.2339,
163
+ "eval_samples_per_second": 39.811,
164
+ "eval_steps_per_second": 4.98,
165
  "step": 1000
166
  },
167
  {
168
+ "epoch": 0.8793969849246231,
169
+ "grad_norm": 4.2074055671691895,
170
+ "learning_rate": 5.235e-05,
171
+ "loss": 0.4625,
172
  "step": 1050
173
  },
174
  {
175
+ "epoch": 0.9212730318257957,
176
+ "grad_norm": 2.215416431427002,
177
+ "learning_rate": 5.485e-05,
178
+ "loss": 0.461,
179
  "step": 1100
180
  },
181
  {
182
+ "epoch": 0.9631490787269682,
183
+ "grad_norm": 3.3321831226348877,
184
+ "learning_rate": 5.7350000000000005e-05,
185
+ "loss": 0.4807,
186
  "step": 1150
187
  },
188
  {
189
+ "epoch": 1.0050251256281406,
190
+ "grad_norm": 3.5796420574188232,
191
+ "learning_rate": 5.9850000000000005e-05,
192
+ "loss": 0.4456,
193
  "step": 1200
194
  },
195
  {
196
+ "epoch": 1.0469011725293131,
197
+ "grad_norm": 2.3565518856048584,
198
+ "learning_rate": 6.235000000000001e-05,
199
+ "loss": 0.4664,
200
  "step": 1250
201
  },
202
  {
203
+ "epoch": 1.0887772194304857,
204
+ "grad_norm": 3.147794485092163,
205
+ "learning_rate": 6.485e-05,
206
+ "loss": 0.4681,
207
  "step": 1300
208
  },
209
  {
210
+ "epoch": 1.1306532663316582,
211
+ "grad_norm": 3.028409957885742,
212
+ "learning_rate": 6.735e-05,
213
+ "loss": 0.4663,
214
  "step": 1350
215
  },
216
  {
217
+ "epoch": 1.1725293132328307,
218
+ "grad_norm": 4.820915222167969,
219
+ "learning_rate": 6.985e-05,
220
+ "loss": 0.4493,
221
  "step": 1400
222
  },
223
  {
224
+ "epoch": 1.2144053601340032,
225
+ "grad_norm": 1.93254816532135,
226
+ "learning_rate": 7.235000000000001e-05,
227
+ "loss": 0.4413,
228
  "step": 1450
229
  },
230
  {
231
+ "epoch": 1.2562814070351758,
232
+ "grad_norm": 2.237015724182129,
233
+ "learning_rate": 7.485e-05,
234
+ "loss": 0.432,
235
  "step": 1500
236
  },
237
  {
238
+ "epoch": 1.2562814070351758,
239
+ "eval_loss": 0.39829039573669434,
240
+ "eval_runtime": 217.3782,
241
+ "eval_samples_per_second": 39.052,
242
+ "eval_steps_per_second": 4.885,
243
  "step": 1500
244
  },
245
  {
246
+ "epoch": 1.2981574539363483,
247
+ "grad_norm": 1.709120273590088,
248
+ "learning_rate": 7.735e-05,
249
+ "loss": 0.4361,
250
  "step": 1550
251
  },
252
  {
253
+ "epoch": 1.3400335008375208,
254
+ "grad_norm": 3.733328104019165,
255
+ "learning_rate": 7.985e-05,
256
+ "loss": 0.4357,
257
  "step": 1600
258
  },
259
  {
260
+ "epoch": 1.3819095477386933,
261
+ "grad_norm": 4.131324768066406,
262
+ "learning_rate": 8.235000000000001e-05,
263
+ "loss": 0.4445,
264
  "step": 1650
265
  },
266
  {
267
+ "epoch": 1.4237855946398659,
268
+ "grad_norm": 2.2893927097320557,
269
+ "learning_rate": 8.485e-05,
270
+ "loss": 0.4379,
271
  "step": 1700
272
  },
273
  {
274
+ "epoch": 1.4656616415410384,
275
+ "grad_norm": 2.1464734077453613,
276
+ "learning_rate": 8.735000000000001e-05,
277
+ "loss": 0.4393,
278
  "step": 1750
279
  },
280
  {
281
+ "epoch": 1.507537688442211,
282
+ "grad_norm": 2.01698637008667,
283
+ "learning_rate": 8.985e-05,
284
+ "loss": 0.4341,
285
  "step": 1800
286
  },
287
  {
288
+ "epoch": 1.5494137353433834,
289
+ "grad_norm": 2.259115219116211,
290
+ "learning_rate": 9.235000000000001e-05,
291
+ "loss": 0.4324,
292
  "step": 1850
293
  },
294
  {
295
+ "epoch": 1.591289782244556,
296
+ "grad_norm": 2.0072858333587646,
297
+ "learning_rate": 9.485e-05,
298
+ "loss": 0.4327,
299
  "step": 1900
300
  },
301
  {
302
+ "epoch": 1.6331658291457285,
303
+ "grad_norm": 2.060303211212158,
304
+ "learning_rate": 9.735000000000001e-05,
305
+ "loss": 0.4279,
306
  "step": 1950
307
  },
308
  {
309
+ "epoch": 1.675041876046901,
310
+ "grad_norm": 1.2846741676330566,
311
+ "learning_rate": 9.985000000000001e-05,
312
+ "loss": 0.429,
313
  "step": 2000
314
  },
315
  {
316
+ "epoch": 1.675041876046901,
317
+ "eval_loss": 0.3953319191932678,
318
+ "eval_runtime": 217.8108,
319
+ "eval_samples_per_second": 38.974,
320
+ "eval_steps_per_second": 4.876,
321
  "step": 2000
322
  },
323
  {
324
+ "epoch": 1.7169179229480735,
325
+ "grad_norm": 2.253840923309326,
326
+ "learning_rate": 9.953e-05,
327
+ "loss": 0.4264,
328
  "step": 2050
329
  },
330
  {
331
+ "epoch": 1.758793969849246,
332
+ "grad_norm": 1.3114299774169922,
333
+ "learning_rate": 9.903e-05,
334
+ "loss": 0.4355,
335
  "step": 2100
336
  },
337
  {
338
+ "epoch": 1.8006700167504186,
339
+ "grad_norm": 1.8178610801696777,
340
+ "learning_rate": 9.853e-05,
341
+ "loss": 0.4256,
342
  "step": 2150
343
  },
344
  {
345
+ "epoch": 1.8425460636515911,
346
+ "grad_norm": 2.1193735599517822,
347
+ "learning_rate": 9.803e-05,
348
+ "loss": 0.4305,
349
  "step": 2200
350
  },
351
  {
352
+ "epoch": 1.8844221105527639,
353
+ "grad_norm": 1.6871263980865479,
354
+ "learning_rate": 9.753e-05,
355
+ "loss": 0.4106,
356
  "step": 2250
357
  },
358
  {
359
+ "epoch": 1.9262981574539364,
360
+ "grad_norm": 1.2699921131134033,
361
+ "learning_rate": 9.703000000000001e-05,
362
+ "loss": 0.4156,
363
  "step": 2300
364
  },
365
  {
366
+ "epoch": 1.968174204355109,
367
+ "grad_norm": 1.7966769933700562,
368
+ "learning_rate": 9.653000000000001e-05,
369
+ "loss": 0.4257,
370
  "step": 2350
371
  },
372
  {
373
+ "epoch": 2.0100502512562812,
374
+ "grad_norm": 1.7838683128356934,
375
+ "learning_rate": 9.603000000000001e-05,
376
+ "loss": 0.4156,
377
  "step": 2400
378
  },
379
  {
380
+ "epoch": 2.0519262981574538,
381
+ "grad_norm": 1.4161527156829834,
382
+ "learning_rate": 9.553e-05,
383
+ "loss": 0.4096,
384
  "step": 2450
385
  },
386
  {
387
+ "epoch": 2.0938023450586263,
388
+ "grad_norm": 1.5864896774291992,
389
+ "learning_rate": 9.503e-05,
390
+ "loss": 0.4168,
391
  "step": 2500
392
  },
393
  {
394
+ "epoch": 2.0938023450586263,
395
+ "eval_loss": 0.3701198697090149,
396
+ "eval_runtime": 220.4399,
397
+ "eval_samples_per_second": 38.509,
398
+ "eval_steps_per_second": 4.818,
399
  "step": 2500
400
  },
401
  {
402
+ "epoch": 2.135678391959799,
403
+ "grad_norm": 1.2460856437683105,
404
+ "learning_rate": 9.453e-05,
405
+ "loss": 0.415,
406
  "step": 2550
407
  },
408
  {
409
+ "epoch": 2.1775544388609713,
410
+ "grad_norm": 1.994315266609192,
411
+ "learning_rate": 9.403e-05,
412
+ "loss": 0.4075,
413
  "step": 2600
414
  },
415
  {
416
+ "epoch": 2.219430485762144,
417
+ "grad_norm": 2.148324966430664,
418
+ "learning_rate": 9.353000000000001e-05,
419
+ "loss": 0.4077,
420
  "step": 2650
421
  },
422
  {
423
+ "epoch": 2.2613065326633164,
424
+ "grad_norm": 1.7242035865783691,
425
+ "learning_rate": 9.303000000000001e-05,
426
+ "loss": 0.4044,
427
  "step": 2700
428
  },
429
  {
430
+ "epoch": 2.3031825795644894,
431
+ "grad_norm": 1.912110447883606,
432
+ "learning_rate": 9.253000000000001e-05,
433
+ "loss": 0.4075,
434
  "step": 2750
435
  },
436
  {
437
+ "epoch": 2.3450586264656614,
438
+ "grad_norm": 1.3661189079284668,
439
+ "learning_rate": 9.203000000000001e-05,
440
+ "loss": 0.4041,
441
  "step": 2800
442
  },
443
  {
444
+ "epoch": 2.3869346733668344,
445
+ "grad_norm": 1.7083531618118286,
446
+ "learning_rate": 9.153e-05,
447
+ "loss": 0.396,
448
  "step": 2850
449
  },
450
  {
451
+ "epoch": 2.4288107202680065,
452
+ "grad_norm": 1.7829670906066895,
453
+ "learning_rate": 9.103e-05,
454
+ "loss": 0.4042,
455
  "step": 2900
456
  },
457
  {
458
+ "epoch": 2.4706867671691795,
459
+ "grad_norm": 2.188753366470337,
460
+ "learning_rate": 9.053e-05,
461
+ "loss": 0.3963,
462
  "step": 2950
463
  },
464
  {
465
+ "epoch": 2.5125628140703515,
466
+ "grad_norm": 1.2608036994934082,
467
+ "learning_rate": 9.003e-05,
468
+ "loss": 0.4021,
469
  "step": 3000
470
  },
471
  {
472
+ "epoch": 2.5125628140703515,
473
+ "eval_loss": 0.3613213300704956,
474
+ "eval_runtime": 214.7226,
475
+ "eval_samples_per_second": 39.535,
476
+ "eval_steps_per_second": 4.946,
477
  "step": 3000
478
  }
479
  ],
 
494
  "attributes": {}
495
  }
496
  },
497
+ "total_flos": 2.915701212952915e+16,
498
+ "train_batch_size": 32,
499
  "trial_name": null,
500
  "trial_params": null
501
  }
checkpoint-3000/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fa8f41edfc5db87ff9f21c8561c4e1b9e9e17c2b5ee34213317807a5e0e41371
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb8203a5324b57ae1a64f032a5185016d65cbc7738fd0978239bc5b2518d051b
3
  size 5304