AlekseyKorshuk commited on
Commit
4fe4412
·
1 Parent(s): 51ee662

huggingartists

Browse files
README.md CHANGED
@@ -45,15 +45,15 @@ from datasets import load_dataset
45
  dataset = load_dataset("huggingartists/eminem")
46
  ```
47
 
48
- [Explore the data](https://wandb.ai/huggingartists/huggingartists/runs/2tmmbk4q/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
49
 
50
  ## Training procedure
51
 
52
  The model is based on a pre-trained [GPT-2](https://huggingface.co/gpt2) which is fine-tuned on Eminem's lyrics.
53
 
54
- Hyperparameters and metrics are recorded in the [W&B training run](https://wandb.ai/huggingartists/huggingartists/runs/1n3sf0z7) for full transparency and reproducibility.
55
 
56
- At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/1n3sf0z7/artifacts) is logged and versioned.
57
 
58
  ## How to use
59
 
 
45
  dataset = load_dataset("huggingartists/eminem")
46
  ```
47
 
48
+ [Explore the data](https://wandb.ai/huggingartists/huggingartists/runs/15cym4b5/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
49
 
50
  ## Training procedure
51
 
52
  The model is based on a pre-trained [GPT-2](https://huggingface.co/gpt2) which is fine-tuned on Eminem's lyrics.
53
 
54
+ Hyperparameters and metrics are recorded in the [W&B training run](https://wandb.ai/huggingartists/huggingartists/runs/1x3uqmm0) for full transparency and reproducibility.
55
 
56
+ At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/1x3uqmm0/artifacts) is logged and versioned.
57
 
58
  ## How to use
59
 
evaluation.txt CHANGED
@@ -1 +1 @@
1
- {"eval_loss": 0.6822583675384521, "eval_runtime": 27.1385, "eval_samples_per_second": 22.256, "eval_steps_per_second": 2.8, "epoch": 3.0}
 
1
+ {"eval_loss": 0.8025060296058655, "eval_runtime": 8.4071, "eval_samples_per_second": 76.84, "eval_steps_per_second": 9.635, "epoch": 3.0}
flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bcf0afe6139148f265c80e26d4bfc5c7039a20828882f87c6b3037890711e56e
3
  size 497764120
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b739a8ccddc25542afe5ea2119ebba6e65e6112338dbb78979b77f70e8686623
3
  size 497764120
optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7a65d3443a82bf2e41da97ef8f633284eadd651caa7a18815b9196f1e2b0847c
3
  size 995604017
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e7cb42f5dffe82bcd3b920ad61f5f93a9acc9fbfabe99f82bbe40dad162fef9
3
  size 995604017
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ef17b9c3a0ca44ea4230c596d212eb41870f4f94818e18bf2e56aa4e30f46ce8
3
  size 510403817
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67e98567401881145d820a525614414d2e99e92a4565ac3f62537226f10421a5
3
  size 510403817
rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fdee0035455f96487346df6e0d66b94b0a0cd4bb48870d0dd9b34bd67dca7c62
3
- size 14503
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db724fec037fee48b46d10275cfadd56b2c386d7a8e13404c7083dc7b9d1713f
3
+ size 14439
scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b294a94dc01fb76be071ae8b385cb1227c60af155ff6121e65edf420f2ffa8f
3
  size 623
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cac36feda9756e3d13d61874960f11354da701987062c5c64b16b3e4d3148948
3
  size 623
trainer_state.json CHANGED
@@ -1,1144 +1,1118 @@
1
  {
2
- "best_metric": 0.6822583675384521,
3
- "best_model_checkpoint": "output/eminem/checkpoint-920",
4
  "epoch": 2.0,
5
- "global_step": 920,
6
  "is_hyper_param_search": false,
7
  "is_local_process_zero": true,
8
  "is_world_process_zero": true,
9
  "log_history": [
10
  {
11
  "epoch": 0.01,
12
- "learning_rate": 0.00013618053464655743,
13
- "loss": 1.1972,
14
  "step": 5
15
  },
16
  {
17
  "epoch": 0.02,
18
- "learning_rate": 0.0001365469584575468,
19
- "loss": 1.0976,
20
  "step": 10
21
  },
22
  {
23
  "epoch": 0.03,
24
- "learning_rate": 0.00013683240859409016,
25
- "loss": 1.0008,
26
  "step": 15
27
  },
28
  {
29
  "epoch": 0.04,
30
- "learning_rate": 0.00013703654487986559,
31
- "loss": 1.1789,
32
  "step": 20
33
  },
34
  {
35
- "epoch": 0.05,
36
- "learning_rate": 0.00013715912404180336,
37
- "loss": 1.1345,
38
  "step": 25
39
  },
40
  {
41
  "epoch": 0.07,
42
- "learning_rate": 0.0001372,
43
- "loss": 1.1009,
44
  "step": 30
45
  },
46
  {
47
  "epoch": 0.08,
48
- "learning_rate": 0.00013715912404180342,
49
- "loss": 1.0427,
50
  "step": 35
51
  },
52
  {
53
  "epoch": 0.09,
54
- "learning_rate": 0.00013703654487986564,
55
- "loss": 1.078,
56
  "step": 40
57
  },
58
  {
59
  "epoch": 0.1,
60
- "learning_rate": 0.0001368324085940902,
61
- "loss": 1.1279,
62
  "step": 45
63
  },
64
  {
65
  "epoch": 0.11,
66
- "learning_rate": 0.0001365469584575469,
67
- "loss": 1.3308,
68
  "step": 50
69
  },
70
  {
71
  "epoch": 0.12,
72
- "learning_rate": 0.0001361805346465575,
73
- "loss": 1.3708,
74
  "step": 55
75
  },
76
  {
77
  "epoch": 0.13,
78
- "learning_rate": 0.00013573357383530257,
79
- "loss": 1.1679,
80
  "step": 60
81
  },
82
  {
83
  "epoch": 0.14,
84
- "learning_rate": 0.00013520660867542708,
85
- "loss": 1.1438,
86
  "step": 65
87
  },
88
  {
89
  "epoch": 0.15,
90
- "learning_rate": 0.000134600267161271,
91
- "loss": 1.2393,
92
  "step": 70
93
  },
94
  {
95
- "epoch": 0.16,
96
- "learning_rate": 0.00013391527188147482,
97
- "loss": 1.3522,
98
  "step": 75
99
  },
100
  {
101
  "epoch": 0.18,
102
- "learning_rate": 0.0001331524391578593,
103
- "loss": 1.2583,
104
  "step": 80
105
  },
106
  {
107
  "epoch": 0.19,
108
- "learning_rate": 0.00013231267807259546,
109
- "loss": 1.2512,
110
  "step": 85
111
  },
112
  {
113
  "epoch": 0.2,
114
- "learning_rate": 0.00013139698938484037,
115
- "loss": 1.1431,
116
  "step": 90
117
  },
118
  {
119
  "epoch": 0.21,
120
- "learning_rate": 0.00013040646433810598,
121
- "loss": 1.2775,
122
  "step": 95
123
  },
124
  {
125
  "epoch": 0.22,
126
- "learning_rate": 0.0001293422833598104,
127
- "loss": 1.2136,
128
  "step": 100
129
  },
130
  {
131
  "epoch": 0.23,
132
- "learning_rate": 0.0001282057146545354,
133
- "loss": 1.1051,
134
  "step": 105
135
  },
136
  {
137
  "epoch": 0.24,
138
- "learning_rate": 0.00012699811269268664,
139
- "loss": 1.1814,
140
  "step": 110
141
  },
142
  {
143
  "epoch": 0.25,
144
- "learning_rate": 0.00012572091659634216,
145
- "loss": 1.1777,
146
  "step": 115
147
  },
148
  {
149
  "epoch": 0.26,
150
- "learning_rate": 0.00012437564842422737,
151
- "loss": 1.2222,
152
  "step": 120
153
  },
154
  {
155
- "epoch": 0.27,
156
- "learning_rate": 0.0001229639113578445,
157
- "loss": 1.1447,
158
  "step": 125
159
  },
160
  {
161
  "epoch": 0.29,
162
- "learning_rate": 0.00012148738779093613,
163
- "loss": 1.2241,
164
  "step": 130
165
  },
166
  {
167
  "epoch": 0.3,
168
- "learning_rate": 0.00011994783732453708,
169
- "loss": 1.3034,
170
  "step": 135
171
  },
172
  {
173
  "epoch": 0.31,
174
- "learning_rate": 0.00011834709467003608,
175
- "loss": 1.1023,
176
  "step": 140
177
  },
178
  {
179
  "epoch": 0.32,
180
- "learning_rate": 0.0001166870674627019,
181
- "loss": 1.2467,
182
  "step": 145
183
  },
184
  {
185
  "epoch": 0.33,
186
- "learning_rate": 0.0001149697339883305,
187
- "loss": 1.4339,
188
  "step": 150
189
  },
190
  {
191
  "epoch": 0.34,
192
- "learning_rate": 0.00011319714082567578,
193
- "loss": 1.3216,
194
  "step": 155
195
  },
196
  {
197
  "epoch": 0.35,
198
- "learning_rate": 0.00011137140040750936,
199
- "loss": 1.2588,
200
  "step": 160
201
  },
202
  {
203
  "epoch": 0.36,
204
- "learning_rate": 0.00010949468850318863,
205
- "loss": 1.1163,
206
  "step": 165
207
  },
208
  {
209
  "epoch": 0.37,
210
- "learning_rate": 0.00010756924162575738,
211
- "loss": 1.2174,
212
  "step": 170
213
  },
214
  {
215
- "epoch": 0.38,
216
- "learning_rate": 0.00010559735436664466,
217
- "loss": 1.1407,
218
  "step": 175
219
  },
220
  {
221
  "epoch": 0.4,
222
- "learning_rate": 0.00010358137666116407,
223
- "loss": 1.2427,
224
  "step": 180
225
  },
226
  {
227
  "epoch": 0.41,
228
- "learning_rate": 0.00010152371098804076,
229
- "loss": 1.2078,
230
  "step": 185
231
  },
232
  {
233
  "epoch": 0.42,
234
- "learning_rate": 9.942680950634789e-05,
235
- "loss": 1.1613,
236
  "step": 190
237
  },
238
  {
239
  "epoch": 0.43,
240
- "learning_rate": 9.729317113320387e-05,
241
- "loss": 1.2122,
242
  "step": 195
243
  },
244
  {
245
  "epoch": 0.44,
246
- "learning_rate": 9.51253385657818e-05,
247
- "loss": 1.2021,
248
  "step": 200
249
  },
250
  {
251
  "epoch": 0.45,
252
- "learning_rate": 9.292589525111788e-05,
253
- "loss": 1.1738,
254
  "step": 205
255
  },
256
  {
257
  "epoch": 0.46,
258
- "learning_rate": 9.069746230737676e-05,
259
- "loss": 1.4022,
260
  "step": 210
261
  },
262
  {
263
  "epoch": 0.47,
264
- "learning_rate": 8.844269540020829e-05,
265
- "loss": 1.068,
266
  "step": 215
267
  },
268
  {
269
  "epoch": 0.48,
270
- "learning_rate": 8.616428157794787e-05,
271
- "loss": 1.227,
272
  "step": 220
273
  },
274
  {
275
- "epoch": 0.49,
276
- "learning_rate": 8.386493606940273e-05,
277
- "loss": 1.3073,
278
  "step": 225
279
  },
280
  {
281
  "epoch": 0.51,
282
- "learning_rate": 8.1547399048071e-05,
283
- "loss": 1.3174,
284
  "step": 230
285
  },
286
  {
287
  "epoch": 0.52,
288
- "learning_rate": 7.921443236661364e-05,
289
- "loss": 1.2401,
290
  "step": 235
291
  },
292
  {
293
  "epoch": 0.53,
294
- "learning_rate": 7.68688162655159e-05,
295
- "loss": 1.0838,
296
  "step": 240
297
  },
298
  {
299
  "epoch": 0.54,
300
- "learning_rate": 7.451334605981065e-05,
301
- "loss": 1.2547,
302
  "step": 245
303
  },
304
  {
305
  "epoch": 0.55,
306
- "learning_rate": 7.215082880785731e-05,
307
- "loss": 1.168,
308
  "step": 250
309
  },
310
  {
311
  "epoch": 0.56,
312
- "learning_rate": 6.978407996610963e-05,
313
- "loss": 1.1954,
314
  "step": 255
315
  },
316
  {
317
  "epoch": 0.57,
318
- "learning_rate": 6.741592003389071e-05,
319
- "loss": 1.1562,
320
  "step": 260
321
  },
322
  {
323
  "epoch": 0.58,
324
- "learning_rate": 6.504917119214301e-05,
325
- "loss": 1.129,
326
  "step": 265
327
  },
328
  {
329
  "epoch": 0.59,
330
- "learning_rate": 6.268665394019066e-05,
331
- "loss": 1.1188,
332
  "step": 270
333
  },
334
  {
335
- "epoch": 0.6,
336
- "learning_rate": 6.0331183734484437e-05,
337
- "loss": 1.2232,
338
  "step": 275
339
  },
340
  {
341
  "epoch": 0.62,
342
- "learning_rate": 5.7985567633387656e-05,
343
- "loss": 1.1544,
344
  "step": 280
345
  },
346
  {
347
  "epoch": 0.63,
348
- "learning_rate": 5.56526009519303e-05,
349
- "loss": 1.169,
350
  "step": 285
351
  },
352
  {
353
  "epoch": 0.64,
354
- "learning_rate": 5.333506393059759e-05,
355
- "loss": 1.1042,
356
  "step": 290
357
  },
358
  {
359
  "epoch": 0.65,
360
- "learning_rate": 5.1035718422052454e-05,
361
- "loss": 1.2713,
362
  "step": 295
363
  },
364
  {
365
  "epoch": 0.66,
366
- "learning_rate": 4.8757304599792025e-05,
367
- "loss": 1.0417,
368
  "step": 300
369
  },
370
  {
371
  "epoch": 0.67,
372
- "learning_rate": 4.6502537692623556e-05,
373
- "loss": 1.3032,
374
  "step": 305
375
  },
376
  {
377
  "epoch": 0.68,
378
- "learning_rate": 4.4274104748882436e-05,
379
- "loss": 1.1426,
380
  "step": 310
381
  },
382
  {
383
  "epoch": 0.69,
384
- "learning_rate": 4.20746614342185e-05,
385
- "loss": 1.132,
386
  "step": 315
387
  },
388
  {
389
  "epoch": 0.7,
390
- "learning_rate": 3.9906828866797315e-05,
391
- "loss": 1.0621,
392
  "step": 320
393
  },
394
  {
395
- "epoch": 0.71,
396
- "learning_rate": 3.7773190493652406e-05,
397
- "loss": 1.2695,
398
  "step": 325
399
  },
400
  {
401
  "epoch": 0.73,
402
- "learning_rate": 3.567628901195954e-05,
403
- "loss": 1.2674,
404
  "step": 330
405
  },
406
  {
407
  "epoch": 0.74,
408
- "learning_rate": 3.3618623338837045e-05,
409
- "loss": 1.2634,
410
  "step": 335
411
  },
412
  {
413
  "epoch": 0.75,
414
- "learning_rate": 3.160264563335562e-05,
415
- "loss": 1.3034,
416
  "step": 340
417
  },
418
  {
419
  "epoch": 0.76,
420
- "learning_rate": 2.963075837424291e-05,
421
- "loss": 1.1978,
422
  "step": 345
423
  },
424
  {
425
  "epoch": 0.77,
426
- "learning_rate": 2.770531149681164e-05,
427
- "loss": 1.147,
428
  "step": 350
429
  },
430
  {
431
  "epoch": 0.78,
432
- "learning_rate": 2.5828599592490913e-05,
433
- "loss": 1.1312,
434
  "step": 355
435
  },
436
  {
437
  "epoch": 0.79,
438
- "learning_rate": 2.400285917432448e-05,
439
- "loss": 1.0567,
440
  "step": 360
441
  },
442
  {
443
  "epoch": 0.8,
444
- "learning_rate": 2.223026601166975e-05,
445
- "loss": 1.223,
446
  "step": 365
447
  },
448
  {
449
  "epoch": 0.81,
450
- "learning_rate": 2.0512932537299025e-05,
451
- "loss": 1.0728,
452
  "step": 370
453
  },
454
  {
455
- "epoch": 0.82,
456
- "learning_rate": 1.885290532996415e-05,
457
- "loss": 1.0335,
458
  "step": 375
459
  },
460
  {
461
  "epoch": 0.84,
462
- "learning_rate": 1.7252162675463158e-05,
463
- "loss": 1.1881,
464
  "step": 380
465
  },
466
  {
467
  "epoch": 0.85,
468
- "learning_rate": 1.5712612209064698e-05,
469
- "loss": 1.2686,
470
  "step": 385
471
  },
472
  {
473
  "epoch": 0.86,
474
- "learning_rate": 1.423608864215572e-05,
475
- "loss": 1.1489,
476
  "step": 390
477
  },
478
  {
479
  "epoch": 0.87,
480
- "learning_rate": 1.282435157577283e-05,
481
- "loss": 1.1948,
482
  "step": 395
483
  },
484
  {
485
  "epoch": 0.88,
486
- "learning_rate": 1.147908340365801e-05,
487
- "loss": 1.0644,
488
  "step": 400
489
  },
490
  {
491
  "epoch": 0.89,
492
- "learning_rate": 1.020188730731355e-05,
493
- "loss": 1.3175,
494
  "step": 405
495
  },
496
  {
497
  "epoch": 0.9,
498
- "learning_rate": 8.99428534546479e-06,
499
- "loss": 1.2393,
500
  "step": 410
501
  },
502
  {
503
  "epoch": 0.91,
504
- "learning_rate": 7.857716640189754e-06,
505
- "loss": 1.2108,
506
  "step": 415
507
  },
508
  {
509
- "epoch": 0.92,
510
- "learning_rate": 6.793535661894595e-06,
511
- "loss": 1.0917,
512
  "step": 420
513
  },
514
  {
515
- "epoch": 0.93,
516
- "learning_rate": 5.80301061515975e-06,
517
- "loss": 1.2168,
518
  "step": 425
519
  },
520
  {
521
  "epoch": 0.95,
522
- "learning_rate": 4.8873219274046634e-06,
523
- "loss": 0.9956,
524
  "step": 430
525
  },
526
  {
527
  "epoch": 0.96,
528
- "learning_rate": 4.047560842141143e-06,
529
- "loss": 1.2478,
530
  "step": 435
531
  },
532
  {
533
  "epoch": 0.97,
534
- "learning_rate": 3.2847281185252856e-06,
535
- "loss": 1.2265,
536
  "step": 440
537
  },
538
  {
539
  "epoch": 0.98,
540
- "learning_rate": 2.5997328387290914e-06,
541
- "loss": 1.1521,
542
  "step": 445
543
  },
544
  {
545
  "epoch": 0.99,
546
- "learning_rate": 1.9933913245729997e-06,
547
- "loss": 1.1974,
548
  "step": 450
549
  },
550
  {
551
  "epoch": 1.0,
552
- "learning_rate": 1.4664261646974962e-06,
553
- "loss": 1.166,
554
- "step": 455
 
 
555
  },
556
  {
557
  "epoch": 1.0,
558
- "eval_loss": 0.8371484875679016,
559
- "eval_runtime": 28.6472,
560
- "eval_samples_per_second": 22.341,
561
- "eval_steps_per_second": 2.793,
562
  "step": 455
563
  },
564
  {
565
- "epoch": 1.02,
566
- "learning_rate": 3.313560072903732e-06,
567
- "loss": 0.9642,
568
  "step": 460
569
  },
570
  {
571
- "epoch": 1.03,
572
- "learning_rate": 4.083020186586042e-06,
573
- "loss": 1.0926,
574
  "step": 465
575
  },
576
  {
577
  "epoch": 1.04,
578
- "learning_rate": 4.930046740852481e-06,
579
- "loss": 1.1107,
580
  "step": 470
581
  },
582
  {
583
  "epoch": 1.05,
584
- "learning_rate": 5.85362138619196e-06,
585
- "loss": 1.0343,
586
  "step": 475
587
  },
588
  {
589
  "epoch": 1.06,
590
- "learning_rate": 6.852633742079277e-06,
591
- "loss": 1.0895,
592
  "step": 480
593
  },
594
  {
595
  "epoch": 1.07,
596
- "learning_rate": 7.925882731949876e-06,
597
- "loss": 1.0942,
598
  "step": 485
599
  },
600
  {
601
  "epoch": 1.08,
602
- "learning_rate": 9.072078027207483e-06,
603
- "loss": 1.0349,
604
  "step": 490
605
  },
606
  {
607
  "epoch": 1.09,
608
- "learning_rate": 1.0289841598539205e-05,
609
- "loss": 1.1153,
610
  "step": 495
611
  },
612
  {
613
  "epoch": 1.1,
614
- "learning_rate": 1.1577709372675019e-05,
615
- "loss": 1.1266,
616
  "step": 500
617
  },
618
  {
619
  "epoch": 1.11,
620
- "learning_rate": 1.2934132992586013e-05,
621
- "loss": 1.0478,
622
  "step": 505
623
  },
624
  {
625
- "epoch": 1.13,
626
- "learning_rate": 1.4357481679018476e-05,
627
- "loss": 1.0834,
628
  "step": 510
629
  },
630
  {
631
- "epoch": 1.14,
632
- "learning_rate": 1.5846044191123214e-05,
633
- "loss": 0.9411,
634
  "step": 515
635
  },
636
  {
637
  "epoch": 1.15,
638
- "learning_rate": 1.739803088381975e-05,
639
- "loss": 1.1014,
640
  "step": 520
641
  },
642
  {
643
  "epoch": 1.16,
644
- "learning_rate": 1.9011575859416407e-05,
645
- "loss": 1.0989,
646
  "step": 525
647
  },
648
  {
649
  "epoch": 1.17,
650
- "learning_rate": 2.0684739210917447e-05,
651
- "loss": 0.9124,
652
  "step": 530
653
  },
654
  {
655
  "epoch": 1.18,
656
- "learning_rate": 2.2415509354303698e-05,
657
- "loss": 1.0756,
658
  "step": 535
659
  },
660
  {
661
  "epoch": 1.19,
662
- "learning_rate": 2.4201805446981234e-05,
663
- "loss": 1.0015,
664
  "step": 540
665
  },
666
  {
667
  "epoch": 1.2,
668
- "learning_rate": 2.604147988950966e-05,
669
- "loss": 1.2663,
670
  "step": 545
671
  },
672
  {
673
  "epoch": 1.21,
674
- "learning_rate": 2.793232090758389e-05,
675
- "loss": 1.1183,
676
  "step": 550
677
  },
678
  {
679
- "epoch": 1.23,
680
- "learning_rate": 2.9872055211163767e-05,
681
- "loss": 1.2909,
682
  "step": 555
683
  },
684
  {
685
- "epoch": 1.24,
686
- "learning_rate": 3.185835072757725e-05,
687
- "loss": 1.1669,
688
  "step": 560
689
  },
690
  {
691
- "epoch": 1.25,
692
- "learning_rate": 3.388881940527942e-05,
693
- "loss": 1.062,
694
  "step": 565
695
  },
696
  {
697
  "epoch": 1.26,
698
- "learning_rate": 3.596102008492985e-05,
699
- "loss": 1.1253,
700
  "step": 570
701
  },
702
  {
703
  "epoch": 1.27,
704
- "learning_rate": 3.807246143430314e-05,
705
- "loss": 1.1079,
706
  "step": 575
707
  },
708
  {
709
  "epoch": 1.28,
710
- "learning_rate": 4.02206049435268e-05,
711
- "loss": 1.2146,
712
  "step": 580
713
  },
714
  {
715
  "epoch": 1.29,
716
- "learning_rate": 4.2402867977052555e-05,
717
- "loss": 1.0767,
718
  "step": 585
719
  },
720
  {
721
  "epoch": 1.3,
722
- "learning_rate": 4.461662687864929e-05,
723
- "loss": 1.0771,
724
  "step": 590
725
  },
726
  {
727
  "epoch": 1.31,
728
- "learning_rate": 4.685922012574009e-05,
729
- "loss": 1.1997,
730
  "step": 595
731
  },
732
  {
733
  "epoch": 1.32,
734
- "learning_rate": 4.912795152925409e-05,
735
- "loss": 1.107,
736
  "step": 600
737
  },
738
  {
739
- "epoch": 1.34,
740
- "learning_rate": 5.14200934751585e-05,
741
- "loss": 1.1217,
742
  "step": 605
743
  },
744
  {
745
- "epoch": 1.35,
746
- "learning_rate": 5.373289020376694e-05,
747
- "loss": 1.0955,
748
  "step": 610
749
  },
750
  {
751
- "epoch": 1.36,
752
- "learning_rate": 5.606356112288902e-05,
753
- "loss": 1.1355,
754
  "step": 615
755
  },
756
  {
757
  "epoch": 1.37,
758
- "learning_rate": 5.840930415084469e-05,
759
- "loss": 1.2501,
760
  "step": 620
761
  },
762
  {
763
  "epoch": 1.38,
764
- "learning_rate": 6.0767299085297635e-05,
765
- "loss": 1.0553,
766
  "step": 625
767
  },
768
  {
769
  "epoch": 1.39,
770
- "learning_rate": 6.313471099388376e-05,
771
- "loss": 1.0718,
772
  "step": 630
773
  },
774
  {
775
  "epoch": 1.4,
776
- "learning_rate": 6.550869362255213e-05,
777
- "loss": 1.2052,
778
  "step": 635
779
  },
780
  {
781
  "epoch": 1.41,
782
- "learning_rate": 6.788639281751474e-05,
783
- "loss": 0.9941,
784
  "step": 640
785
  },
786
  {
787
  "epoch": 1.42,
788
- "learning_rate": 7.026494995668396e-05,
789
- "loss": 1.0171,
790
  "step": 645
791
  },
792
  {
793
  "epoch": 1.43,
794
- "learning_rate": 7.264150538649964e-05,
795
- "loss": 1.0264,
796
  "step": 650
797
  },
798
  {
799
- "epoch": 1.45,
800
- "learning_rate": 7.501320185998764e-05,
801
- "loss": 0.9926,
802
  "step": 655
803
  },
804
  {
805
- "epoch": 1.46,
806
- "learning_rate": 7.737718797191607e-05,
807
- "loss": 1.0749,
808
  "step": 660
809
  },
810
  {
811
- "epoch": 1.47,
812
- "learning_rate": 7.97306215869479e-05,
813
- "loss": 1.0997,
814
  "step": 665
815
  },
816
  {
817
  "epoch": 1.48,
818
- "learning_rate": 8.207067325662808e-05,
819
- "loss": 1.1054,
820
  "step": 670
821
  },
822
  {
823
  "epoch": 1.49,
824
- "learning_rate": 8.439452962113818e-05,
825
- "loss": 1.2298,
826
  "step": 675
827
  },
828
  {
829
  "epoch": 1.5,
830
- "learning_rate": 8.669939679168881e-05,
831
- "loss": 1.0679,
832
  "step": 680
833
  },
834
  {
835
  "epoch": 1.51,
836
- "learning_rate": 8.898250370951158e-05,
837
- "loss": 0.93,
838
  "step": 685
839
  },
840
  {
841
  "epoch": 1.52,
842
- "learning_rate": 9.124110547741169e-05,
843
- "loss": 1.0726,
844
  "step": 690
845
  },
846
  {
847
  "epoch": 1.53,
848
- "learning_rate": 9.347248665985043e-05,
849
- "loss": 1.038,
850
  "step": 695
851
  },
852
  {
853
- "epoch": 1.55,
854
- "learning_rate": 9.567396454761656e-05,
855
- "loss": 1.219,
856
  "step": 700
857
  },
858
  {
859
- "epoch": 1.56,
860
- "learning_rate": 9.78428923831546e-05,
861
- "loss": 1.1036,
862
  "step": 705
863
  },
864
  {
865
- "epoch": 1.57,
866
- "learning_rate": 9.997666254266687e-05,
867
- "loss": 1.2553,
868
  "step": 710
869
  },
870
  {
871
- "epoch": 1.58,
872
- "learning_rate": 0.0001020727096711577,
873
- "loss": 0.9944,
874
  "step": 715
875
  },
876
  {
877
  "epoch": 1.59,
878
- "learning_rate": 0.00010412851376667537,
879
- "loss": 1.0103,
880
  "step": 720
881
  },
882
  {
883
  "epoch": 1.6,
884
- "learning_rate": 0.00010614160321001933,
885
- "loss": 1.1856,
886
  "step": 725
887
  },
888
  {
889
  "epoch": 1.61,
890
- "learning_rate": 0.00010810955773627152,
891
- "loss": 1.1214,
892
  "step": 730
893
  },
894
  {
895
  "epoch": 1.62,
896
- "learning_rate": 0.00011003001134460133,
897
- "loss": 1.226,
898
  "step": 735
899
  },
900
  {
901
  "epoch": 1.63,
902
- "learning_rate": 0.00011190065514282273,
903
- "loss": 1.1587,
904
  "step": 740
905
  },
906
  {
907
  "epoch": 1.64,
908
- "learning_rate": 0.00011371924012328539,
909
- "loss": 1.3559,
910
  "step": 745
911
  },
912
  {
913
- "epoch": 1.66,
914
- "learning_rate": 0.00011548357986678387,
915
- "loss": 1.0196,
916
  "step": 750
917
  },
918
  {
919
- "epoch": 1.67,
920
- "learning_rate": 0.00011719155317120334,
921
- "loss": 1.0605,
922
  "step": 755
923
  },
924
  {
925
- "epoch": 1.68,
926
- "learning_rate": 0.00011884110660177233,
927
- "loss": 1.3014,
928
  "step": 760
929
  },
930
  {
931
  "epoch": 1.69,
932
- "learning_rate": 0.0001204302569598269,
933
- "loss": 1.104,
934
  "step": 765
935
  },
936
  {
937
  "epoch": 1.7,
938
- "learning_rate": 0.00012195709366713928,
939
- "loss": 1.1758,
940
  "step": 770
941
  },
942
  {
943
  "epoch": 1.71,
944
- "learning_rate": 0.00012341978106294302,
945
- "loss": 1.1321,
946
  "step": 775
947
  },
948
  {
949
  "epoch": 1.72,
950
- "learning_rate": 0.00012481656061087584,
951
- "loss": 1.0998,
952
  "step": 780
953
  },
954
  {
955
  "epoch": 1.73,
956
- "learning_rate": 0.00012614575301320556,
957
- "loss": 1.2422,
958
  "step": 785
959
  },
960
  {
961
  "epoch": 1.74,
962
- "learning_rate": 0.00012740576022979148,
963
- "loss": 1.0643,
964
  "step": 790
965
  },
966
  {
967
  "epoch": 1.75,
968
- "learning_rate": 0.00012859506739935084,
969
- "loss": 1.224,
970
  "step": 795
971
  },
972
  {
973
- "epoch": 1.77,
974
- "learning_rate": 0.0001297122446607173,
975
- "loss": 1.2715,
976
  "step": 800
977
  },
978
  {
979
- "epoch": 1.78,
980
- "learning_rate": 0.00013075594887191635,
981
- "loss": 1.2678,
982
  "step": 805
983
  },
984
  {
985
- "epoch": 1.79,
986
- "learning_rate": 0.00013172492522497582,
987
- "loss": 1.3525,
988
  "step": 810
989
  },
990
  {
991
  "epoch": 1.8,
992
- "learning_rate": 0.00013261800875453292,
993
- "loss": 1.1699,
994
  "step": 815
995
  },
996
  {
997
  "epoch": 1.81,
998
- "learning_rate": 0.0001334341257384344,
999
- "loss": 1.2546,
1000
  "step": 820
1001
  },
1002
  {
1003
  "epoch": 1.82,
1004
- "learning_rate": 0.00013417229498863045,
1005
- "loss": 1.1295,
1006
  "step": 825
1007
  },
1008
  {
1009
  "epoch": 1.83,
1010
- "learning_rate": 0.00013483162903082528,
1011
- "loss": 1.0542,
1012
  "step": 830
1013
  },
1014
  {
1015
  "epoch": 1.84,
1016
- "learning_rate": 0.00013541133517145263,
1017
- "loss": 1.1614,
1018
  "step": 835
1019
  },
1020
  {
1021
  "epoch": 1.85,
1022
- "learning_rate": 0.000135910716450703,
1023
- "loss": 1.2422,
1024
  "step": 840
1025
  },
1026
  {
1027
- "epoch": 1.87,
1028
- "learning_rate": 0.00013632917248045546,
1029
- "loss": 1.3636,
1030
  "step": 845
1031
  },
1032
  {
1033
- "epoch": 1.88,
1034
- "learning_rate": 0.00013666620016610032,
1035
- "loss": 1.1613,
1036
  "step": 850
1037
  },
1038
  {
1039
- "epoch": 1.89,
1040
- "learning_rate": 0.00013692139431139235,
1041
- "loss": 1.1166,
1042
  "step": 855
1043
  },
1044
  {
1045
- "epoch": 1.9,
1046
- "learning_rate": 0.00013709444810560428,
1047
- "loss": 1.3142,
1048
  "step": 860
1049
  },
1050
  {
1051
  "epoch": 1.91,
1052
- "learning_rate": 0.00013718515349239374,
1053
- "loss": 1.2432,
1054
  "step": 865
1055
  },
1056
  {
1057
  "epoch": 1.92,
1058
- "learning_rate": 0.0001371934014199418,
1059
- "loss": 1.2136,
1060
  "step": 870
1061
  },
1062
  {
1063
  "epoch": 1.93,
1064
- "learning_rate": 0.00013711918197206236,
1065
- "loss": 1.244,
1066
  "step": 875
1067
  },
1068
  {
1069
  "epoch": 1.94,
1070
- "learning_rate": 0.00013696258438012369,
1071
- "loss": 1.197,
1072
  "step": 880
1073
  },
1074
  {
1075
  "epoch": 1.95,
1076
- "learning_rate": 0.0001367237969157692,
1077
- "loss": 1.2077,
1078
  "step": 885
1079
  },
1080
  {
1081
  "epoch": 1.96,
1082
- "learning_rate": 0.0001364031066645648,
1083
- "loss": 1.219,
1084
  "step": 890
1085
  },
1086
  {
1087
- "epoch": 1.98,
1088
- "learning_rate": 0.00013600089918084689,
1089
- "loss": 1.1498,
1090
  "step": 895
1091
  },
1092
- {
1093
- "epoch": 1.99,
1094
- "learning_rate": 0.00013551765802418306,
1095
- "loss": 1.3182,
1096
- "step": 900
1097
- },
1098
- {
1099
- "epoch": 2.0,
1100
- "learning_rate": 0.00013495396417800696,
1101
- "loss": 1.2463,
1102
- "step": 905
1103
- },
1104
- {
1105
- "epoch": 2.0,
1106
- "eval_loss": 0.7939189076423645,
1107
- "eval_runtime": 29.7555,
1108
- "eval_samples_per_second": 22.08,
1109
- "eval_steps_per_second": 2.789,
1110
- "step": 906
1111
- },
1112
  {
1113
  "epoch": 1.98,
1114
- "learning_rate": 7.7941032128002e-05,
1115
- "loss": 1.1586,
1116
- "step": 910
1117
  },
1118
  {
1119
  "epoch": 1.99,
1120
- "learning_rate": 8.025585241296564e-05,
1121
- "loss": 1.0424,
1122
- "step": 915
1123
- },
1124
- {
1125
- "epoch": 2.0,
1126
- "learning_rate": 8.255708249541132e-05,
1127
- "loss": 0.9012,
1128
- "step": 920
1129
  },
1130
  {
1131
  "epoch": 2.0,
1132
- "eval_loss": 0.6822583675384521,
1133
- "eval_runtime": 26.4465,
1134
- "eval_samples_per_second": 22.839,
1135
- "eval_steps_per_second": 2.874,
1136
- "step": 920
1137
  }
1138
  ],
1139
- "max_steps": 1380,
1140
  "num_train_epochs": 3,
1141
- "total_flos": 959725633536000.0,
1142
  "trial_name": null,
1143
  "trial_params": null
1144
  }
 
1
  {
2
+ "best_metric": 0.8025060296058655,
3
+ "best_model_checkpoint": "output/eminem/checkpoint-908",
4
  "epoch": 2.0,
5
+ "global_step": 908,
6
  "is_hyper_param_search": false,
7
  "is_local_process_zero": true,
8
  "is_world_process_zero": true,
9
  "log_history": [
10
  {
11
  "epoch": 0.01,
12
+ "learning_rate": 1.3299239388716378e-07,
13
+ "loss": 0.9607,
14
  "step": 5
15
  },
16
  {
17
  "epoch": 0.02,
18
+ "learning_rate": 3.216610996650382e-07,
19
+ "loss": 0.966,
20
  "step": 10
21
  },
22
  {
23
  "epoch": 0.03,
24
+ "learning_rate": 5.920572031213835e-07,
25
+ "loss": 0.9796,
26
  "step": 15
27
  },
28
  {
29
  "epoch": 0.04,
30
+ "learning_rate": 9.43857047145701e-07,
31
+ "loss": 0.8739,
32
  "step": 20
33
  },
34
  {
35
+ "epoch": 0.06,
36
+ "learning_rate": 1.3766395364652307e-06,
37
+ "loss": 1.0099,
38
  "step": 25
39
  },
40
  {
41
  "epoch": 0.07,
42
+ "learning_rate": 1.88988664168433e-06,
43
+ "loss": 0.9153,
44
  "step": 30
45
  },
46
  {
47
  "epoch": 0.08,
48
+ "learning_rate": 2.482984019353729e-06,
49
+ "loss": 1.0177,
50
  "step": 35
51
  },
52
  {
53
  "epoch": 0.09,
54
+ "learning_rate": 3.1552217473210573e-06,
55
+ "loss": 0.9735,
56
  "step": 40
57
  },
58
  {
59
  "epoch": 0.1,
60
+ "learning_rate": 3.905795174492246e-06,
61
+ "loss": 0.8828,
62
  "step": 45
63
  },
64
  {
65
  "epoch": 0.11,
66
+ "learning_rate": 4.733805883976127e-06,
67
+ "loss": 1.0567,
68
  "step": 50
69
  },
70
  {
71
  "epoch": 0.12,
72
+ "learning_rate": 5.638262768466535e-06,
73
+ "loss": 0.9174,
74
  "step": 55
75
  },
76
  {
77
  "epoch": 0.13,
78
+ "learning_rate": 6.618083216572472e-06,
79
+ "loss": 0.8727,
80
  "step": 60
81
  },
82
  {
83
  "epoch": 0.14,
84
+ "learning_rate": 7.6720944086741e-06,
85
+ "loss": 0.9884,
86
  "step": 65
87
  },
88
  {
89
  "epoch": 0.15,
90
+ "learning_rate": 8.799034720759847e-06,
91
+ "loss": 0.9629,
92
  "step": 70
93
  },
94
  {
95
+ "epoch": 0.17,
96
+ "learning_rate": 9.997555234555936e-06,
97
+ "loss": 0.927,
98
  "step": 75
99
  },
100
  {
101
  "epoch": 0.18,
102
+ "learning_rate": 1.1266221352147439e-05,
103
+ "loss": 0.9882,
104
  "step": 80
105
  },
106
  {
107
  "epoch": 0.19,
108
+ "learning_rate": 1.260351451315264e-05,
109
+ "loss": 1.1071,
110
  "step": 85
111
  },
112
  {
113
  "epoch": 0.2,
114
+ "learning_rate": 1.4007834012403411e-05,
115
+ "loss": 0.9968,
116
  "step": 90
117
  },
118
  {
119
  "epoch": 0.21,
120
+ "learning_rate": 1.5477498915945732e-05,
121
+ "loss": 0.8446,
122
  "step": 95
123
  },
124
  {
125
  "epoch": 0.22,
126
+ "learning_rate": 1.7010750073074396e-05,
127
+ "loss": 1.1172,
128
  "step": 100
129
  },
130
  {
131
  "epoch": 0.23,
132
+ "learning_rate": 1.8605752221991424e-05,
133
+ "loss": 1.0285,
134
  "step": 105
135
  },
136
  {
137
  "epoch": 0.24,
138
+ "learning_rate": 2.026059618656475e-05,
139
+ "loss": 1.1558,
140
  "step": 110
141
  },
142
  {
143
  "epoch": 0.25,
144
+ "learning_rate": 2.19733011615644e-05,
145
+ "loss": 1.0706,
146
  "step": 115
147
  },
148
  {
149
  "epoch": 0.26,
150
+ "learning_rate": 2.3741817083633492e-05,
151
+ "loss": 0.9304,
152
  "step": 120
153
  },
154
  {
155
+ "epoch": 0.28,
156
+ "learning_rate": 2.5564027085162985e-05,
157
+ "loss": 1.0108,
158
  "step": 125
159
  },
160
  {
161
  "epoch": 0.29,
162
+ "learning_rate": 2.7437750028125123e-05,
163
+ "loss": 1.022,
164
  "step": 130
165
  },
166
  {
167
  "epoch": 0.3,
168
+ "learning_rate": 2.9360743114840418e-05,
169
+ "loss": 0.9518,
170
  "step": 135
171
  },
172
  {
173
  "epoch": 0.31,
174
+ "learning_rate": 3.133070457254507e-05,
175
+ "loss": 0.9963,
176
  "step": 140
177
  },
178
  {
179
  "epoch": 0.32,
180
+ "learning_rate": 3.334527640855787e-05,
181
+ "loss": 0.9435,
182
  "step": 145
183
  },
184
  {
185
  "epoch": 0.33,
186
+ "learning_rate": 3.5402047232733056e-05,
187
+ "loss": 0.955,
188
  "step": 150
189
  },
190
  {
191
  "epoch": 0.34,
192
+ "learning_rate": 3.749855514383242e-05,
193
+ "loss": 0.9793,
194
  "step": 155
195
  },
196
  {
197
  "epoch": 0.35,
198
+ "learning_rate": 3.963229067635804e-05,
199
+ "loss": 0.9688,
200
  "step": 160
201
  },
202
  {
203
  "epoch": 0.36,
204
+ "learning_rate": 4.180069980431374e-05,
205
+ "loss": 0.9478,
206
  "step": 165
207
  },
208
  {
209
  "epoch": 0.37,
210
+ "learning_rate": 4.400118699831312e-05,
211
+ "loss": 0.9073,
212
  "step": 170
213
  },
214
  {
215
+ "epoch": 0.39,
216
+ "learning_rate": 4.62311183323573e-05,
217
+ "loss": 0.8938,
218
  "step": 175
219
  },
220
  {
221
  "epoch": 0.4,
222
+ "learning_rate": 4.848782463657674e-05,
223
+ "loss": 1.0837,
224
  "step": 180
225
  },
226
  {
227
  "epoch": 0.41,
228
+ "learning_rate": 5.076860469215926e-05,
229
+ "loss": 0.964,
230
  "step": 185
231
  },
232
  {
233
  "epoch": 0.42,
234
+ "learning_rate": 5.307072846463586e-05,
235
+ "loss": 0.9684,
236
  "step": 190
237
  },
238
  {
239
  "epoch": 0.43,
240
+ "learning_rate": 5.539144037166391e-05,
241
+ "loss": 0.9341,
242
  "step": 195
243
  },
244
  {
245
  "epoch": 0.44,
246
+ "learning_rate": 5.772796258138569e-05,
247
+ "loss": 0.8384,
248
  "step": 200
249
  },
250
  {
251
  "epoch": 0.45,
252
+ "learning_rate": 6.007749833742388e-05,
253
+ "loss": 0.8859,
254
  "step": 205
255
  },
256
  {
257
  "epoch": 0.46,
258
+ "learning_rate": 6.243723530652334e-05,
259
+ "loss": 0.9459,
260
  "step": 210
261
  },
262
  {
263
  "epoch": 0.47,
264
+ "learning_rate": 6.480434894484296e-05,
265
+ "loss": 0.9548,
266
  "step": 215
267
  },
268
  {
269
  "epoch": 0.48,
270
+ "learning_rate": 6.717600587885728e-05,
271
+ "loss": 0.9153,
272
  "step": 220
273
  },
274
  {
275
+ "epoch": 0.5,
276
+ "learning_rate": 6.954936729683686e-05,
277
+ "loss": 0.9542,
278
  "step": 225
279
  },
280
  {
281
  "epoch": 0.51,
282
+ "learning_rate": 7.192159234682828e-05,
283
+ "loss": 0.9709,
284
  "step": 230
285
  },
286
  {
287
  "epoch": 0.52,
288
+ "learning_rate": 7.428984153708106e-05,
289
+ "loss": 0.9535,
290
  "step": 235
291
  },
292
  {
293
  "epoch": 0.53,
294
+ "learning_rate": 7.665128013484648e-05,
295
+ "loss": 0.9314,
296
  "step": 240
297
  },
298
  {
299
  "epoch": 0.54,
300
+ "learning_rate": 7.900308155947524e-05,
301
+ "loss": 0.9481,
302
  "step": 245
303
  },
304
  {
305
  "epoch": 0.55,
306
+ "learning_rate": 8.134243076576675e-05,
307
+ "loss": 1.0072,
308
  "step": 250
309
  },
310
  {
311
  "epoch": 0.56,
312
+ "learning_rate": 8.366652761350124e-05,
313
+ "loss": 0.9006,
314
  "step": 255
315
  },
316
  {
317
  "epoch": 0.57,
318
+ "learning_rate": 8.597259021913684e-05,
319
+ "loss": 1.1408,
320
  "step": 260
321
  },
322
  {
323
  "epoch": 0.58,
324
+ "learning_rate": 8.825785828564903e-05,
325
+ "loss": 0.8856,
326
  "step": 265
327
  },
328
  {
329
  "epoch": 0.59,
330
+ "learning_rate": 9.051959640653727e-05,
331
+ "loss": 0.8189,
332
  "step": 270
333
  },
334
  {
335
+ "epoch": 0.61,
336
+ "learning_rate": 9.275509734003362e-05,
337
+ "loss": 0.9642,
338
  "step": 275
339
  },
340
  {
341
  "epoch": 0.62,
342
+ "learning_rate": 9.496168524960374e-05,
343
+ "loss": 0.9964,
344
  "step": 280
345
  },
346
  {
347
  "epoch": 0.63,
348
+ "learning_rate": 9.713671890685066e-05,
349
+ "loss": 1.0822,
350
  "step": 285
351
  },
352
  {
353
  "epoch": 0.64,
354
+ "learning_rate": 9.927759485299768e-05,
355
+ "loss": 1.0476,
356
  "step": 290
357
  },
358
  {
359
  "epoch": 0.65,
360
+ "learning_rate": 0.00010138175051516164,
361
+ "loss": 1.182,
362
  "step": 295
363
  },
364
  {
365
  "epoch": 0.66,
366
+ "learning_rate": 0.0001034466672736827,
367
+ "loss": 1.0057,
368
  "step": 300
369
  },
370
  {
371
  "epoch": 0.67,
372
+ "learning_rate": 0.00010546987347685174,
373
+ "loss": 0.9401,
374
  "step": 305
375
  },
376
  {
377
  "epoch": 0.68,
378
+ "learning_rate": 0.00010744894739940985,
379
+ "loss": 1.1325,
380
  "step": 310
381
  },
382
  {
383
  "epoch": 0.69,
384
+ "learning_rate": 0.0001093815201412914,
385
+ "loss": 1.0879,
386
  "step": 315
387
  },
388
  {
389
  "epoch": 0.7,
390
+ "learning_rate": 0.00011126527846313655,
391
+ "loss": 1.048,
392
  "step": 320
393
  },
394
  {
395
+ "epoch": 0.72,
396
+ "learning_rate": 0.0001130979675551757,
397
+ "loss": 0.9551,
398
  "step": 325
399
  },
400
  {
401
  "epoch": 0.73,
402
+ "learning_rate": 0.00011487739373618278,
403
+ "loss": 0.9586,
404
  "step": 330
405
  },
406
  {
407
  "epoch": 0.74,
408
+ "learning_rate": 0.00011660142707925162,
409
+ "loss": 0.9103,
410
  "step": 335
411
  },
412
  {
413
  "epoch": 0.75,
414
+ "learning_rate": 0.00011826800396126513,
415
+ "loss": 1.0702,
416
  "step": 340
417
  },
418
  {
419
  "epoch": 0.76,
420
+ "learning_rate": 0.0001198751295329933,
421
+ "loss": 0.9897,
422
  "step": 345
423
  },
424
  {
425
  "epoch": 0.77,
426
+ "learning_rate": 0.00012142088010688453,
427
+ "loss": 0.9608,
428
  "step": 350
429
  },
430
  {
431
  "epoch": 0.78,
432
+ "learning_rate": 0.00012290340545966183,
433
+ "loss": 1.1265,
434
  "step": 355
435
  },
436
  {
437
  "epoch": 0.79,
438
+ "learning_rate": 0.00012432093104699707,
439
+ "loss": 1.0023,
440
  "step": 360
441
  },
442
  {
443
  "epoch": 0.8,
444
+ "learning_rate": 0.00012567176012759186,
445
+ "loss": 1.0231,
446
  "step": 365
447
  },
448
  {
449
  "epoch": 0.81,
450
+ "learning_rate": 0.0001269542757941335,
451
+ "loss": 1.1342,
452
  "step": 370
453
  },
454
  {
455
+ "epoch": 0.83,
456
+ "learning_rate": 0.00012816694290869156,
457
+ "loss": 1.0269,
458
  "step": 375
459
  },
460
  {
461
  "epoch": 0.84,
462
+ "learning_rate": 0.00012930830994023604,
463
+ "loss": 1.1784,
464
  "step": 380
465
  },
466
  {
467
  "epoch": 0.85,
468
+ "learning_rate": 0.0001303770107020853,
469
+ "loss": 1.1591,
470
  "step": 385
471
  },
472
  {
473
  "epoch": 0.86,
474
+ "learning_rate": 0.00013137176598719403,
475
+ "loss": 1.0692,
476
  "step": 390
477
  },
478
  {
479
  "epoch": 0.87,
480
+ "learning_rate": 0.00013229138509933172,
481
+ "loss": 0.9776,
482
  "step": 395
483
  },
484
  {
485
  "epoch": 0.88,
486
+ "learning_rate": 0.00013313476727831364,
487
+ "loss": 1.1472,
488
  "step": 400
489
  },
490
  {
491
  "epoch": 0.89,
492
+ "learning_rate": 0.0001339009030175844,
493
+ "loss": 1.1092,
494
  "step": 405
495
  },
496
  {
497
  "epoch": 0.9,
498
+ "learning_rate": 0.00013458887527256958,
499
+ "loss": 1.0475,
500
  "step": 410
501
  },
502
  {
503
  "epoch": 0.91,
504
+ "learning_rate": 0.00013519786055835514,
505
+ "loss": 1.0603,
506
  "step": 415
507
  },
508
  {
509
+ "epoch": 0.93,
510
+ "learning_rate": 0.0001357271299353754,
511
+ "loss": 1.0979,
512
  "step": 420
513
  },
514
  {
515
+ "epoch": 0.94,
516
+ "learning_rate": 0.00013617604988193791,
517
+ "loss": 1.1643,
518
  "step": 425
519
  },
520
  {
521
  "epoch": 0.95,
522
+ "learning_rate": 0.00013654408305253046,
523
+ "loss": 1.1787,
524
  "step": 430
525
  },
526
  {
527
  "epoch": 0.96,
528
+ "learning_rate": 0.00013683078892100969,
529
+ "loss": 1.0679,
530
  "step": 435
531
  },
532
  {
533
  "epoch": 0.97,
534
+ "learning_rate": 0.00013703582430790026,
535
+ "loss": 1.1034,
536
  "step": 440
537
  },
538
  {
539
  "epoch": 0.98,
540
+ "learning_rate": 0.00013715894379117124,
541
+ "loss": 1.2196,
542
  "step": 445
543
  },
544
  {
545
  "epoch": 0.99,
546
+ "learning_rate": 0.0001372,
547
+ "loss": 1.1973,
548
  "step": 450
549
  },
550
  {
551
  "epoch": 1.0,
552
+ "eval_loss": 0.8132181763648987,
553
+ "eval_runtime": 8.3954,
554
+ "eval_samples_per_second": 76.947,
555
+ "eval_steps_per_second": 9.648,
556
+ "step": 454
557
  },
558
  {
559
  "epoch": 1.0,
560
+ "learning_rate": 0.00013715894379117126,
561
+ "loss": 1.097,
 
 
562
  "step": 455
563
  },
564
  {
565
+ "epoch": 1.01,
566
+ "learning_rate": 0.0001370358243079003,
567
+ "loss": 1.0361,
568
  "step": 460
569
  },
570
  {
571
+ "epoch": 1.02,
572
+ "learning_rate": 0.00013683078892100974,
573
+ "loss": 1.0419,
574
  "step": 465
575
  },
576
  {
577
  "epoch": 1.04,
578
+ "learning_rate": 0.00013654408305253056,
579
+ "loss": 0.9943,
580
  "step": 470
581
  },
582
  {
583
  "epoch": 1.05,
584
+ "learning_rate": 0.00013617604988193802,
585
+ "loss": 0.8492,
586
  "step": 475
587
  },
588
  {
589
  "epoch": 1.06,
590
+ "learning_rate": 0.0001357271299353755,
591
+ "loss": 0.8659,
592
  "step": 480
593
  },
594
  {
595
  "epoch": 1.07,
596
+ "learning_rate": 0.00013519786055835527,
597
+ "loss": 1.0681,
598
  "step": 485
599
  },
600
  {
601
  "epoch": 1.08,
602
+ "learning_rate": 0.00013458887527256977,
603
+ "loss": 0.8057,
604
  "step": 490
605
  },
606
  {
607
  "epoch": 1.09,
608
+ "learning_rate": 0.0001339009030175846,
609
+ "loss": 0.8735,
610
  "step": 495
611
  },
612
  {
613
  "epoch": 1.1,
614
+ "learning_rate": 0.00013313476727831386,
615
+ "loss": 1.0159,
616
  "step": 500
617
  },
618
  {
619
  "epoch": 1.11,
620
+ "learning_rate": 0.00013229138509933196,
621
+ "loss": 1.0634,
622
  "step": 505
623
  },
624
  {
625
+ "epoch": 1.12,
626
+ "learning_rate": 0.0001313717659871943,
627
+ "loss": 1.1331,
628
  "step": 510
629
  },
630
  {
631
+ "epoch": 1.13,
632
+ "learning_rate": 0.00013037701070208558,
633
+ "loss": 1.0009,
634
  "step": 515
635
  },
636
  {
637
  "epoch": 1.15,
638
+ "learning_rate": 0.00012930830994023633,
639
+ "loss": 0.9342,
640
  "step": 520
641
  },
642
  {
643
  "epoch": 1.16,
644
+ "learning_rate": 0.0001281669429086919,
645
+ "loss": 0.9788,
646
  "step": 525
647
  },
648
  {
649
  "epoch": 1.17,
650
+ "learning_rate": 0.00012695427579413385,
651
+ "loss": 0.9793,
652
  "step": 530
653
  },
654
  {
655
  "epoch": 1.18,
656
+ "learning_rate": 0.0001256717601275922,
657
+ "loss": 1.0848,
658
  "step": 535
659
  },
660
  {
661
  "epoch": 1.19,
662
+ "learning_rate": 0.00012432093104699745,
663
+ "loss": 0.9297,
664
  "step": 540
665
  },
666
  {
667
  "epoch": 1.2,
668
+ "learning_rate": 0.00012290340545966224,
669
+ "loss": 1.1117,
670
  "step": 545
671
  },
672
  {
673
  "epoch": 1.21,
674
+ "learning_rate": 0.00012142088010688494,
675
+ "loss": 1.1474,
676
  "step": 550
677
  },
678
  {
679
+ "epoch": 1.22,
680
+ "learning_rate": 0.00011987512953299372,
681
+ "loss": 0.9244,
682
  "step": 555
683
  },
684
  {
685
+ "epoch": 1.23,
686
+ "learning_rate": 0.00011826800396126557,
687
+ "loss": 1.1159,
688
  "step": 560
689
  },
690
  {
691
+ "epoch": 1.24,
692
+ "learning_rate": 0.00011660142707925208,
693
+ "loss": 1.0104,
694
  "step": 565
695
  },
696
  {
697
  "epoch": 1.26,
698
+ "learning_rate": 0.00011487739373618326,
699
+ "loss": 1.0571,
700
  "step": 570
701
  },
702
  {
703
  "epoch": 1.27,
704
+ "learning_rate": 0.00011309796755517619,
705
+ "loss": 1.1123,
706
  "step": 575
707
  },
708
  {
709
  "epoch": 1.28,
710
+ "learning_rate": 0.00011126527846313704,
711
+ "loss": 0.8683,
712
  "step": 580
713
  },
714
  {
715
  "epoch": 1.29,
716
+ "learning_rate": 0.00010938152014129192,
717
+ "loss": 0.9522,
718
  "step": 585
719
  },
720
  {
721
  "epoch": 1.3,
722
+ "learning_rate": 0.00010744894739941038,
723
+ "loss": 0.9323,
724
  "step": 590
725
  },
726
  {
727
  "epoch": 1.31,
728
+ "learning_rate": 0.00010546987347685228,
729
+ "loss": 0.8976,
730
  "step": 595
731
  },
732
  {
733
  "epoch": 1.32,
734
+ "learning_rate": 0.00010344666727368325,
735
+ "loss": 0.8812,
736
  "step": 600
737
  },
738
  {
739
+ "epoch": 1.33,
740
+ "learning_rate": 0.00010138175051516221,
741
+ "loss": 0.9468,
742
  "step": 605
743
  },
744
  {
745
+ "epoch": 1.34,
746
+ "learning_rate": 9.927759485299826e-05,
747
+ "loss": 0.9565,
748
  "step": 610
749
  },
750
  {
751
+ "epoch": 1.35,
752
+ "learning_rate": 9.713671890685124e-05,
753
+ "loss": 1.0534,
754
  "step": 615
755
  },
756
  {
757
  "epoch": 1.37,
758
+ "learning_rate": 9.496168524960432e-05,
759
+ "loss": 1.1313,
760
  "step": 620
761
  },
762
  {
763
  "epoch": 1.38,
764
+ "learning_rate": 9.275509734003423e-05,
765
+ "loss": 1.1094,
766
  "step": 625
767
  },
768
  {
769
  "epoch": 1.39,
770
+ "learning_rate": 9.051959640653786e-05,
771
+ "loss": 1.0261,
772
  "step": 630
773
  },
774
  {
775
  "epoch": 1.4,
776
+ "learning_rate": 8.825785828564964e-05,
777
+ "loss": 1.0947,
778
  "step": 635
779
  },
780
  {
781
  "epoch": 1.41,
782
+ "learning_rate": 8.597259021913746e-05,
783
+ "loss": 1.014,
784
  "step": 640
785
  },
786
  {
787
  "epoch": 1.42,
788
+ "learning_rate": 8.366652761350186e-05,
789
+ "loss": 0.9513,
790
  "step": 645
791
  },
792
  {
793
  "epoch": 1.43,
794
+ "learning_rate": 8.134243076576737e-05,
795
+ "loss": 1.0445,
796
  "step": 650
797
  },
798
  {
799
+ "epoch": 1.44,
800
+ "learning_rate": 7.900308155947589e-05,
801
+ "loss": 0.9315,
802
  "step": 655
803
  },
804
  {
805
+ "epoch": 1.45,
806
+ "learning_rate": 7.665128013484713e-05,
807
+ "loss": 1.0082,
808
  "step": 660
809
  },
810
  {
811
+ "epoch": 1.46,
812
+ "learning_rate": 7.42898415370817e-05,
813
+ "loss": 1.0051,
814
  "step": 665
815
  },
816
  {
817
  "epoch": 1.48,
818
+ "learning_rate": 7.192159234682893e-05,
819
+ "loss": 1.0477,
820
  "step": 670
821
  },
822
  {
823
  "epoch": 1.49,
824
+ "learning_rate": 6.95493672968375e-05,
825
+ "loss": 1.0178,
826
  "step": 675
827
  },
828
  {
829
  "epoch": 1.5,
830
+ "learning_rate": 6.717600587885793e-05,
831
+ "loss": 0.9478,
832
  "step": 680
833
  },
834
  {
835
  "epoch": 1.51,
836
+ "learning_rate": 6.48043489448436e-05,
837
+ "loss": 1.0712,
838
  "step": 685
839
  },
840
  {
841
  "epoch": 1.52,
842
+ "learning_rate": 6.243723530652398e-05,
843
+ "loss": 1.0546,
844
  "step": 690
845
  },
846
  {
847
  "epoch": 1.53,
848
+ "learning_rate": 6.0077498337424516e-05,
849
+ "loss": 0.9993,
850
  "step": 695
851
  },
852
  {
853
+ "epoch": 1.54,
854
+ "learning_rate": 5.772796258138633e-05,
855
+ "loss": 1.0692,
856
  "step": 700
857
  },
858
  {
859
+ "epoch": 1.55,
860
+ "learning_rate": 5.5391440371664534e-05,
861
+ "loss": 0.784,
862
  "step": 705
863
  },
864
  {
865
+ "epoch": 1.56,
866
+ "learning_rate": 5.307072846463648e-05,
867
+ "loss": 1.0026,
868
  "step": 710
869
  },
870
  {
871
+ "epoch": 1.57,
872
+ "learning_rate": 5.0768604692159876e-05,
873
+ "loss": 1.0188,
874
  "step": 715
875
  },
876
  {
877
  "epoch": 1.59,
878
+ "learning_rate": 4.848782463657735e-05,
879
+ "loss": 1.1097,
880
  "step": 720
881
  },
882
  {
883
  "epoch": 1.6,
884
+ "learning_rate": 4.623111833235789e-05,
885
+ "loss": 0.8797,
886
  "step": 725
887
  },
888
  {
889
  "epoch": 1.61,
890
+ "learning_rate": 4.400118699831371e-05,
891
+ "loss": 0.9968,
892
  "step": 730
893
  },
894
  {
895
  "epoch": 1.62,
896
+ "learning_rate": 4.180069980431433e-05,
897
+ "loss": 0.9446,
898
  "step": 735
899
  },
900
  {
901
  "epoch": 1.63,
902
+ "learning_rate": 3.9632290676358616e-05,
903
+ "loss": 0.9644,
904
  "step": 740
905
  },
906
  {
907
  "epoch": 1.64,
908
+ "learning_rate": 3.749855514383298e-05,
909
+ "loss": 0.9931,
910
  "step": 745
911
  },
912
  {
913
+ "epoch": 1.65,
914
+ "learning_rate": 3.540204723273361e-05,
915
+ "loss": 0.846,
916
  "step": 750
917
  },
918
  {
919
+ "epoch": 1.66,
920
+ "learning_rate": 3.3345276408558416e-05,
921
+ "loss": 0.8853,
922
  "step": 755
923
  },
924
  {
925
+ "epoch": 1.67,
926
+ "learning_rate": 3.1330704572545604e-05,
927
+ "loss": 1.0022,
928
  "step": 760
929
  },
930
  {
931
  "epoch": 1.69,
932
+ "learning_rate": 2.9360743114840946e-05,
933
+ "loss": 0.9714,
934
  "step": 765
935
  },
936
  {
937
  "epoch": 1.7,
938
+ "learning_rate": 2.743775002812563e-05,
939
+ "loss": 0.9134,
940
  "step": 770
941
  },
942
  {
943
  "epoch": 1.71,
944
+ "learning_rate": 2.556402708516349e-05,
945
+ "loss": 1.0328,
946
  "step": 775
947
  },
948
  {
949
  "epoch": 1.72,
950
+ "learning_rate": 2.374181708363397e-05,
951
+ "loss": 0.9452,
952
  "step": 780
953
  },
954
  {
955
  "epoch": 1.73,
956
+ "learning_rate": 2.197330116156487e-05,
957
+ "loss": 0.9128,
958
  "step": 785
959
  },
960
  {
961
  "epoch": 1.74,
962
+ "learning_rate": 2.0260596186565206e-05,
963
+ "loss": 0.8532,
964
  "step": 790
965
  },
966
  {
967
  "epoch": 1.75,
968
+ "learning_rate": 1.8605752221991858e-05,
969
+ "loss": 1.0068,
970
  "step": 795
971
  },
972
  {
973
+ "epoch": 1.76,
974
+ "learning_rate": 1.7010750073074816e-05,
975
+ "loss": 0.954,
976
  "step": 800
977
  },
978
  {
979
+ "epoch": 1.77,
980
+ "learning_rate": 1.5477498915946135e-05,
981
+ "loss": 0.955,
982
  "step": 805
983
  },
984
  {
985
+ "epoch": 1.78,
986
+ "learning_rate": 1.40078340124038e-05,
987
+ "loss": 0.8937,
988
  "step": 810
989
  },
990
  {
991
  "epoch": 1.8,
992
+ "learning_rate": 1.2603514513153006e-05,
993
+ "loss": 0.8846,
994
  "step": 815
995
  },
996
  {
997
  "epoch": 1.81,
998
+ "learning_rate": 1.126622135214779e-05,
999
+ "loss": 0.9227,
1000
  "step": 820
1001
  },
1002
  {
1003
  "epoch": 1.82,
1004
+ "learning_rate": 9.997555234556263e-06,
1005
+ "loss": 0.9755,
1006
  "step": 825
1007
  },
1008
  {
1009
  "epoch": 1.83,
1010
+ "learning_rate": 8.799034720760158e-06,
1011
+ "loss": 1.0972,
1012
  "step": 830
1013
  },
1014
  {
1015
  "epoch": 1.84,
1016
+ "learning_rate": 7.672094408674397e-06,
1017
+ "loss": 1.0254,
1018
  "step": 835
1019
  },
1020
  {
1021
  "epoch": 1.85,
1022
+ "learning_rate": 6.618083216572745e-06,
1023
+ "loss": 0.9993,
1024
  "step": 840
1025
  },
1026
  {
1027
+ "epoch": 1.86,
1028
+ "learning_rate": 5.638262768466794e-06,
1029
+ "loss": 0.9833,
1030
  "step": 845
1031
  },
1032
  {
1033
+ "epoch": 1.87,
1034
+ "learning_rate": 4.733805883976356e-06,
1035
+ "loss": 0.9683,
1036
  "step": 850
1037
  },
1038
  {
1039
+ "epoch": 1.88,
1040
+ "learning_rate": 3.905795174492459e-06,
1041
+ "loss": 1.0143,
1042
  "step": 855
1043
  },
1044
  {
1045
+ "epoch": 1.89,
1046
+ "learning_rate": 3.155221747321248e-06,
1047
+ "loss": 1.0293,
1048
  "step": 860
1049
  },
1050
  {
1051
  "epoch": 1.91,
1052
+ "learning_rate": 2.482984019353904e-06,
1053
+ "loss": 0.9645,
1054
  "step": 865
1055
  },
1056
  {
1057
  "epoch": 1.92,
1058
+ "learning_rate": 1.8898866416844749e-06,
1059
+ "loss": 0.9395,
1060
  "step": 870
1061
  },
1062
  {
1063
  "epoch": 1.93,
1064
+ "learning_rate": 1.37663953646536e-06,
1065
+ "loss": 1.0553,
1066
  "step": 875
1067
  },
1068
  {
1069
  "epoch": 1.94,
1070
+ "learning_rate": 9.438570471458076e-07,
1071
+ "loss": 1.0763,
1072
  "step": 880
1073
  },
1074
  {
1075
  "epoch": 1.95,
1076
+ "learning_rate": 5.920572031214673e-07,
1077
+ "loss": 0.987,
1078
  "step": 885
1079
  },
1080
  {
1081
  "epoch": 1.96,
1082
+ "learning_rate": 3.216610996650991e-07,
1083
+ "loss": 1.0021,
1084
  "step": 890
1085
  },
1086
  {
1087
+ "epoch": 1.97,
1088
+ "learning_rate": 1.3299239388720947e-07,
1089
+ "loss": 0.8033,
1090
  "step": 895
1091
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1092
  {
1093
  "epoch": 1.98,
1094
+ "learning_rate": 2.6276917326665837e-08,
1095
+ "loss": 1.0829,
1096
+ "step": 900
1097
  },
1098
  {
1099
  "epoch": 1.99,
1100
+ "learning_rate": 1.6424056362936757e-09,
1101
+ "loss": 0.9621,
1102
+ "step": 905
 
 
 
 
 
 
1103
  },
1104
  {
1105
  "epoch": 2.0,
1106
+ "eval_loss": 0.8025060296058655,
1107
+ "eval_runtime": 8.3951,
1108
+ "eval_samples_per_second": 76.95,
1109
+ "eval_steps_per_second": 9.649,
1110
+ "step": 908
1111
  }
1112
  ],
1113
+ "max_steps": 1362,
1114
  "num_train_epochs": 3,
1115
+ "total_flos": 948751368192000.0,
1116
  "trial_name": null,
1117
  "trial_params": null
1118
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:af133d06cf9107eff32426f3a7587a6180a61f69752eaf4fc23152d22d231624
3
  size 2863
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af3d16db63fcd2a5987c663780ce8fd33dc7a85804a193836b0800c52e0bcc3a
3
  size 2863