alexue4 commited on
Commit
a917f64
1 Parent(s): 2678be8

End of training

Browse files
Files changed (4) hide show
  1. README.md +27 -27
  2. pytorch_model.bin +1 -1
  3. trainer_state.json +797 -779
  4. training_args.bin +2 -2
README.md CHANGED
@@ -15,9 +15,9 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [alexue4/text-normalization-ru-new](https://huggingface.co/alexue4/text-normalization-ru-new) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.0014
19
  - Mean Distance: 0
20
- - Max Distance: 3
21
 
22
  ## Model description
23
 
@@ -49,31 +49,31 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Mean Distance | Max Distance |
51
  |:-------------:|:-----:|:----:|:---------------:|:-------------:|:------------:|
52
- | 0.0368 | 1.0 | 68 | 0.0619 | 0 | 3 |
53
- | 0.0057 | 2.0 | 136 | 0.0346 | 0 | 3 |
54
- | 0.0113 | 3.0 | 204 | 0.0252 | 0 | 4 |
55
- | 0.0244 | 4.0 | 272 | 0.0111 | 0 | 3 |
56
- | 0.001 | 5.0 | 340 | 0.0065 | 0 | 3 |
57
- | 0.0021 | 6.0 | 408 | 0.0051 | 0 | 5 |
58
- | 0.0008 | 7.0 | 476 | 0.0077 | 0 | 3 |
59
- | 0.0008 | 8.0 | 544 | 0.0040 | 0 | 3 |
60
- | 0.0025 | 9.0 | 612 | 0.0029 | 0 | 2 |
61
- | 0.0012 | 10.0 | 680 | 0.0020 | 0 | 3 |
62
- | 0.0012 | 11.0 | 748 | 0.0010 | 0 | 3 |
63
- | 0.0006 | 12.0 | 816 | 0.0008 | 0 | 0 |
64
- | 0.0012 | 13.0 | 884 | 0.0005 | 0 | 0 |
65
- | 0.001 | 14.0 | 952 | 0.0007 | 0 | 3 |
66
- | 0.0009 | 15.0 | 1020 | 0.0014 | 0 | 3 |
67
- | 0.0005 | 16.0 | 1088 | 0.0015 | 0 | 3 |
68
- | 0.0009 | 17.0 | 1156 | 0.0014 | 0 | 3 |
69
- | 0.0006 | 18.0 | 1224 | 0.0013 | 0 | 3 |
70
- | 0.0012 | 19.0 | 1292 | 0.0014 | 0 | 3 |
71
- | 0.0007 | 20.0 | 1360 | 0.0014 | 0 | 3 |
72
- | 0.0012 | 21.0 | 1428 | 0.0013 | 0 | 3 |
73
- | 0.0013 | 22.0 | 1496 | 0.0014 | 0 | 3 |
74
- | 0.0008 | 23.0 | 1564 | 0.0014 | 0 | 3 |
75
- | 0.0005 | 24.0 | 1632 | 0.0014 | 0 | 3 |
76
- | 0.001 | 25.0 | 1700 | 0.0014 | 0 | 3 |
77
 
78
 
79
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [alexue4/text-normalization-ru-new](https://huggingface.co/alexue4/text-normalization-ru-new) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.0003
19
  - Mean Distance: 0
20
+ - Max Distance: 0
21
 
22
  ## Model description
23
 
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Mean Distance | Max Distance |
51
  |:-------------:|:-----:|:----:|:---------------:|:-------------:|:------------:|
52
+ | 0.0013 | 1.0 | 69 | 0.0028 | 0 | 2 |
53
+ | 0.0006 | 2.0 | 138 | 0.0026 | 0 | 3 |
54
+ | 0.0025 | 3.0 | 207 | 0.0039 | 0 | 3 |
55
+ | 0.0004 | 4.0 | 276 | 0.0037 | 0 | 3 |
56
+ | 0.0005 | 5.0 | 345 | 0.0091 | 0 | 3 |
57
+ | 0.0009 | 6.0 | 414 | 0.0006 | 0 | 0 |
58
+ | 0.0016 | 7.0 | 483 | 0.0003 | 0 | 0 |
59
+ | 0.0012 | 8.0 | 552 | 0.0111 | 0 | 5 |
60
+ | 0.0008 | 9.0 | 621 | 0.0004 | 0 | 0 |
61
+ | 0.0018 | 10.0 | 690 | 0.0003 | 0 | 0 |
62
+ | 0.0028 | 11.0 | 759 | 0.0003 | 0 | 0 |
63
+ | 0.0008 | 12.0 | 828 | 0.0003 | 0 | 0 |
64
+ | 0.001 | 13.0 | 897 | 0.0004 | 0 | 2 |
65
+ | 0.0026 | 14.0 | 966 | 0.0005 | 0 | 2 |
66
+ | 0.0015 | 15.0 | 1035 | 0.0007 | 0 | 3 |
67
+ | 0.0009 | 16.0 | 1104 | 0.0007 | 0 | 3 |
68
+ | 0.0014 | 17.0 | 1173 | 0.0003 | 0 | 0 |
69
+ | 0.001 | 18.0 | 1242 | 0.0004 | 0 | 0 |
70
+ | 0.0007 | 19.0 | 1311 | 0.0013 | 0 | 3 |
71
+ | 0.0013 | 20.0 | 1380 | 0.0013 | 0 | 3 |
72
+ | 0.0007 | 21.0 | 1449 | 0.0003 | 0 | 0 |
73
+ | 0.0016 | 22.0 | 1518 | 0.0003 | 0 | 0 |
74
+ | 0.0013 | 23.0 | 1587 | 0.0003 | 0 | 0 |
75
+ | 0.0004 | 24.0 | 1656 | 0.0003 | 0 | 0 |
76
+ | 0.001 | 25.0 | 1725 | 0.0003 | 0 | 0 |
77
 
78
 
79
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9c78ae41e27176e6da318232c2b38c680ba5bed653558b5af6c48e9134e6fe90
3
  size 258643461
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69a322d8df9333542c41863f9dfd626d1fa4f1b142f13cf8ac5477a1f46f3819
3
  size 258643461
trainer_state.json CHANGED
@@ -1,1412 +1,1430 @@
1
  {
2
- "best_metric": 0.0005441381363198161,
3
- "best_model_checkpoint": "text-normalization-ru-new\\checkpoint-884",
4
  "epoch": 25.0,
5
  "eval_steps": 500,
6
- "global_step": 1700,
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.01,
13
- "learning_rate": 5.882352941176471e-07,
14
- "loss": 0.0269,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.13,
19
- "learning_rate": 5.294117647058824e-06,
20
- "loss": 0.032,
21
  "step": 9
22
  },
23
  {
24
  "epoch": 0.26,
25
- "learning_rate": 1.0588235294117648e-05,
26
- "loss": 0.0142,
27
  "step": 18
28
  },
29
  {
30
- "epoch": 0.4,
31
- "learning_rate": 1.588235294117647e-05,
32
- "loss": 0.003,
33
  "step": 27
34
  },
35
  {
36
- "epoch": 0.53,
37
- "learning_rate": 2.1176470588235296e-05,
38
- "loss": 0.0185,
39
  "step": 36
40
  },
41
  {
42
- "epoch": 0.66,
43
- "learning_rate": 2.647058823529412e-05,
44
- "loss": 0.0431,
45
  "step": 45
46
  },
47
  {
48
- "epoch": 0.79,
49
- "learning_rate": 3.176470588235294e-05,
50
- "loss": 0.0182,
51
  "step": 54
52
  },
53
  {
54
- "epoch": 0.93,
55
- "learning_rate": 3.705882352941177e-05,
56
- "loss": 0.0368,
57
  "step": 63
58
  },
59
  {
60
  "epoch": 1.0,
61
- "eval_loss": 0.061875343322753906,
62
- "eval_max_distance": 3,
63
  "eval_mean_distance": 0,
64
- "eval_runtime": 1.7967,
65
- "eval_samples_per_second": 42.855,
66
- "eval_steps_per_second": 1.67,
67
- "step": 68
68
  },
69
  {
70
- "epoch": 1.06,
71
- "learning_rate": 4.235294117647059e-05,
72
- "loss": 0.0151,
73
  "step": 72
74
  },
75
  {
76
- "epoch": 1.19,
77
- "learning_rate": 4.7647058823529414e-05,
78
- "loss": 0.0193,
79
  "step": 81
80
  },
81
  {
82
- "epoch": 1.32,
83
- "learning_rate": 5.294117647058824e-05,
84
- "loss": 0.0076,
85
  "step": 90
86
  },
87
  {
88
- "epoch": 1.46,
89
- "learning_rate": 5.823529411764707e-05,
90
- "loss": 0.0061,
91
  "step": 99
92
  },
93
  {
94
- "epoch": 1.59,
95
- "learning_rate": 6.352941176470588e-05,
96
- "loss": 0.0111,
97
  "step": 108
98
  },
99
  {
100
- "epoch": 1.72,
101
- "learning_rate": 6.882352941176471e-05,
102
- "loss": 0.0314,
103
  "step": 117
104
  },
105
  {
106
- "epoch": 1.85,
107
- "learning_rate": 7.411764705882354e-05,
108
- "loss": 0.0253,
109
  "step": 126
110
  },
111
  {
112
- "epoch": 1.99,
113
- "learning_rate": 7.941176470588235e-05,
114
- "loss": 0.0057,
115
  "step": 135
116
  },
117
  {
118
  "epoch": 2.0,
119
- "eval_loss": 0.034646954387426376,
120
  "eval_max_distance": 3,
121
  "eval_mean_distance": 0,
122
- "eval_runtime": 1.8378,
123
- "eval_samples_per_second": 41.898,
124
- "eval_steps_per_second": 1.632,
125
- "step": 136
126
  },
127
  {
128
- "epoch": 2.12,
129
- "learning_rate": 8.470588235294118e-05,
130
- "loss": 0.0184,
131
  "step": 144
132
  },
133
  {
134
- "epoch": 2.25,
135
- "learning_rate": 9e-05,
136
- "loss": 0.0062,
137
  "step": 153
138
  },
139
  {
140
- "epoch": 2.38,
141
- "learning_rate": 9.529411764705883e-05,
142
- "loss": 0.0055,
143
  "step": 162
144
  },
145
  {
146
- "epoch": 2.51,
147
- "learning_rate": 9.993464052287582e-05,
148
- "loss": 0.0144,
149
  "step": 171
150
  },
151
  {
152
- "epoch": 2.65,
153
- "learning_rate": 9.934640522875818e-05,
154
- "loss": 0.0006,
155
  "step": 180
156
  },
157
  {
158
- "epoch": 2.78,
159
- "learning_rate": 9.875816993464053e-05,
160
- "loss": 0.0101,
161
  "step": 189
162
  },
163
  {
164
- "epoch": 2.91,
165
- "learning_rate": 9.816993464052288e-05,
166
- "loss": 0.0113,
167
  "step": 198
168
  },
169
  {
170
  "epoch": 3.0,
171
- "eval_loss": 0.025189895182847977,
172
- "eval_max_distance": 4,
173
- "eval_mean_distance": 0,
174
- "eval_runtime": 1.7969,
175
- "eval_samples_per_second": 42.851,
176
- "eval_steps_per_second": 1.67,
177
- "step": 204
178
  },
179
  {
180
- "epoch": 3.04,
181
- "learning_rate": 9.758169934640524e-05,
182
- "loss": 0.0007,
 
 
 
 
183
  "step": 207
184
  },
185
  {
186
- "epoch": 3.18,
187
- "learning_rate": 9.699346405228759e-05,
188
- "loss": 0.0026,
189
  "step": 216
190
  },
191
  {
192
- "epoch": 3.31,
193
- "learning_rate": 9.640522875816994e-05,
194
- "loss": 0.0023,
195
  "step": 225
196
  },
197
  {
198
- "epoch": 3.44,
199
- "learning_rate": 9.58169934640523e-05,
200
- "loss": 0.0012,
201
  "step": 234
202
  },
203
  {
204
- "epoch": 3.57,
205
- "learning_rate": 9.522875816993463e-05,
206
- "loss": 0.0029,
207
  "step": 243
208
  },
209
  {
210
- "epoch": 3.71,
211
- "learning_rate": 9.4640522875817e-05,
212
- "loss": 0.0028,
213
  "step": 252
214
  },
215
  {
216
- "epoch": 3.84,
217
- "learning_rate": 9.405228758169935e-05,
218
- "loss": 0.0049,
219
  "step": 261
220
  },
221
  {
222
- "epoch": 3.97,
223
- "learning_rate": 9.34640522875817e-05,
224
- "loss": 0.0244,
225
  "step": 270
226
  },
227
  {
228
  "epoch": 4.0,
229
- "eval_loss": 0.011130807921290398,
230
  "eval_max_distance": 3,
231
  "eval_mean_distance": 0,
232
- "eval_runtime": 1.9031,
233
- "eval_samples_per_second": 40.459,
234
- "eval_steps_per_second": 1.576,
235
- "step": 272
236
  },
237
  {
238
- "epoch": 4.1,
239
- "learning_rate": 9.287581699346406e-05,
240
- "loss": 0.0163,
241
  "step": 279
242
  },
243
  {
244
- "epoch": 4.24,
245
- "learning_rate": 9.228758169934641e-05,
246
- "loss": 0.0035,
247
  "step": 288
248
  },
249
  {
250
- "epoch": 4.37,
251
- "learning_rate": 9.169934640522877e-05,
252
- "loss": 0.0011,
253
  "step": 297
254
  },
255
  {
256
- "epoch": 4.5,
257
- "learning_rate": 9.111111111111112e-05,
258
- "loss": 0.0127,
259
  "step": 306
260
  },
261
  {
262
- "epoch": 4.63,
263
- "learning_rate": 9.052287581699347e-05,
264
- "loss": 0.0017,
265
  "step": 315
266
  },
267
  {
268
- "epoch": 4.76,
269
- "learning_rate": 8.993464052287582e-05,
270
- "loss": 0.0018,
271
  "step": 324
272
  },
273
  {
274
- "epoch": 4.9,
275
- "learning_rate": 8.934640522875818e-05,
276
- "loss": 0.001,
277
  "step": 333
278
  },
 
 
 
 
 
 
279
  {
280
  "epoch": 5.0,
281
- "eval_loss": 0.006467848084867001,
282
  "eval_max_distance": 3,
283
  "eval_mean_distance": 0,
284
- "eval_runtime": 1.7867,
285
- "eval_samples_per_second": 43.096,
286
- "eval_steps_per_second": 1.679,
287
- "step": 340
288
  },
289
  {
290
- "epoch": 5.03,
291
- "learning_rate": 8.875816993464052e-05,
292
- "loss": 0.0136,
293
- "step": 342
294
- },
295
- {
296
- "epoch": 5.16,
297
- "learning_rate": 8.816993464052288e-05,
298
- "loss": 0.0054,
299
  "step": 351
300
  },
301
  {
302
- "epoch": 5.29,
303
- "learning_rate": 8.758169934640524e-05,
304
- "loss": 0.0016,
305
  "step": 360
306
  },
307
  {
308
- "epoch": 5.43,
309
- "learning_rate": 8.699346405228759e-05,
310
- "loss": 0.0093,
311
  "step": 369
312
  },
313
  {
314
- "epoch": 5.56,
315
- "learning_rate": 8.640522875816994e-05,
316
- "loss": 0.0058,
317
  "step": 378
318
  },
319
  {
320
- "epoch": 5.69,
321
- "learning_rate": 8.581699346405228e-05,
322
- "loss": 0.0016,
323
  "step": 387
324
  },
325
  {
326
- "epoch": 5.82,
327
- "learning_rate": 8.522875816993465e-05,
328
- "loss": 0.0007,
329
  "step": 396
330
  },
331
  {
332
- "epoch": 5.96,
333
- "learning_rate": 8.4640522875817e-05,
334
- "loss": 0.0021,
335
  "step": 405
336
  },
337
  {
338
  "epoch": 6.0,
339
- "eval_loss": 0.0050986092537641525,
340
- "eval_max_distance": 5,
341
- "eval_mean_distance": 0,
342
- "eval_runtime": 1.8684,
343
- "eval_samples_per_second": 41.212,
344
- "eval_steps_per_second": 1.606,
345
- "step": 408
346
  },
347
  {
348
- "epoch": 6.09,
349
- "learning_rate": 8.405228758169935e-05,
350
- "loss": 0.0008,
 
 
 
 
351
  "step": 414
352
  },
353
  {
354
- "epoch": 6.22,
355
- "learning_rate": 8.346405228758171e-05,
356
- "loss": 0.0022,
357
  "step": 423
358
  },
359
  {
360
- "epoch": 6.35,
361
- "learning_rate": 8.287581699346406e-05,
362
- "loss": 0.0071,
363
  "step": 432
364
  },
365
  {
366
- "epoch": 6.49,
367
- "learning_rate": 8.22875816993464e-05,
368
- "loss": 0.004,
369
  "step": 441
370
  },
371
  {
372
- "epoch": 6.62,
373
- "learning_rate": 8.169934640522877e-05,
374
- "loss": 0.0004,
375
  "step": 450
376
  },
377
  {
378
- "epoch": 6.75,
379
- "learning_rate": 8.111111111111112e-05,
380
- "loss": 0.0051,
381
  "step": 459
382
  },
383
  {
384
- "epoch": 6.88,
385
- "learning_rate": 8.052287581699347e-05,
386
  "loss": 0.0008,
387
  "step": 468
388
  },
389
  {
390
- "epoch": 7.0,
391
- "eval_loss": 0.007741495501250029,
392
- "eval_max_distance": 3,
393
- "eval_mean_distance": 0,
394
- "eval_runtime": 1.8063,
395
- "eval_samples_per_second": 42.629,
396
- "eval_steps_per_second": 1.661,
397
- "step": 476
398
  },
399
  {
400
- "epoch": 7.01,
401
- "learning_rate": 7.993464052287583e-05,
402
- "loss": 0.0071,
403
- "step": 477
 
 
 
 
404
  },
405
  {
406
- "epoch": 7.15,
407
- "learning_rate": 7.934640522875816e-05,
408
- "loss": 0.004,
409
  "step": 486
410
  },
411
  {
412
- "epoch": 7.28,
413
- "learning_rate": 7.875816993464053e-05,
414
- "loss": 0.0014,
415
  "step": 495
416
  },
417
  {
418
- "epoch": 7.41,
419
- "learning_rate": 7.816993464052288e-05,
420
- "loss": 0.003,
421
  "step": 504
422
  },
423
  {
424
- "epoch": 7.54,
425
- "learning_rate": 7.758169934640524e-05,
426
- "loss": 0.0008,
427
  "step": 513
428
  },
429
  {
430
- "epoch": 7.68,
431
- "learning_rate": 7.699346405228759e-05,
432
- "loss": 0.001,
433
  "step": 522
434
  },
435
  {
436
- "epoch": 7.81,
437
- "learning_rate": 7.640522875816993e-05,
438
- "loss": 0.0016,
439
  "step": 531
440
  },
441
  {
442
- "epoch": 7.94,
443
- "learning_rate": 7.581699346405228e-05,
444
- "loss": 0.0008,
445
  "step": 540
446
  },
 
 
 
 
 
 
447
  {
448
  "epoch": 8.0,
449
- "eval_loss": 0.004006722941994667,
450
- "eval_max_distance": 3,
451
  "eval_mean_distance": 0,
452
- "eval_runtime": 1.7939,
453
- "eval_samples_per_second": 42.923,
454
- "eval_steps_per_second": 1.672,
455
- "step": 544
456
  },
457
  {
458
- "epoch": 8.07,
459
- "learning_rate": 7.522875816993465e-05,
460
  "loss": 0.0008,
461
- "step": 549
462
- },
463
- {
464
- "epoch": 8.21,
465
- "learning_rate": 7.4640522875817e-05,
466
- "loss": 0.0015,
467
  "step": 558
468
  },
469
  {
470
- "epoch": 8.34,
471
- "learning_rate": 7.405228758169935e-05,
472
- "loss": 0.0018,
473
  "step": 567
474
  },
475
  {
476
- "epoch": 8.47,
477
- "learning_rate": 7.346405228758171e-05,
478
- "loss": 0.0015,
479
  "step": 576
480
  },
481
  {
482
- "epoch": 8.6,
483
- "learning_rate": 7.287581699346405e-05,
484
- "loss": 0.0028,
485
  "step": 585
486
  },
487
  {
488
- "epoch": 8.74,
489
- "learning_rate": 7.228758169934641e-05,
490
- "loss": 0.0017,
491
  "step": 594
492
  },
493
  {
494
- "epoch": 8.87,
495
- "learning_rate": 7.169934640522877e-05,
496
- "loss": 0.001,
497
  "step": 603
498
  },
499
  {
500
- "epoch": 9.0,
501
- "learning_rate": 7.111111111111112e-05,
502
- "loss": 0.0025,
503
  "step": 612
504
  },
505
  {
506
  "epoch": 9.0,
507
- "eval_loss": 0.002946252468973398,
508
- "eval_max_distance": 2,
509
- "eval_mean_distance": 0,
510
- "eval_runtime": 1.778,
511
- "eval_samples_per_second": 43.306,
512
- "eval_steps_per_second": 1.687,
513
- "step": 612
514
  },
515
  {
516
- "epoch": 9.13,
517
- "learning_rate": 7.052287581699347e-05,
518
- "loss": 0.0008,
 
 
 
 
519
  "step": 621
520
  },
521
  {
522
- "epoch": 9.26,
523
- "learning_rate": 6.993464052287581e-05,
524
- "loss": 0.0014,
525
  "step": 630
526
  },
527
  {
528
- "epoch": 9.4,
529
- "learning_rate": 6.934640522875817e-05,
530
- "loss": 0.0036,
531
  "step": 639
532
  },
533
  {
534
- "epoch": 9.53,
535
- "learning_rate": 6.875816993464053e-05,
536
- "loss": 0.0022,
537
  "step": 648
538
  },
539
  {
540
- "epoch": 9.66,
541
- "learning_rate": 6.816993464052288e-05,
542
- "loss": 0.0006,
543
  "step": 657
544
  },
545
  {
546
- "epoch": 9.79,
547
- "learning_rate": 6.758169934640524e-05,
548
- "loss": 0.0034,
549
  "step": 666
550
  },
551
  {
552
- "epoch": 9.93,
553
- "learning_rate": 6.699346405228758e-05,
554
- "loss": 0.0012,
555
  "step": 675
556
  },
557
  {
558
- "epoch": 10.0,
559
- "eval_loss": 0.00195131148211658,
560
- "eval_max_distance": 3,
561
- "eval_mean_distance": 0,
562
- "eval_runtime": 1.7911,
563
- "eval_samples_per_second": 42.99,
564
- "eval_steps_per_second": 1.675,
565
- "step": 680
566
  },
567
  {
568
- "epoch": 10.06,
569
- "learning_rate": 6.640522875816993e-05,
570
- "loss": 0.0003,
571
- "step": 684
 
 
 
 
572
  },
573
  {
574
- "epoch": 10.19,
575
- "learning_rate": 6.581699346405228e-05,
576
- "loss": 0.0014,
577
  "step": 693
578
  },
579
  {
580
- "epoch": 10.32,
581
- "learning_rate": 6.522875816993465e-05,
582
- "loss": 0.0013,
583
  "step": 702
584
  },
585
  {
586
- "epoch": 10.46,
587
- "learning_rate": 6.4640522875817e-05,
588
- "loss": 0.0021,
589
  "step": 711
590
  },
591
  {
592
- "epoch": 10.59,
593
- "learning_rate": 6.405228758169934e-05,
594
- "loss": 0.0035,
595
  "step": 720
596
  },
597
  {
598
- "epoch": 10.72,
599
- "learning_rate": 6.34640522875817e-05,
600
- "loss": 0.0011,
601
  "step": 729
602
  },
603
  {
604
- "epoch": 10.85,
605
- "learning_rate": 6.287581699346405e-05,
606
- "loss": 0.0005,
607
  "step": 738
608
  },
609
  {
610
- "epoch": 10.99,
611
- "learning_rate": 6.228758169934641e-05,
612
- "loss": 0.0012,
613
  "step": 747
614
  },
615
  {
616
- "epoch": 11.0,
617
- "eval_loss": 0.0009844042360782623,
618
- "eval_max_distance": 3,
619
- "eval_mean_distance": 0,
620
- "eval_runtime": 1.7912,
621
- "eval_samples_per_second": 42.989,
622
- "eval_steps_per_second": 1.675,
623
- "step": 748
624
  },
625
  {
626
- "epoch": 11.12,
627
- "learning_rate": 6.169934640522877e-05,
628
- "loss": 0.0068,
629
- "step": 756
 
 
 
 
630
  },
631
  {
632
- "epoch": 11.25,
633
- "learning_rate": 6.111111111111112e-05,
634
- "loss": 0.001,
635
  "step": 765
636
  },
637
  {
638
- "epoch": 11.38,
639
- "learning_rate": 6.052287581699346e-05,
640
- "loss": 0.0009,
641
  "step": 774
642
  },
643
  {
644
- "epoch": 11.51,
645
- "learning_rate": 5.993464052287582e-05,
646
- "loss": 0.0002,
647
  "step": 783
648
  },
649
  {
650
- "epoch": 11.65,
651
- "learning_rate": 5.934640522875817e-05,
652
- "loss": 0.0014,
653
  "step": 792
654
  },
655
  {
656
- "epoch": 11.78,
657
- "learning_rate": 5.8758169934640526e-05,
658
- "loss": 0.0028,
659
  "step": 801
660
  },
661
  {
662
- "epoch": 11.91,
663
- "learning_rate": 5.8169934640522886e-05,
664
- "loss": 0.0006,
665
  "step": 810
666
  },
667
  {
668
- "epoch": 12.0,
669
- "eval_loss": 0.0007958057103678584,
670
- "eval_max_distance": 0,
671
- "eval_mean_distance": 0,
672
- "eval_runtime": 1.8244,
673
- "eval_samples_per_second": 42.205,
674
- "eval_steps_per_second": 1.644,
675
- "step": 816
676
  },
677
  {
678
- "epoch": 12.04,
679
- "learning_rate": 5.7581699346405225e-05,
680
- "loss": 0.0028,
681
- "step": 819
682
  },
683
  {
684
- "epoch": 12.18,
685
- "learning_rate": 5.6993464052287585e-05,
686
- "loss": 0.0015,
 
 
 
 
687
  "step": 828
688
  },
689
  {
690
- "epoch": 12.31,
691
- "learning_rate": 5.640522875816994e-05,
692
- "loss": 0.0009,
693
  "step": 837
694
  },
695
  {
696
- "epoch": 12.44,
697
- "learning_rate": 5.581699346405229e-05,
698
- "loss": 0.0018,
699
  "step": 846
700
  },
701
  {
702
- "epoch": 12.57,
703
- "learning_rate": 5.5228758169934644e-05,
704
- "loss": 0.0008,
705
  "step": 855
706
  },
707
  {
708
- "epoch": 12.71,
709
- "learning_rate": 5.464052287581699e-05,
710
- "loss": 0.0004,
711
  "step": 864
712
  },
713
  {
714
- "epoch": 12.84,
715
- "learning_rate": 5.405228758169934e-05,
716
- "loss": 0.002,
717
  "step": 873
718
  },
719
  {
720
- "epoch": 12.97,
721
- "learning_rate": 5.34640522875817e-05,
722
- "loss": 0.0012,
723
  "step": 882
724
  },
725
  {
726
- "epoch": 13.0,
727
- "eval_loss": 0.0005441381363198161,
728
- "eval_max_distance": 0,
729
- "eval_mean_distance": 0,
730
- "eval_runtime": 1.7816,
731
- "eval_samples_per_second": 43.219,
732
- "eval_steps_per_second": 1.684,
733
- "step": 884
734
  },
735
  {
736
- "epoch": 13.1,
737
- "learning_rate": 5.2875816993464056e-05,
738
- "loss": 0.0013,
739
- "step": 891
 
 
 
 
740
  },
741
  {
742
- "epoch": 13.24,
743
- "learning_rate": 5.228758169934641e-05,
744
- "loss": 0.0004,
745
  "step": 900
746
  },
747
  {
748
- "epoch": 13.37,
749
- "learning_rate": 5.169934640522877e-05,
750
  "loss": 0.0008,
751
  "step": 909
752
  },
753
  {
754
- "epoch": 13.5,
755
- "learning_rate": 5.111111111111111e-05,
756
- "loss": 0.0018,
757
  "step": 918
758
  },
759
  {
760
- "epoch": 13.63,
761
- "learning_rate": 5.052287581699347e-05,
762
- "loss": 0.0026,
763
  "step": 927
764
  },
765
  {
766
- "epoch": 13.76,
767
- "learning_rate": 4.993464052287582e-05,
768
- "loss": 0.0025,
769
  "step": 936
770
  },
771
  {
772
- "epoch": 13.9,
773
- "learning_rate": 4.9346405228758174e-05,
774
- "loss": 0.001,
775
  "step": 945
776
  },
777
  {
778
- "epoch": 14.0,
779
- "eval_loss": 0.0007363850018009543,
780
- "eval_max_distance": 3,
781
- "eval_mean_distance": 0,
782
- "eval_runtime": 1.7798,
783
- "eval_samples_per_second": 43.263,
784
- "eval_steps_per_second": 1.686,
785
- "step": 952
786
  },
787
  {
788
- "epoch": 14.03,
789
- "learning_rate": 4.875816993464053e-05,
790
- "loss": 0.0012,
791
- "step": 954
792
  },
793
  {
794
- "epoch": 14.16,
795
- "learning_rate": 4.816993464052288e-05,
796
- "loss": 0.0006,
797
- "step": 963
 
 
 
 
798
  },
799
  {
800
- "epoch": 14.29,
801
- "learning_rate": 4.7581699346405226e-05,
802
- "loss": 0.0022,
803
  "step": 972
804
  },
805
  {
806
- "epoch": 14.43,
807
- "learning_rate": 4.6993464052287586e-05,
808
- "loss": 0.001,
809
  "step": 981
810
  },
811
  {
812
- "epoch": 14.56,
813
- "learning_rate": 4.640522875816994e-05,
814
- "loss": 0.0015,
815
  "step": 990
816
  },
817
  {
818
- "epoch": 14.69,
819
- "learning_rate": 4.5816993464052285e-05,
820
- "loss": 0.0015,
821
  "step": 999
822
  },
823
  {
824
- "epoch": 14.82,
825
- "learning_rate": 4.5228758169934645e-05,
826
- "loss": 0.0013,
827
  "step": 1008
828
  },
829
  {
830
- "epoch": 14.96,
831
- "learning_rate": 4.4640522875817e-05,
832
- "loss": 0.0009,
833
  "step": 1017
834
  },
835
  {
836
- "epoch": 15.0,
837
- "eval_loss": 0.0013804116751998663,
838
- "eval_max_distance": 3,
839
- "eval_mean_distance": 0,
840
- "eval_runtime": 1.7896,
841
- "eval_samples_per_second": 43.027,
842
- "eval_steps_per_second": 1.676,
843
- "step": 1020
844
  },
845
  {
846
- "epoch": 15.09,
847
- "learning_rate": 4.405228758169935e-05,
848
- "loss": 0.0013,
849
- "step": 1026
850
  },
851
  {
852
- "epoch": 15.22,
853
- "learning_rate": 4.3464052287581704e-05,
854
- "loss": 0.0011,
 
 
 
 
855
  "step": 1035
856
  },
857
  {
858
- "epoch": 15.35,
859
- "learning_rate": 4.287581699346405e-05,
860
- "loss": 0.0024,
861
  "step": 1044
862
  },
863
  {
864
- "epoch": 15.49,
865
- "learning_rate": 4.228758169934641e-05,
866
- "loss": 0.0006,
867
  "step": 1053
868
  },
869
  {
870
- "epoch": 15.62,
871
- "learning_rate": 4.169934640522876e-05,
872
- "loss": 0.0013,
873
  "step": 1062
874
  },
875
  {
876
- "epoch": 15.75,
877
- "learning_rate": 4.111111111111111e-05,
878
- "loss": 0.0012,
879
  "step": 1071
880
  },
881
  {
882
- "epoch": 15.88,
883
- "learning_rate": 4.052287581699347e-05,
884
- "loss": 0.0005,
885
  "step": 1080
886
  },
887
  {
888
- "epoch": 16.0,
889
- "eval_loss": 0.0015163180651143193,
890
- "eval_max_distance": 3,
891
- "eval_mean_distance": 0,
892
- "eval_runtime": 1.7783,
893
- "eval_samples_per_second": 43.3,
894
- "eval_steps_per_second": 1.687,
895
- "step": 1088
896
- },
897
- {
898
- "epoch": 16.01,
899
- "learning_rate": 3.993464052287582e-05,
900
- "loss": 0.0012,
901
  "step": 1089
902
  },
903
  {
904
- "epoch": 16.15,
905
- "learning_rate": 3.934640522875817e-05,
906
- "loss": 0.0006,
907
  "step": 1098
908
  },
909
  {
910
- "epoch": 16.28,
911
- "learning_rate": 3.875816993464053e-05,
912
- "loss": 0.0018,
 
 
 
 
 
 
 
 
 
 
913
  "step": 1107
914
  },
915
  {
916
- "epoch": 16.41,
917
- "learning_rate": 3.8169934640522874e-05,
918
- "loss": 0.0008,
919
  "step": 1116
920
  },
921
  {
922
- "epoch": 16.54,
923
- "learning_rate": 3.758169934640523e-05,
924
  "loss": 0.0006,
925
  "step": 1125
926
  },
927
  {
928
- "epoch": 16.68,
929
- "learning_rate": 3.699346405228759e-05,
930
- "loss": 0.0007,
931
  "step": 1134
932
  },
933
  {
934
- "epoch": 16.81,
935
- "learning_rate": 3.640522875816993e-05,
936
- "loss": 0.0013,
937
  "step": 1143
938
  },
939
  {
940
- "epoch": 16.94,
941
- "learning_rate": 3.581699346405229e-05,
942
- "loss": 0.0009,
943
  "step": 1152
944
  },
945
  {
946
- "epoch": 17.0,
947
- "eval_loss": 0.0014126226305961609,
948
- "eval_max_distance": 3,
949
- "eval_mean_distance": 0,
950
- "eval_runtime": 1.7701,
951
- "eval_samples_per_second": 43.501,
952
- "eval_steps_per_second": 1.695,
953
- "step": 1156
954
- },
955
- {
956
- "epoch": 17.07,
957
- "learning_rate": 3.5228758169934646e-05,
958
- "loss": 0.0016,
959
  "step": 1161
960
  },
961
  {
962
- "epoch": 17.21,
963
- "learning_rate": 3.464052287581699e-05,
964
- "loss": 0.0008,
965
  "step": 1170
966
  },
967
  {
968
- "epoch": 17.34,
969
- "learning_rate": 3.405228758169935e-05,
970
- "loss": 0.001,
 
 
 
 
 
 
 
 
 
 
971
  "step": 1179
972
  },
973
  {
974
- "epoch": 17.47,
975
- "learning_rate": 3.34640522875817e-05,
976
- "loss": 0.0006,
977
  "step": 1188
978
  },
979
  {
980
- "epoch": 17.6,
981
- "learning_rate": 3.287581699346405e-05,
982
- "loss": 0.0025,
983
  "step": 1197
984
  },
985
  {
986
- "epoch": 17.74,
987
- "learning_rate": 3.228758169934641e-05,
988
- "loss": 0.0011,
989
  "step": 1206
990
  },
991
  {
992
- "epoch": 17.87,
993
- "learning_rate": 3.169934640522876e-05,
994
- "loss": 0.0008,
995
  "step": 1215
996
  },
997
  {
998
- "epoch": 18.0,
999
- "learning_rate": 3.111111111111111e-05,
1000
  "loss": 0.0006,
1001
  "step": 1224
1002
  },
1003
  {
1004
- "epoch": 18.0,
1005
- "eval_loss": 0.0012967990478500724,
1006
- "eval_max_distance": 3,
1007
- "eval_mean_distance": 0,
1008
- "eval_runtime": 1.7808,
1009
- "eval_samples_per_second": 43.238,
1010
- "eval_steps_per_second": 1.685,
1011
- "step": 1224
1012
  },
1013
  {
1014
- "epoch": 18.13,
1015
- "learning_rate": 3.052287581699347e-05,
1016
- "loss": 0.0011,
1017
- "step": 1233
1018
  },
1019
  {
1020
- "epoch": 18.26,
1021
- "learning_rate": 2.9934640522875816e-05,
1022
- "loss": 0.0009,
 
 
 
 
1023
  "step": 1242
1024
  },
1025
  {
1026
- "epoch": 18.4,
1027
- "learning_rate": 2.9346405228758172e-05,
1028
- "loss": 0.0009,
1029
  "step": 1251
1030
  },
1031
  {
1032
- "epoch": 18.53,
1033
- "learning_rate": 2.8758169934640522e-05,
1034
- "loss": 0.0008,
1035
  "step": 1260
1036
  },
1037
  {
1038
- "epoch": 18.66,
1039
- "learning_rate": 2.8169934640522878e-05,
1040
- "loss": 0.0005,
1041
  "step": 1269
1042
  },
1043
  {
1044
- "epoch": 18.79,
1045
- "learning_rate": 2.758169934640523e-05,
1046
- "loss": 0.001,
1047
  "step": 1278
1048
  },
1049
  {
1050
- "epoch": 18.93,
1051
- "learning_rate": 2.699346405228758e-05,
1052
- "loss": 0.0012,
1053
  "step": 1287
1054
  },
1055
  {
1056
- "epoch": 19.0,
1057
- "eval_loss": 0.001441239845007658,
1058
- "eval_max_distance": 3,
1059
- "eval_mean_distance": 0,
1060
- "eval_runtime": 1.824,
1061
- "eval_samples_per_second": 42.215,
1062
- "eval_steps_per_second": 1.645,
1063
- "step": 1292
1064
- },
1065
- {
1066
- "epoch": 19.06,
1067
- "learning_rate": 2.6405228758169937e-05,
1068
- "loss": 0.0008,
1069
  "step": 1296
1070
  },
1071
  {
1072
- "epoch": 19.19,
1073
- "learning_rate": 2.581699346405229e-05,
1074
- "loss": 0.0009,
1075
  "step": 1305
1076
  },
1077
  {
1078
- "epoch": 19.32,
1079
- "learning_rate": 2.522875816993464e-05,
1080
- "loss": 0.0009,
 
 
 
 
 
 
 
 
 
 
1081
  "step": 1314
1082
  },
1083
  {
1084
- "epoch": 19.46,
1085
- "learning_rate": 2.4640522875816996e-05,
1086
- "loss": 0.0002,
1087
  "step": 1323
1088
  },
1089
  {
1090
- "epoch": 19.59,
1091
- "learning_rate": 2.405228758169935e-05,
1092
  "loss": 0.0006,
1093
  "step": 1332
1094
  },
1095
  {
1096
- "epoch": 19.72,
1097
- "learning_rate": 2.34640522875817e-05,
1098
  "loss": 0.0004,
1099
  "step": 1341
1100
  },
1101
  {
1102
- "epoch": 19.85,
1103
- "learning_rate": 2.2875816993464052e-05,
1104
- "loss": 0.0006,
1105
  "step": 1350
1106
  },
1107
  {
1108
- "epoch": 19.99,
1109
- "learning_rate": 2.2287581699346408e-05,
1110
- "loss": 0.0007,
1111
  "step": 1359
1112
  },
1113
  {
1114
- "epoch": 20.0,
1115
- "eval_loss": 0.0013804284390062094,
1116
- "eval_max_distance": 3,
1117
- "eval_mean_distance": 0,
1118
- "eval_runtime": 1.9155,
1119
- "eval_samples_per_second": 40.198,
1120
- "eval_steps_per_second": 1.566,
1121
- "step": 1360
1122
  },
1123
  {
1124
- "epoch": 20.12,
1125
- "learning_rate": 2.169934640522876e-05,
1126
  "loss": 0.0013,
1127
- "step": 1368
1128
  },
1129
  {
1130
- "epoch": 20.25,
1131
- "learning_rate": 2.111111111111111e-05,
1132
- "loss": 0.0014,
1133
- "step": 1377
 
 
 
 
1134
  },
1135
  {
1136
- "epoch": 20.38,
1137
- "learning_rate": 2.0522875816993464e-05,
1138
- "loss": 0.0001,
1139
  "step": 1386
1140
  },
1141
  {
1142
- "epoch": 20.51,
1143
- "learning_rate": 1.993464052287582e-05,
1144
- "loss": 0.0015,
1145
  "step": 1395
1146
  },
1147
  {
1148
- "epoch": 20.65,
1149
- "learning_rate": 1.9346405228758173e-05,
1150
- "loss": 0.0008,
1151
  "step": 1404
1152
  },
1153
  {
1154
- "epoch": 20.78,
1155
- "learning_rate": 1.8758169934640523e-05,
1156
  "loss": 0.0006,
1157
  "step": 1413
1158
  },
1159
  {
1160
- "epoch": 20.91,
1161
- "learning_rate": 1.8169934640522876e-05,
1162
- "loss": 0.0012,
1163
  "step": 1422
1164
  },
1165
  {
1166
- "epoch": 21.0,
1167
- "eval_loss": 0.0013173692859709263,
1168
- "eval_max_distance": 3,
1169
- "eval_mean_distance": 0,
1170
- "eval_runtime": 1.8242,
1171
- "eval_samples_per_second": 42.21,
1172
- "eval_steps_per_second": 1.645,
1173
- "step": 1428
1174
- },
1175
- {
1176
- "epoch": 21.04,
1177
- "learning_rate": 1.7581699346405232e-05,
1178
- "loss": 0.0027,
1179
  "step": 1431
1180
  },
1181
  {
1182
- "epoch": 21.18,
1183
- "learning_rate": 1.6993464052287582e-05,
1184
- "loss": 0.0008,
1185
  "step": 1440
1186
  },
1187
  {
1188
- "epoch": 21.31,
1189
- "learning_rate": 1.6405228758169935e-05,
1190
- "loss": 0.0006,
1191
  "step": 1449
1192
  },
1193
  {
1194
- "epoch": 21.44,
1195
- "learning_rate": 1.5816993464052288e-05,
1196
- "loss": 0.0002,
 
 
 
 
 
 
 
 
 
 
1197
  "step": 1458
1198
  },
1199
  {
1200
- "epoch": 21.57,
1201
- "learning_rate": 1.522875816993464e-05,
1202
- "loss": 0.001,
1203
  "step": 1467
1204
  },
1205
  {
1206
- "epoch": 21.71,
1207
- "learning_rate": 1.4640522875816995e-05,
1208
- "loss": 0.0012,
1209
  "step": 1476
1210
  },
1211
  {
1212
- "epoch": 21.84,
1213
- "learning_rate": 1.4052287581699347e-05,
1214
- "loss": 0.0008,
1215
  "step": 1485
1216
  },
1217
  {
1218
- "epoch": 21.97,
1219
- "learning_rate": 1.34640522875817e-05,
1220
- "loss": 0.0013,
1221
  "step": 1494
1222
  },
1223
  {
1224
- "epoch": 22.0,
1225
- "eval_loss": 0.001370716723613441,
1226
- "eval_max_distance": 3,
1227
- "eval_mean_distance": 0,
1228
- "eval_runtime": 1.8261,
1229
- "eval_samples_per_second": 42.166,
1230
- "eval_steps_per_second": 1.643,
1231
- "step": 1496
1232
- },
1233
- {
1234
- "epoch": 22.1,
1235
- "learning_rate": 1.2875816993464051e-05,
1236
- "loss": 0.0004,
1237
  "step": 1503
1238
  },
1239
  {
1240
- "epoch": 22.24,
1241
- "learning_rate": 1.2287581699346406e-05,
1242
- "loss": 0.001,
1243
  "step": 1512
1244
  },
1245
  {
1246
- "epoch": 22.37,
1247
- "learning_rate": 1.1699346405228759e-05,
1248
- "loss": 0.0008,
 
 
 
 
 
 
 
 
 
 
1249
  "step": 1521
1250
  },
1251
  {
1252
- "epoch": 22.5,
1253
- "learning_rate": 1.1111111111111112e-05,
1254
- "loss": 0.0013,
1255
  "step": 1530
1256
  },
1257
  {
1258
- "epoch": 22.63,
1259
- "learning_rate": 1.0522875816993465e-05,
1260
- "loss": 0.0007,
1261
  "step": 1539
1262
  },
1263
  {
1264
- "epoch": 22.76,
1265
- "learning_rate": 9.934640522875818e-06,
1266
- "loss": 0.0013,
1267
  "step": 1548
1268
  },
1269
  {
1270
- "epoch": 22.9,
1271
- "learning_rate": 9.34640522875817e-06,
1272
- "loss": 0.0008,
1273
  "step": 1557
1274
  },
1275
  {
1276
- "epoch": 23.0,
1277
- "eval_loss": 0.001398879336193204,
1278
- "eval_max_distance": 3,
1279
- "eval_mean_distance": 0,
1280
- "eval_runtime": 1.8234,
1281
- "eval_samples_per_second": 42.23,
1282
- "eval_steps_per_second": 1.645,
1283
- "step": 1564
1284
- },
1285
- {
1286
- "epoch": 23.03,
1287
- "learning_rate": 8.758169934640522e-06,
1288
- "loss": 0.0025,
1289
  "step": 1566
1290
  },
1291
  {
1292
- "epoch": 23.16,
1293
- "learning_rate": 8.169934640522877e-06,
1294
- "loss": 0.0003,
1295
  "step": 1575
1296
  },
1297
  {
1298
- "epoch": 23.29,
1299
- "learning_rate": 7.581699346405229e-06,
1300
- "loss": 0.0009,
1301
  "step": 1584
1302
  },
1303
  {
1304
- "epoch": 23.43,
1305
- "learning_rate": 6.993464052287582e-06,
1306
- "loss": 0.0011,
 
 
 
 
 
 
 
 
 
 
1307
  "step": 1593
1308
  },
1309
  {
1310
- "epoch": 23.56,
1311
- "learning_rate": 6.405228758169935e-06,
1312
- "loss": 0.0026,
1313
  "step": 1602
1314
  },
1315
  {
1316
- "epoch": 23.69,
1317
- "learning_rate": 5.816993464052288e-06,
1318
- "loss": 0.0013,
1319
  "step": 1611
1320
  },
1321
  {
1322
- "epoch": 23.82,
1323
- "learning_rate": 5.228758169934641e-06,
1324
- "loss": 0.0012,
1325
  "step": 1620
1326
  },
1327
  {
1328
- "epoch": 23.96,
1329
- "learning_rate": 4.640522875816994e-06,
1330
- "loss": 0.0005,
1331
  "step": 1629
1332
  },
1333
  {
1334
- "epoch": 24.0,
1335
- "eval_loss": 0.0014485500287264585,
1336
- "eval_max_distance": 3,
1337
- "eval_mean_distance": 0,
1338
- "eval_runtime": 1.8281,
1339
- "eval_samples_per_second": 42.12,
1340
- "eval_steps_per_second": 1.641,
1341
- "step": 1632
1342
- },
1343
- {
1344
- "epoch": 24.09,
1345
- "learning_rate": 4.052287581699347e-06,
1346
- "loss": 0.0015,
1347
  "step": 1638
1348
  },
1349
  {
1350
- "epoch": 24.22,
1351
- "learning_rate": 3.4640522875816997e-06,
1352
  "loss": 0.0013,
1353
  "step": 1647
1354
  },
1355
  {
1356
- "epoch": 24.35,
1357
- "learning_rate": 2.8758169934640523e-06,
1358
- "loss": 0.0019,
1359
  "step": 1656
1360
  },
1361
  {
1362
- "epoch": 24.49,
1363
- "learning_rate": 2.2875816993464053e-06,
1364
- "loss": 0.0005,
 
 
 
 
 
 
 
 
 
 
1365
  "step": 1665
1366
  },
1367
  {
1368
- "epoch": 24.62,
1369
- "learning_rate": 1.699346405228758e-06,
1370
- "loss": 0.0012,
1371
  "step": 1674
1372
  },
1373
  {
1374
- "epoch": 24.75,
1375
- "learning_rate": 1.1111111111111112e-06,
1376
- "loss": 0.001,
1377
  "step": 1683
1378
  },
1379
  {
1380
- "epoch": 24.88,
1381
- "learning_rate": 5.22875816993464e-07,
1382
- "loss": 0.001,
1383
  "step": 1692
1384
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1385
  {
1386
  "epoch": 25.0,
1387
- "eval_loss": 0.0014130847994238138,
1388
- "eval_max_distance": 3,
1389
  "eval_mean_distance": 0,
1390
- "eval_runtime": 1.9218,
1391
- "eval_samples_per_second": 40.066,
1392
- "eval_steps_per_second": 1.561,
1393
- "step": 1700
1394
  },
1395
  {
1396
  "epoch": 25.0,
1397
- "step": 1700,
1398
- "total_flos": 456120480651264.0,
1399
- "train_loss": 0.003576931923401871,
1400
- "train_runtime": 194.6035,
1401
- "train_samples_per_second": 259.373,
1402
- "train_steps_per_second": 8.736
1403
  }
1404
  ],
1405
  "logging_steps": 9,
1406
- "max_steps": 1700,
1407
  "num_train_epochs": 25,
1408
- "save_steps": 17,
1409
- "total_flos": 456120480651264.0,
1410
  "trial_name": null,
1411
  "trial_params": null
1412
  }
 
1
  {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
  "epoch": 25.0,
5
  "eval_steps": 500,
6
+ "global_step": 1725,
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.01,
13
+ "learning_rate": 5.780346820809248e-07,
14
+ "loss": 0.005,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.13,
19
+ "learning_rate": 5.202312138728324e-06,
20
+ "loss": 0.0008,
21
  "step": 9
22
  },
23
  {
24
  "epoch": 0.26,
25
+ "learning_rate": 1.0404624277456647e-05,
26
+ "loss": 0.0019,
27
  "step": 18
28
  },
29
  {
30
+ "epoch": 0.39,
31
+ "learning_rate": 1.5606936416184973e-05,
32
+ "loss": 0.0019,
33
  "step": 27
34
  },
35
  {
36
+ "epoch": 0.52,
37
+ "learning_rate": 2.0809248554913295e-05,
38
+ "loss": 0.0016,
39
  "step": 36
40
  },
41
  {
42
+ "epoch": 0.65,
43
+ "learning_rate": 2.6011560693641617e-05,
44
+ "loss": 0.0027,
45
  "step": 45
46
  },
47
  {
48
+ "epoch": 0.78,
49
+ "learning_rate": 3.1213872832369946e-05,
50
+ "loss": 0.0032,
51
  "step": 54
52
  },
53
  {
54
+ "epoch": 0.91,
55
+ "learning_rate": 3.6416184971098265e-05,
56
+ "loss": 0.0013,
57
  "step": 63
58
  },
59
  {
60
  "epoch": 1.0,
61
+ "eval_loss": 0.002811400219798088,
62
+ "eval_max_distance": 2,
63
  "eval_mean_distance": 0,
64
+ "eval_runtime": 1.9415,
65
+ "eval_samples_per_second": 42.235,
66
+ "eval_steps_per_second": 1.545,
67
+ "step": 69
68
  },
69
  {
70
+ "epoch": 1.04,
71
+ "learning_rate": 4.161849710982659e-05,
72
+ "loss": 0.0033,
73
  "step": 72
74
  },
75
  {
76
+ "epoch": 1.17,
77
+ "learning_rate": 4.6820809248554915e-05,
78
+ "loss": 0.0034,
79
  "step": 81
80
  },
81
  {
82
+ "epoch": 1.3,
83
+ "learning_rate": 5.2023121387283234e-05,
84
+ "loss": 0.0015,
85
  "step": 90
86
  },
87
  {
88
+ "epoch": 1.43,
89
+ "learning_rate": 5.722543352601156e-05,
90
+ "loss": 0.0042,
91
  "step": 99
92
  },
93
  {
94
+ "epoch": 1.57,
95
+ "learning_rate": 6.242774566473989e-05,
96
+ "loss": 0.0025,
97
  "step": 108
98
  },
99
  {
100
+ "epoch": 1.7,
101
+ "learning_rate": 6.763005780346822e-05,
102
+ "loss": 0.0012,
103
  "step": 117
104
  },
105
  {
106
+ "epoch": 1.83,
107
+ "learning_rate": 7.283236994219653e-05,
108
+ "loss": 0.002,
109
  "step": 126
110
  },
111
  {
112
+ "epoch": 1.96,
113
+ "learning_rate": 7.803468208092485e-05,
114
+ "loss": 0.0006,
115
  "step": 135
116
  },
117
  {
118
  "epoch": 2.0,
119
+ "eval_loss": 0.0026117784436792135,
120
  "eval_max_distance": 3,
121
  "eval_mean_distance": 0,
122
+ "eval_runtime": 1.9167,
123
+ "eval_samples_per_second": 42.781,
124
+ "eval_steps_per_second": 1.565,
125
+ "step": 138
126
  },
127
  {
128
+ "epoch": 2.09,
129
+ "learning_rate": 8.323699421965318e-05,
130
+ "loss": 0.001,
131
  "step": 144
132
  },
133
  {
134
+ "epoch": 2.22,
135
+ "learning_rate": 8.84393063583815e-05,
136
+ "loss": 0.0016,
137
  "step": 153
138
  },
139
  {
140
+ "epoch": 2.35,
141
+ "learning_rate": 9.364161849710983e-05,
142
+ "loss": 0.0011,
143
  "step": 162
144
  },
145
  {
146
+ "epoch": 2.48,
147
+ "learning_rate": 9.884393063583816e-05,
148
+ "loss": 0.001,
149
  "step": 171
150
  },
151
  {
152
+ "epoch": 2.61,
153
+ "learning_rate": 9.954896907216495e-05,
154
+ "loss": 0.0026,
155
  "step": 180
156
  },
157
  {
158
+ "epoch": 2.74,
159
+ "learning_rate": 9.896907216494846e-05,
160
+ "loss": 0.0008,
161
  "step": 189
162
  },
163
  {
164
+ "epoch": 2.87,
165
+ "learning_rate": 9.838917525773196e-05,
166
+ "loss": 0.0022,
167
  "step": 198
168
  },
169
  {
170
  "epoch": 3.0,
171
+ "learning_rate": 9.780927835051546e-05,
172
+ "loss": 0.0025,
173
+ "step": 207
 
 
 
 
174
  },
175
  {
176
+ "epoch": 3.0,
177
+ "eval_loss": 0.003930480219423771,
178
+ "eval_max_distance": 3,
179
+ "eval_mean_distance": 0,
180
+ "eval_runtime": 1.9341,
181
+ "eval_samples_per_second": 42.396,
182
+ "eval_steps_per_second": 1.551,
183
  "step": 207
184
  },
185
  {
186
+ "epoch": 3.13,
187
+ "learning_rate": 9.722938144329897e-05,
188
+ "loss": 0.0019,
189
  "step": 216
190
  },
191
  {
192
+ "epoch": 3.26,
193
+ "learning_rate": 9.664948453608248e-05,
194
+ "loss": 0.0024,
195
  "step": 225
196
  },
197
  {
198
+ "epoch": 3.39,
199
+ "learning_rate": 9.606958762886598e-05,
200
+ "loss": 0.0007,
201
  "step": 234
202
  },
203
  {
204
+ "epoch": 3.52,
205
+ "learning_rate": 9.54896907216495e-05,
206
+ "loss": 0.0026,
207
  "step": 243
208
  },
209
  {
210
+ "epoch": 3.65,
211
+ "learning_rate": 9.490979381443299e-05,
212
+ "loss": 0.0009,
213
  "step": 252
214
  },
215
  {
216
+ "epoch": 3.78,
217
+ "learning_rate": 9.43298969072165e-05,
218
+ "loss": 0.0012,
219
  "step": 261
220
  },
221
  {
222
+ "epoch": 3.91,
223
+ "learning_rate": 9.375e-05,
224
+ "loss": 0.0004,
225
  "step": 270
226
  },
227
  {
228
  "epoch": 4.0,
229
+ "eval_loss": 0.0036886015441268682,
230
  "eval_max_distance": 3,
231
  "eval_mean_distance": 0,
232
+ "eval_runtime": 1.9271,
233
+ "eval_samples_per_second": 42.551,
234
+ "eval_steps_per_second": 1.557,
235
+ "step": 276
236
  },
237
  {
238
+ "epoch": 4.04,
239
+ "learning_rate": 9.317010309278351e-05,
240
+ "loss": 0.0018,
241
  "step": 279
242
  },
243
  {
244
+ "epoch": 4.17,
245
+ "learning_rate": 9.259020618556701e-05,
246
+ "loss": 0.0017,
247
  "step": 288
248
  },
249
  {
250
+ "epoch": 4.3,
251
+ "learning_rate": 9.201030927835051e-05,
252
+ "loss": 0.0014,
253
  "step": 297
254
  },
255
  {
256
+ "epoch": 4.43,
257
+ "learning_rate": 9.143041237113402e-05,
258
+ "loss": 0.0005,
259
  "step": 306
260
  },
261
  {
262
+ "epoch": 4.57,
263
+ "learning_rate": 9.085051546391753e-05,
264
+ "loss": 0.001,
265
  "step": 315
266
  },
267
  {
268
+ "epoch": 4.7,
269
+ "learning_rate": 9.027061855670103e-05,
270
+ "loss": 0.002,
271
  "step": 324
272
  },
273
  {
274
+ "epoch": 4.83,
275
+ "learning_rate": 8.969072164948454e-05,
276
+ "loss": 0.0016,
277
  "step": 333
278
  },
279
+ {
280
+ "epoch": 4.96,
281
+ "learning_rate": 8.911082474226806e-05,
282
+ "loss": 0.0005,
283
+ "step": 342
284
+ },
285
  {
286
  "epoch": 5.0,
287
+ "eval_loss": 0.009117466397583485,
288
  "eval_max_distance": 3,
289
  "eval_mean_distance": 0,
290
+ "eval_runtime": 1.9694,
291
+ "eval_samples_per_second": 41.638,
292
+ "eval_steps_per_second": 1.523,
293
+ "step": 345
294
  },
295
  {
296
+ "epoch": 5.09,
297
+ "learning_rate": 8.853092783505154e-05,
298
+ "loss": 0.0005,
 
 
 
 
 
 
299
  "step": 351
300
  },
301
  {
302
+ "epoch": 5.22,
303
+ "learning_rate": 8.795103092783505e-05,
304
+ "loss": 0.0051,
305
  "step": 360
306
  },
307
  {
308
+ "epoch": 5.35,
309
+ "learning_rate": 8.737113402061856e-05,
310
+ "loss": 0.0004,
311
  "step": 369
312
  },
313
  {
314
+ "epoch": 5.48,
315
+ "learning_rate": 8.679123711340206e-05,
316
+ "loss": 0.0012,
317
  "step": 378
318
  },
319
  {
320
+ "epoch": 5.61,
321
+ "learning_rate": 8.621134020618558e-05,
322
+ "loss": 0.001,
323
  "step": 387
324
  },
325
  {
326
+ "epoch": 5.74,
327
+ "learning_rate": 8.563144329896907e-05,
328
+ "loss": 0.0015,
329
  "step": 396
330
  },
331
  {
332
+ "epoch": 5.87,
333
+ "learning_rate": 8.505154639175259e-05,
334
+ "loss": 0.0016,
335
  "step": 405
336
  },
337
  {
338
  "epoch": 6.0,
339
+ "learning_rate": 8.447164948453608e-05,
340
+ "loss": 0.0009,
341
+ "step": 414
 
 
 
 
342
  },
343
  {
344
+ "epoch": 6.0,
345
+ "eval_loss": 0.0005720060435123742,
346
+ "eval_max_distance": 0,
347
+ "eval_mean_distance": 0,
348
+ "eval_runtime": 1.9399,
349
+ "eval_samples_per_second": 42.271,
350
+ "eval_steps_per_second": 1.546,
351
  "step": 414
352
  },
353
  {
354
+ "epoch": 6.13,
355
+ "learning_rate": 8.38917525773196e-05,
356
+ "loss": 0.0005,
357
  "step": 423
358
  },
359
  {
360
+ "epoch": 6.26,
361
+ "learning_rate": 8.331185567010311e-05,
362
+ "loss": 0.0006,
363
  "step": 432
364
  },
365
  {
366
+ "epoch": 6.39,
367
+ "learning_rate": 8.273195876288659e-05,
368
+ "loss": 0.0014,
369
  "step": 441
370
  },
371
  {
372
+ "epoch": 6.52,
373
+ "learning_rate": 8.21520618556701e-05,
374
+ "loss": 0.0019,
375
  "step": 450
376
  },
377
  {
378
+ "epoch": 6.65,
379
+ "learning_rate": 8.157216494845362e-05,
380
+ "loss": 0.001,
381
  "step": 459
382
  },
383
  {
384
+ "epoch": 6.78,
385
+ "learning_rate": 8.099226804123711e-05,
386
  "loss": 0.0008,
387
  "step": 468
388
  },
389
  {
390
+ "epoch": 6.91,
391
+ "learning_rate": 8.041237113402063e-05,
392
+ "loss": 0.0016,
393
+ "step": 477
 
 
 
 
394
  },
395
  {
396
+ "epoch": 7.0,
397
+ "eval_loss": 0.00027213190332986414,
398
+ "eval_max_distance": 0,
399
+ "eval_mean_distance": 0,
400
+ "eval_runtime": 1.9188,
401
+ "eval_samples_per_second": 42.735,
402
+ "eval_steps_per_second": 1.563,
403
+ "step": 483
404
  },
405
  {
406
+ "epoch": 7.04,
407
+ "learning_rate": 7.983247422680414e-05,
408
+ "loss": 0.0006,
409
  "step": 486
410
  },
411
  {
412
+ "epoch": 7.17,
413
+ "learning_rate": 7.925257731958762e-05,
414
+ "loss": 0.007,
415
  "step": 495
416
  },
417
  {
418
+ "epoch": 7.3,
419
+ "learning_rate": 7.867268041237113e-05,
420
+ "loss": 0.0004,
421
  "step": 504
422
  },
423
  {
424
+ "epoch": 7.43,
425
+ "learning_rate": 7.809278350515465e-05,
426
+ "loss": 0.0016,
427
  "step": 513
428
  },
429
  {
430
+ "epoch": 7.57,
431
+ "learning_rate": 7.751288659793814e-05,
432
+ "loss": 0.0006,
433
  "step": 522
434
  },
435
  {
436
+ "epoch": 7.7,
437
+ "learning_rate": 7.693298969072166e-05,
438
+ "loss": 0.0011,
439
  "step": 531
440
  },
441
  {
442
+ "epoch": 7.83,
443
+ "learning_rate": 7.635309278350515e-05,
444
+ "loss": 0.0014,
445
  "step": 540
446
  },
447
+ {
448
+ "epoch": 7.96,
449
+ "learning_rate": 7.577319587628867e-05,
450
+ "loss": 0.0012,
451
+ "step": 549
452
+ },
453
  {
454
  "epoch": 8.0,
455
+ "eval_loss": 0.011139851063489914,
456
+ "eval_max_distance": 5,
457
  "eval_mean_distance": 0,
458
+ "eval_runtime": 1.9435,
459
+ "eval_samples_per_second": 42.193,
460
+ "eval_steps_per_second": 1.544,
461
+ "step": 552
462
  },
463
  {
464
+ "epoch": 8.09,
465
+ "learning_rate": 7.519329896907217e-05,
466
  "loss": 0.0008,
 
 
 
 
 
 
467
  "step": 558
468
  },
469
  {
470
+ "epoch": 8.22,
471
+ "learning_rate": 7.461340206185568e-05,
472
+ "loss": 0.0011,
473
  "step": 567
474
  },
475
  {
476
+ "epoch": 8.35,
477
+ "learning_rate": 7.403350515463919e-05,
478
+ "loss": 0.0025,
479
  "step": 576
480
  },
481
  {
482
+ "epoch": 8.48,
483
+ "learning_rate": 7.345360824742269e-05,
484
+ "loss": 0.003,
485
  "step": 585
486
  },
487
  {
488
+ "epoch": 8.61,
489
+ "learning_rate": 7.287371134020619e-05,
490
+ "loss": 0.004,
491
  "step": 594
492
  },
493
  {
494
+ "epoch": 8.74,
495
+ "learning_rate": 7.22938144329897e-05,
496
+ "loss": 0.002,
497
  "step": 603
498
  },
499
  {
500
+ "epoch": 8.87,
501
+ "learning_rate": 7.17139175257732e-05,
502
+ "loss": 0.0007,
503
  "step": 612
504
  },
505
  {
506
  "epoch": 9.0,
507
+ "learning_rate": 7.113402061855671e-05,
508
+ "loss": 0.0008,
509
+ "step": 621
 
 
 
 
510
  },
511
  {
512
+ "epoch": 9.0,
513
+ "eval_loss": 0.0003953798732254654,
514
+ "eval_max_distance": 0,
515
+ "eval_mean_distance": 0,
516
+ "eval_runtime": 1.995,
517
+ "eval_samples_per_second": 41.102,
518
+ "eval_steps_per_second": 1.504,
519
  "step": 621
520
  },
521
  {
522
+ "epoch": 9.13,
523
+ "learning_rate": 7.055412371134022e-05,
524
+ "loss": 0.0005,
525
  "step": 630
526
  },
527
  {
528
+ "epoch": 9.26,
529
+ "learning_rate": 6.99742268041237e-05,
530
+ "loss": 0.0004,
531
  "step": 639
532
  },
533
  {
534
+ "epoch": 9.39,
535
+ "learning_rate": 6.939432989690722e-05,
536
+ "loss": 0.0013,
537
  "step": 648
538
  },
539
  {
540
+ "epoch": 9.52,
541
+ "learning_rate": 6.881443298969073e-05,
542
+ "loss": 0.0002,
543
  "step": 657
544
  },
545
  {
546
+ "epoch": 9.65,
547
+ "learning_rate": 6.823453608247423e-05,
548
+ "loss": 0.0011,
549
  "step": 666
550
  },
551
  {
552
+ "epoch": 9.78,
553
+ "learning_rate": 6.765463917525774e-05,
554
+ "loss": 0.0018,
555
  "step": 675
556
  },
557
  {
558
+ "epoch": 9.91,
559
+ "learning_rate": 6.707474226804124e-05,
560
+ "loss": 0.0018,
561
+ "step": 684
 
 
 
 
562
  },
563
  {
564
+ "epoch": 10.0,
565
+ "eval_loss": 0.00027754431357607245,
566
+ "eval_max_distance": 0,
567
+ "eval_mean_distance": 0,
568
+ "eval_runtime": 1.9222,
569
+ "eval_samples_per_second": 42.659,
570
+ "eval_steps_per_second": 1.561,
571
+ "step": 690
572
  },
573
  {
574
+ "epoch": 10.04,
575
+ "learning_rate": 6.649484536082475e-05,
576
+ "loss": 0.0011,
577
  "step": 693
578
  },
579
  {
580
+ "epoch": 10.17,
581
+ "learning_rate": 6.591494845360825e-05,
582
+ "loss": 0.0006,
583
  "step": 702
584
  },
585
  {
586
+ "epoch": 10.3,
587
+ "learning_rate": 6.533505154639176e-05,
588
+ "loss": 0.0011,
589
  "step": 711
590
  },
591
  {
592
+ "epoch": 10.43,
593
+ "learning_rate": 6.475515463917527e-05,
594
+ "loss": 0.0013,
595
  "step": 720
596
  },
597
  {
598
+ "epoch": 10.57,
599
+ "learning_rate": 6.417525773195877e-05,
600
+ "loss": 0.0006,
601
  "step": 729
602
  },
603
  {
604
+ "epoch": 10.7,
605
+ "learning_rate": 6.359536082474227e-05,
606
+ "loss": 0.0018,
607
  "step": 738
608
  },
609
  {
610
+ "epoch": 10.83,
611
+ "learning_rate": 6.301546391752578e-05,
612
+ "loss": 0.0016,
613
  "step": 747
614
  },
615
  {
616
+ "epoch": 10.96,
617
+ "learning_rate": 6.243556701030928e-05,
618
+ "loss": 0.0028,
619
+ "step": 756
 
 
 
 
620
  },
621
  {
622
+ "epoch": 11.0,
623
+ "eval_loss": 0.00033258015173487365,
624
+ "eval_max_distance": 0,
625
+ "eval_mean_distance": 0,
626
+ "eval_runtime": 1.9385,
627
+ "eval_samples_per_second": 42.301,
628
+ "eval_steps_per_second": 1.548,
629
+ "step": 759
630
  },
631
  {
632
+ "epoch": 11.09,
633
+ "learning_rate": 6.185567010309279e-05,
634
+ "loss": 0.0009,
635
  "step": 765
636
  },
637
  {
638
+ "epoch": 11.22,
639
+ "learning_rate": 6.12757731958763e-05,
640
+ "loss": 0.0005,
641
  "step": 774
642
  },
643
  {
644
+ "epoch": 11.35,
645
+ "learning_rate": 6.069587628865979e-05,
646
+ "loss": 0.0011,
647
  "step": 783
648
  },
649
  {
650
+ "epoch": 11.48,
651
+ "learning_rate": 6.01159793814433e-05,
652
+ "loss": 0.0007,
653
  "step": 792
654
  },
655
  {
656
+ "epoch": 11.61,
657
+ "learning_rate": 5.953608247422681e-05,
658
+ "loss": 0.0012,
659
  "step": 801
660
  },
661
  {
662
+ "epoch": 11.74,
663
+ "learning_rate": 5.8956185567010315e-05,
664
+ "loss": 0.0021,
665
  "step": 810
666
  },
667
  {
668
+ "epoch": 11.87,
669
+ "learning_rate": 5.837628865979382e-05,
670
+ "loss": 0.0006,
671
+ "step": 819
 
 
 
 
672
  },
673
  {
674
+ "epoch": 12.0,
675
+ "learning_rate": 5.779639175257732e-05,
676
+ "loss": 0.0008,
677
+ "step": 828
678
  },
679
  {
680
+ "epoch": 12.0,
681
+ "eval_loss": 0.0002690624096430838,
682
+ "eval_max_distance": 0,
683
+ "eval_mean_distance": 0,
684
+ "eval_runtime": 1.9448,
685
+ "eval_samples_per_second": 42.163,
686
+ "eval_steps_per_second": 1.543,
687
  "step": 828
688
  },
689
  {
690
+ "epoch": 12.13,
691
+ "learning_rate": 5.721649484536082e-05,
692
+ "loss": 0.001,
693
  "step": 837
694
  },
695
  {
696
+ "epoch": 12.26,
697
+ "learning_rate": 5.663659793814433e-05,
698
+ "loss": 0.0012,
699
  "step": 846
700
  },
701
  {
702
+ "epoch": 12.39,
703
+ "learning_rate": 5.605670103092784e-05,
704
+ "loss": 0.0005,
705
  "step": 855
706
  },
707
  {
708
+ "epoch": 12.52,
709
+ "learning_rate": 5.5476804123711345e-05,
710
+ "loss": 0.0011,
711
  "step": 864
712
  },
713
  {
714
+ "epoch": 12.65,
715
+ "learning_rate": 5.489690721649485e-05,
716
+ "loss": 0.0026,
717
  "step": 873
718
  },
719
  {
720
+ "epoch": 12.78,
721
+ "learning_rate": 5.431701030927835e-05,
722
+ "loss": 0.0009,
723
  "step": 882
724
  },
725
  {
726
+ "epoch": 12.91,
727
+ "learning_rate": 5.3737113402061854e-05,
728
+ "loss": 0.001,
729
+ "step": 891
 
 
 
 
730
  },
731
  {
732
+ "epoch": 13.0,
733
+ "eval_loss": 0.0004277784610167146,
734
+ "eval_max_distance": 2,
735
+ "eval_mean_distance": 0,
736
+ "eval_runtime": 1.9315,
737
+ "eval_samples_per_second": 42.454,
738
+ "eval_steps_per_second": 1.553,
739
+ "step": 897
740
  },
741
  {
742
+ "epoch": 13.04,
743
+ "learning_rate": 5.3157216494845366e-05,
744
+ "loss": 0.0007,
745
  "step": 900
746
  },
747
  {
748
+ "epoch": 13.17,
749
+ "learning_rate": 5.257731958762887e-05,
750
  "loss": 0.0008,
751
  "step": 909
752
  },
753
  {
754
+ "epoch": 13.3,
755
+ "learning_rate": 5.1997422680412376e-05,
756
+ "loss": 0.0005,
757
  "step": 918
758
  },
759
  {
760
+ "epoch": 13.43,
761
+ "learning_rate": 5.1417525773195874e-05,
762
+ "loss": 0.0004,
763
  "step": 927
764
  },
765
  {
766
+ "epoch": 13.57,
767
+ "learning_rate": 5.083762886597938e-05,
768
+ "loss": 0.0006,
769
  "step": 936
770
  },
771
  {
772
+ "epoch": 13.7,
773
+ "learning_rate": 5.025773195876289e-05,
774
+ "loss": 0.0025,
775
  "step": 945
776
  },
777
  {
778
+ "epoch": 13.83,
779
+ "learning_rate": 4.9677835051546396e-05,
780
+ "loss": 0.0009,
781
+ "step": 954
 
 
 
 
782
  },
783
  {
784
+ "epoch": 13.96,
785
+ "learning_rate": 4.9097938144329895e-05,
786
+ "loss": 0.0026,
787
+ "step": 963
788
  },
789
  {
790
+ "epoch": 14.0,
791
+ "eval_loss": 0.0005385838449001312,
792
+ "eval_max_distance": 2,
793
+ "eval_mean_distance": 0,
794
+ "eval_runtime": 1.993,
795
+ "eval_samples_per_second": 41.144,
796
+ "eval_steps_per_second": 1.505,
797
+ "step": 966
798
  },
799
  {
800
+ "epoch": 14.09,
801
+ "learning_rate": 4.8518041237113407e-05,
802
+ "loss": 0.0016,
803
  "step": 972
804
  },
805
  {
806
+ "epoch": 14.22,
807
+ "learning_rate": 4.793814432989691e-05,
808
+ "loss": 0.0014,
809
  "step": 981
810
  },
811
  {
812
+ "epoch": 14.35,
813
+ "learning_rate": 4.735824742268041e-05,
814
+ "loss": 0.0007,
815
  "step": 990
816
  },
817
  {
818
+ "epoch": 14.48,
819
+ "learning_rate": 4.677835051546392e-05,
820
+ "loss": 0.0031,
821
  "step": 999
822
  },
823
  {
824
+ "epoch": 14.61,
825
+ "learning_rate": 4.619845360824743e-05,
826
+ "loss": 0.0008,
827
  "step": 1008
828
  },
829
  {
830
+ "epoch": 14.74,
831
+ "learning_rate": 4.561855670103093e-05,
832
+ "loss": 0.0028,
833
  "step": 1017
834
  },
835
  {
836
+ "epoch": 14.87,
837
+ "learning_rate": 4.503865979381444e-05,
838
+ "loss": 0.0004,
839
+ "step": 1026
 
 
 
 
840
  },
841
  {
842
+ "epoch": 15.0,
843
+ "learning_rate": 4.4458762886597936e-05,
844
+ "loss": 0.0015,
845
+ "step": 1035
846
  },
847
  {
848
+ "epoch": 15.0,
849
+ "eval_loss": 0.0007138837827369571,
850
+ "eval_max_distance": 3,
851
+ "eval_mean_distance": 0,
852
+ "eval_runtime": 1.9688,
853
+ "eval_samples_per_second": 41.651,
854
+ "eval_steps_per_second": 1.524,
855
  "step": 1035
856
  },
857
  {
858
+ "epoch": 15.13,
859
+ "learning_rate": 4.387886597938145e-05,
860
+ "loss": 0.0005,
861
  "step": 1044
862
  },
863
  {
864
+ "epoch": 15.26,
865
+ "learning_rate": 4.329896907216495e-05,
866
+ "loss": 0.0011,
867
  "step": 1053
868
  },
869
  {
870
+ "epoch": 15.39,
871
+ "learning_rate": 4.271907216494845e-05,
872
+ "loss": 0.0001,
873
  "step": 1062
874
  },
875
  {
876
+ "epoch": 15.52,
877
+ "learning_rate": 4.213917525773196e-05,
878
+ "loss": 0.0009,
879
  "step": 1071
880
  },
881
  {
882
+ "epoch": 15.65,
883
+ "learning_rate": 4.155927835051547e-05,
884
+ "loss": 0.0017,
885
  "step": 1080
886
  },
887
  {
888
+ "epoch": 15.78,
889
+ "learning_rate": 4.097938144329897e-05,
890
+ "loss": 0.0013,
 
 
 
 
 
 
 
 
 
 
891
  "step": 1089
892
  },
893
  {
894
+ "epoch": 15.91,
895
+ "learning_rate": 4.039948453608248e-05,
896
+ "loss": 0.0009,
897
  "step": 1098
898
  },
899
  {
900
+ "epoch": 16.0,
901
+ "eval_loss": 0.0006717974320054054,
902
+ "eval_max_distance": 3,
903
+ "eval_mean_distance": 0,
904
+ "eval_runtime": 1.9244,
905
+ "eval_samples_per_second": 42.612,
906
+ "eval_steps_per_second": 1.559,
907
+ "step": 1104
908
+ },
909
+ {
910
+ "epoch": 16.04,
911
+ "learning_rate": 3.9819587628865976e-05,
912
+ "loss": 0.0015,
913
  "step": 1107
914
  },
915
  {
916
+ "epoch": 16.17,
917
+ "learning_rate": 3.923969072164949e-05,
918
+ "loss": 0.0003,
919
  "step": 1116
920
  },
921
  {
922
+ "epoch": 16.3,
923
+ "learning_rate": 3.865979381443299e-05,
924
  "loss": 0.0006,
925
  "step": 1125
926
  },
927
  {
928
+ "epoch": 16.43,
929
+ "learning_rate": 3.807989690721649e-05,
930
+ "loss": 0.0013,
931
  "step": 1134
932
  },
933
  {
934
+ "epoch": 16.57,
935
+ "learning_rate": 3.7500000000000003e-05,
936
+ "loss": 0.0009,
937
  "step": 1143
938
  },
939
  {
940
+ "epoch": 16.7,
941
+ "learning_rate": 3.692010309278351e-05,
942
+ "loss": 0.0004,
943
  "step": 1152
944
  },
945
  {
946
+ "epoch": 16.83,
947
+ "learning_rate": 3.6340206185567014e-05,
948
+ "loss": 0.0008,
 
 
 
 
 
 
 
 
 
 
949
  "step": 1161
950
  },
951
  {
952
+ "epoch": 16.96,
953
+ "learning_rate": 3.576030927835052e-05,
954
+ "loss": 0.0014,
955
  "step": 1170
956
  },
957
  {
958
+ "epoch": 17.0,
959
+ "eval_loss": 0.00033988503855653107,
960
+ "eval_max_distance": 0,
961
+ "eval_mean_distance": 0,
962
+ "eval_runtime": 1.9136,
963
+ "eval_samples_per_second": 42.851,
964
+ "eval_steps_per_second": 1.568,
965
+ "step": 1173
966
+ },
967
+ {
968
+ "epoch": 17.09,
969
+ "learning_rate": 3.5180412371134024e-05,
970
+ "loss": 0.0016,
971
  "step": 1179
972
  },
973
  {
974
+ "epoch": 17.22,
975
+ "learning_rate": 3.460051546391753e-05,
976
+ "loss": 0.0004,
977
  "step": 1188
978
  },
979
  {
980
+ "epoch": 17.35,
981
+ "learning_rate": 3.4020618556701034e-05,
982
+ "loss": 0.002,
983
  "step": 1197
984
  },
985
  {
986
+ "epoch": 17.48,
987
+ "learning_rate": 3.344072164948453e-05,
988
+ "loss": 0.0012,
989
  "step": 1206
990
  },
991
  {
992
+ "epoch": 17.61,
993
+ "learning_rate": 3.2860824742268044e-05,
994
+ "loss": 0.0002,
995
  "step": 1215
996
  },
997
  {
998
+ "epoch": 17.74,
999
+ "learning_rate": 3.228092783505155e-05,
1000
  "loss": 0.0006,
1001
  "step": 1224
1002
  },
1003
  {
1004
+ "epoch": 17.87,
1005
+ "learning_rate": 3.1701030927835054e-05,
1006
+ "loss": 0.0044,
1007
+ "step": 1233
 
 
 
 
1008
  },
1009
  {
1010
+ "epoch": 18.0,
1011
+ "learning_rate": 3.112113402061856e-05,
1012
+ "loss": 0.001,
1013
+ "step": 1242
1014
  },
1015
  {
1016
+ "epoch": 18.0,
1017
+ "eval_loss": 0.00037691937177442014,
1018
+ "eval_max_distance": 0,
1019
+ "eval_mean_distance": 0,
1020
+ "eval_runtime": 1.9496,
1021
+ "eval_samples_per_second": 42.059,
1022
+ "eval_steps_per_second": 1.539,
1023
  "step": 1242
1024
  },
1025
  {
1026
+ "epoch": 18.13,
1027
+ "learning_rate": 3.0541237113402065e-05,
1028
+ "loss": 0.0005,
1029
  "step": 1251
1030
  },
1031
  {
1032
+ "epoch": 18.26,
1033
+ "learning_rate": 2.9961340206185566e-05,
1034
+ "loss": 0.0009,
1035
  "step": 1260
1036
  },
1037
  {
1038
+ "epoch": 18.39,
1039
+ "learning_rate": 2.9381443298969075e-05,
1040
+ "loss": 0.0008,
1041
  "step": 1269
1042
  },
1043
  {
1044
+ "epoch": 18.52,
1045
+ "learning_rate": 2.8801546391752577e-05,
1046
+ "loss": 0.0007,
1047
  "step": 1278
1048
  },
1049
  {
1050
+ "epoch": 18.65,
1051
+ "learning_rate": 2.8221649484536085e-05,
1052
+ "loss": 0.0016,
1053
  "step": 1287
1054
  },
1055
  {
1056
+ "epoch": 18.78,
1057
+ "learning_rate": 2.764175257731959e-05,
1058
+ "loss": 0.0012,
 
 
 
 
 
 
 
 
 
 
1059
  "step": 1296
1060
  },
1061
  {
1062
+ "epoch": 18.91,
1063
+ "learning_rate": 2.7061855670103092e-05,
1064
+ "loss": 0.0007,
1065
  "step": 1305
1066
  },
1067
  {
1068
+ "epoch": 19.0,
1069
+ "eval_loss": 0.001327142701484263,
1070
+ "eval_max_distance": 3,
1071
+ "eval_mean_distance": 0,
1072
+ "eval_runtime": 1.9144,
1073
+ "eval_samples_per_second": 42.834,
1074
+ "eval_steps_per_second": 1.567,
1075
+ "step": 1311
1076
+ },
1077
+ {
1078
+ "epoch": 19.04,
1079
+ "learning_rate": 2.64819587628866e-05,
1080
+ "loss": 0.0005,
1081
  "step": 1314
1082
  },
1083
  {
1084
+ "epoch": 19.17,
1085
+ "learning_rate": 2.5902061855670106e-05,
1086
+ "loss": 0.0011,
1087
  "step": 1323
1088
  },
1089
  {
1090
+ "epoch": 19.3,
1091
+ "learning_rate": 2.5322164948453607e-05,
1092
  "loss": 0.0006,
1093
  "step": 1332
1094
  },
1095
  {
1096
+ "epoch": 19.43,
1097
+ "learning_rate": 2.4742268041237116e-05,
1098
  "loss": 0.0004,
1099
  "step": 1341
1100
  },
1101
  {
1102
+ "epoch": 19.57,
1103
+ "learning_rate": 2.416237113402062e-05,
1104
+ "loss": 0.002,
1105
  "step": 1350
1106
  },
1107
  {
1108
+ "epoch": 19.7,
1109
+ "learning_rate": 2.3582474226804126e-05,
1110
+ "loss": 0.0003,
1111
  "step": 1359
1112
  },
1113
  {
1114
+ "epoch": 19.83,
1115
+ "learning_rate": 2.3002577319587628e-05,
1116
+ "loss": 0.0011,
1117
+ "step": 1368
 
 
 
 
1118
  },
1119
  {
1120
+ "epoch": 19.96,
1121
+ "learning_rate": 2.2422680412371136e-05,
1122
  "loss": 0.0013,
1123
+ "step": 1377
1124
  },
1125
  {
1126
+ "epoch": 20.0,
1127
+ "eval_loss": 0.0012958323350176215,
1128
+ "eval_max_distance": 3,
1129
+ "eval_mean_distance": 0,
1130
+ "eval_runtime": 1.9255,
1131
+ "eval_samples_per_second": 42.587,
1132
+ "eval_steps_per_second": 1.558,
1133
+ "step": 1380
1134
  },
1135
  {
1136
+ "epoch": 20.09,
1137
+ "learning_rate": 2.184278350515464e-05,
1138
+ "loss": 0.0017,
1139
  "step": 1386
1140
  },
1141
  {
1142
+ "epoch": 20.22,
1143
+ "learning_rate": 2.1262886597938146e-05,
1144
+ "loss": 0.0011,
1145
  "step": 1395
1146
  },
1147
  {
1148
+ "epoch": 20.35,
1149
+ "learning_rate": 2.0682989690721648e-05,
1150
+ "loss": 0.0014,
1151
  "step": 1404
1152
  },
1153
  {
1154
+ "epoch": 20.48,
1155
+ "learning_rate": 2.0103092783505157e-05,
1156
  "loss": 0.0006,
1157
  "step": 1413
1158
  },
1159
  {
1160
+ "epoch": 20.61,
1161
+ "learning_rate": 1.952319587628866e-05,
1162
+ "loss": 0.0003,
1163
  "step": 1422
1164
  },
1165
  {
1166
+ "epoch": 20.74,
1167
+ "learning_rate": 1.8943298969072167e-05,
1168
+ "loss": 0.0002,
 
 
 
 
 
 
 
 
 
 
1169
  "step": 1431
1170
  },
1171
  {
1172
+ "epoch": 20.87,
1173
+ "learning_rate": 1.8363402061855672e-05,
1174
+ "loss": 0.0024,
1175
  "step": 1440
1176
  },
1177
  {
1178
+ "epoch": 21.0,
1179
+ "learning_rate": 1.7783505154639177e-05,
1180
+ "loss": 0.0007,
1181
  "step": 1449
1182
  },
1183
  {
1184
+ "epoch": 21.0,
1185
+ "eval_loss": 0.0002567500632721931,
1186
+ "eval_max_distance": 0,
1187
+ "eval_mean_distance": 0,
1188
+ "eval_runtime": 1.9342,
1189
+ "eval_samples_per_second": 42.395,
1190
+ "eval_steps_per_second": 1.551,
1191
+ "step": 1449
1192
+ },
1193
+ {
1194
+ "epoch": 21.13,
1195
+ "learning_rate": 1.7203608247422682e-05,
1196
+ "loss": 0.0006,
1197
  "step": 1458
1198
  },
1199
  {
1200
+ "epoch": 21.26,
1201
+ "learning_rate": 1.6623711340206187e-05,
1202
+ "loss": 0.002,
1203
  "step": 1467
1204
  },
1205
  {
1206
+ "epoch": 21.39,
1207
+ "learning_rate": 1.6043814432989692e-05,
1208
+ "loss": 0.0007,
1209
  "step": 1476
1210
  },
1211
  {
1212
+ "epoch": 21.52,
1213
+ "learning_rate": 1.5463917525773197e-05,
1214
+ "loss": 0.0007,
1215
  "step": 1485
1216
  },
1217
  {
1218
+ "epoch": 21.65,
1219
+ "learning_rate": 1.4884020618556702e-05,
1220
+ "loss": 0.0016,
1221
  "step": 1494
1222
  },
1223
  {
1224
+ "epoch": 21.78,
1225
+ "learning_rate": 1.4304123711340206e-05,
1226
+ "loss": 0.0008,
 
 
 
 
 
 
 
 
 
 
1227
  "step": 1503
1228
  },
1229
  {
1230
+ "epoch": 21.91,
1231
+ "learning_rate": 1.3724226804123713e-05,
1232
+ "loss": 0.0016,
1233
  "step": 1512
1234
  },
1235
  {
1236
+ "epoch": 22.0,
1237
+ "eval_loss": 0.0002821955131366849,
1238
+ "eval_max_distance": 0,
1239
+ "eval_mean_distance": 0,
1240
+ "eval_runtime": 1.9113,
1241
+ "eval_samples_per_second": 42.902,
1242
+ "eval_steps_per_second": 1.57,
1243
+ "step": 1518
1244
+ },
1245
+ {
1246
+ "epoch": 22.04,
1247
+ "learning_rate": 1.3144329896907218e-05,
1248
+ "loss": 0.0018,
1249
  "step": 1521
1250
  },
1251
  {
1252
+ "epoch": 22.17,
1253
+ "learning_rate": 1.2564432989690723e-05,
1254
+ "loss": 0.0027,
1255
  "step": 1530
1256
  },
1257
  {
1258
+ "epoch": 22.3,
1259
+ "learning_rate": 1.1984536082474228e-05,
1260
+ "loss": 0.0002,
1261
  "step": 1539
1262
  },
1263
  {
1264
+ "epoch": 22.43,
1265
+ "learning_rate": 1.1404639175257733e-05,
1266
+ "loss": 0.001,
1267
  "step": 1548
1268
  },
1269
  {
1270
+ "epoch": 22.57,
1271
+ "learning_rate": 1.0824742268041238e-05,
1272
+ "loss": 0.0007,
1273
  "step": 1557
1274
  },
1275
  {
1276
+ "epoch": 22.7,
1277
+ "learning_rate": 1.0244845360824743e-05,
1278
+ "loss": 0.0009,
 
 
 
 
 
 
 
 
 
 
1279
  "step": 1566
1280
  },
1281
  {
1282
+ "epoch": 22.83,
1283
+ "learning_rate": 9.664948453608248e-06,
1284
+ "loss": 0.0012,
1285
  "step": 1575
1286
  },
1287
  {
1288
+ "epoch": 22.96,
1289
+ "learning_rate": 9.085051546391753e-06,
1290
+ "loss": 0.0013,
1291
  "step": 1584
1292
  },
1293
  {
1294
+ "epoch": 23.0,
1295
+ "eval_loss": 0.00030223012436181307,
1296
+ "eval_max_distance": 0,
1297
+ "eval_mean_distance": 0,
1298
+ "eval_runtime": 1.9675,
1299
+ "eval_samples_per_second": 41.677,
1300
+ "eval_steps_per_second": 1.525,
1301
+ "step": 1587
1302
+ },
1303
+ {
1304
+ "epoch": 23.09,
1305
+ "learning_rate": 8.505154639175259e-06,
1306
+ "loss": 0.0025,
1307
  "step": 1593
1308
  },
1309
  {
1310
+ "epoch": 23.22,
1311
+ "learning_rate": 7.925257731958764e-06,
1312
+ "loss": 0.001,
1313
  "step": 1602
1314
  },
1315
  {
1316
+ "epoch": 23.35,
1317
+ "learning_rate": 7.345360824742269e-06,
1318
+ "loss": 0.0004,
1319
  "step": 1611
1320
  },
1321
  {
1322
+ "epoch": 23.48,
1323
+ "learning_rate": 6.765463917525773e-06,
1324
+ "loss": 0.0006,
1325
  "step": 1620
1326
  },
1327
  {
1328
+ "epoch": 23.61,
1329
+ "learning_rate": 6.185567010309279e-06,
1330
+ "loss": 0.001,
1331
  "step": 1629
1332
  },
1333
  {
1334
+ "epoch": 23.74,
1335
+ "learning_rate": 5.605670103092784e-06,
1336
+ "loss": 0.0012,
 
 
 
 
 
 
 
 
 
 
1337
  "step": 1638
1338
  },
1339
  {
1340
+ "epoch": 23.87,
1341
+ "learning_rate": 5.025773195876289e-06,
1342
  "loss": 0.0013,
1343
  "step": 1647
1344
  },
1345
  {
1346
+ "epoch": 24.0,
1347
+ "learning_rate": 4.445876288659794e-06,
1348
+ "loss": 0.0004,
1349
  "step": 1656
1350
  },
1351
  {
1352
+ "epoch": 24.0,
1353
+ "eval_loss": 0.0002631743554957211,
1354
+ "eval_max_distance": 0,
1355
+ "eval_mean_distance": 0,
1356
+ "eval_runtime": 1.9278,
1357
+ "eval_samples_per_second": 42.536,
1358
+ "eval_steps_per_second": 1.556,
1359
+ "step": 1656
1360
+ },
1361
+ {
1362
+ "epoch": 24.13,
1363
+ "learning_rate": 3.865979381443299e-06,
1364
+ "loss": 0.0018,
1365
  "step": 1665
1366
  },
1367
  {
1368
+ "epoch": 24.26,
1369
+ "learning_rate": 3.2860824742268044e-06,
1370
+ "loss": 0.0006,
1371
  "step": 1674
1372
  },
1373
  {
1374
+ "epoch": 24.39,
1375
+ "learning_rate": 2.7061855670103095e-06,
1376
+ "loss": 0.0012,
1377
  "step": 1683
1378
  },
1379
  {
1380
+ "epoch": 24.52,
1381
+ "learning_rate": 2.1262886597938146e-06,
1382
+ "loss": 0.0009,
1383
  "step": 1692
1384
  },
1385
+ {
1386
+ "epoch": 24.65,
1387
+ "learning_rate": 1.5463917525773197e-06,
1388
+ "loss": 0.0007,
1389
+ "step": 1701
1390
+ },
1391
+ {
1392
+ "epoch": 24.78,
1393
+ "learning_rate": 9.664948453608248e-07,
1394
+ "loss": 0.0009,
1395
+ "step": 1710
1396
+ },
1397
+ {
1398
+ "epoch": 24.91,
1399
+ "learning_rate": 3.8659793814432993e-07,
1400
+ "loss": 0.001,
1401
+ "step": 1719
1402
+ },
1403
  {
1404
  "epoch": 25.0,
1405
+ "eval_loss": 0.0002593309909570962,
1406
+ "eval_max_distance": 0,
1407
  "eval_mean_distance": 0,
1408
+ "eval_runtime": 2.0449,
1409
+ "eval_samples_per_second": 40.1,
1410
+ "eval_steps_per_second": 1.467,
1411
+ "step": 1725
1412
  },
1413
  {
1414
  "epoch": 25.0,
1415
+ "step": 1725,
1416
+ "total_flos": 459342194208768.0,
1417
+ "train_loss": 0.0013090899151048043,
1418
+ "train_runtime": 199.817,
1419
+ "train_samples_per_second": 256.61,
1420
+ "train_steps_per_second": 8.633
1421
  }
1422
  ],
1423
  "logging_steps": 9,
1424
+ "max_steps": 1725,
1425
  "num_train_epochs": 25,
1426
+ "save_steps": 18,
1427
+ "total_flos": 459342194208768.0,
1428
  "trial_name": null,
1429
  "trial_params": null
1430
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e3766c94731270598b629b2e54dbd149eec47873f81720b047d2a1d23148ad6e
3
- size 4155
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1715be697a9be6fe0784bf769267cffcc821d071738c8fdd99dec6f06f39db0
3
+ size 4091