AlekseyKorshuk commited on
Commit
aa8012e
·
1 Parent(s): d9d24b5

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/14l0irt0/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/1buv5m9i) for full transparency and reproducibility.
55
 
56
- At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/1buv5m9i/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/wqeebjm6/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/1r91z1u3) for full transparency and reproducibility.
55
 
56
+ At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/1r91z1u3/artifacts) is logged and versioned.
57
 
58
  ## How to use
59
 
evaluation.txt CHANGED
@@ -1 +1 @@
1
- {"eval_loss": 0.37163448333740234, "eval_runtime": 28.704, "eval_samples_per_second": 20.45, "eval_steps_per_second": 2.578, "epoch": 3.0}
 
1
+ {"eval_loss": 0.41102197766304016, "eval_runtime": 14.7285, "eval_samples_per_second": 43.182, "eval_steps_per_second": 5.432, "epoch": 3.0}
flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bb44bd892620724348b1f8bd98aff6e03e38fe1376bcd754f4b89769c6ed210a
3
  size 497764120
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:179861d06f26a39a625bade68ca595e81d6de04012686551d418e512f108a547
3
  size 497764120
optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bfa216e29dab868efb9674298799ceecd462e1b341fde7e584fcff00171c7b33
3
  size 995604017
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5af45af4262862dc6b6a14037e759fcea974defe54215dbe6268cb309d499e5b
3
  size 995604017
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce13b2714bbf717a6d07d2e7d112009b4ee85fc468a7056e1df4e22a76c4b967
3
  size 510403817
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcdd242b4ca2943da34998f3437f3d233bc20424c948fe5cd014c826a37894d1
3
  size 510403817
rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7753baebd09fb9ef19a9cbd379c34b2acd8cbc7bd213ace8e1d0bf591bb2bcfc
3
  size 14503
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dae1d462e506ffceccce9f9c31d892f0fe4bd0d84bb7d72de19955d2ea5bfc2
3
  size 14503
scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:63fe74d287e7e5851bce8bf478b5c5444779821627837777ac4e6b2d39e75580
3
  size 623
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed0916d61de33d62f521fe37251dde481dd7b032866021fa20b2813604e4e361
3
  size 623
trainer_state.json CHANGED
@@ -1,1152 +1,570 @@
1
  {
2
- "best_metric": 0.37163448333740234,
3
- "best_model_checkpoint": "output/eminem/checkpoint-924",
4
- "epoch": 2.0,
5
- "global_step": 924,
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.0001144556130966329,
13
- "loss": 0.7526,
14
  "step": 5
15
  },
16
  {
17
  "epoch": 0.02,
18
- "learning_rate": 0.00011618189922507932,
19
- "loss": 0.7708,
20
  "step": 10
21
  },
22
  {
23
  "epoch": 0.03,
24
- "learning_rate": 0.00011785197628750708,
25
- "loss": 0.869,
26
  "step": 15
27
  },
28
  {
29
  "epoch": 0.04,
30
- "learning_rate": 0.00011946387140192834,
31
- "loss": 0.8428,
32
  "step": 20
33
  },
34
  {
35
  "epoch": 0.05,
36
- "learning_rate": 0.00012101568041738691,
37
- "loss": 0.7731,
38
  "step": 25
39
  },
40
  {
41
  "epoch": 0.07,
42
- "learning_rate": 0.00012250557016335326,
43
- "loss": 0.8297,
44
  "step": 30
45
  },
46
  {
47
  "epoch": 0.08,
48
- "learning_rate": 0.00012393178061527061,
49
- "loss": 0.8621,
50
  "step": 35
51
  },
52
  {
53
  "epoch": 0.09,
54
- "learning_rate": 0.00012529262697369198,
55
- "loss": 0.7284,
56
  "step": 40
57
  },
58
  {
59
  "epoch": 0.1,
60
- "learning_rate": 0.00012658650165455924,
61
- "loss": 0.7879,
62
  "step": 45
63
  },
64
  {
65
  "epoch": 0.11,
66
- "learning_rate": 0.00012781187618824956,
67
- "loss": 0.9203,
68
  "step": 50
69
  },
70
  {
71
  "epoch": 0.12,
72
- "learning_rate": 0.00012896730302519073,
73
- "loss": 0.8996,
74
  "step": 55
75
  },
76
  {
77
  "epoch": 0.13,
78
- "learning_rate": 0.0001300514172458524,
79
- "loss": 0.949,
80
  "step": 60
81
  },
82
  {
83
  "epoch": 0.14,
84
- "learning_rate": 0.0001310629381731565,
85
- "loss": 0.8166,
86
  "step": 65
87
  },
88
  {
89
  "epoch": 0.15,
90
- "learning_rate": 0.0001320006708853459,
91
- "loss": 0.8125,
92
  "step": 70
93
  },
94
  {
95
  "epoch": 0.16,
96
- "learning_rate": 0.00013286350762756457,
97
- "loss": 0.8287,
98
  "step": 75
99
  },
100
  {
101
  "epoch": 0.18,
102
- "learning_rate": 0.0001336504291204605,
103
- "loss": 0.8779,
104
  "step": 80
105
  },
106
  {
107
  "epoch": 0.19,
108
- "learning_rate": 0.00013436050576427267,
109
- "loss": 0.8684,
110
  "step": 85
111
  },
112
  {
113
  "epoch": 0.2,
114
- "learning_rate": 0.00013499289873697794,
115
- "loss": 0.8235,
116
  "step": 90
117
  },
118
  {
119
  "epoch": 0.21,
120
- "learning_rate": 0.00013554686098520387,
121
- "loss": 0.9249,
122
  "step": 95
123
  },
124
  {
125
  "epoch": 0.22,
126
- "learning_rate": 0.00013602173810672657,
127
- "loss": 0.8379,
128
  "step": 100
129
  },
130
  {
131
  "epoch": 0.23,
132
- "learning_rate": 0.0001364169691235302,
133
- "loss": 0.9086,
134
  "step": 105
135
  },
136
  {
137
  "epoch": 0.24,
138
- "learning_rate": 0.0001367320871444924,
139
- "loss": 0.8637,
140
  "step": 110
141
  },
142
  {
143
  "epoch": 0.25,
144
- "learning_rate": 0.0001369667199169302,
145
- "loss": 0.8941,
146
  "step": 115
147
  },
148
  {
149
  "epoch": 0.26,
150
- "learning_rate": 0.00013712059026634634,
151
- "loss": 0.8238,
152
  "step": 120
153
  },
154
  {
155
  "epoch": 0.27,
156
- "learning_rate": 0.00013719351642385878,
157
- "loss": 0.8578,
158
  "step": 125
159
  },
160
  {
161
- "epoch": 0.28,
162
- "learning_rate": 0.0001371854122409264,
163
- "loss": 0.8231,
164
  "step": 130
165
  },
166
  {
167
  "epoch": 0.3,
168
- "learning_rate": 0.0001370962872911174,
169
- "loss": 0.815,
170
  "step": 135
171
  },
172
  {
173
  "epoch": 0.31,
174
- "learning_rate": 0.00013692624685879965,
175
- "loss": 0.726,
176
  "step": 140
177
  },
178
  {
179
  "epoch": 0.32,
180
- "learning_rate": 0.00013667549181476735,
181
- "loss": 0.8906,
182
  "step": 145
183
  },
184
  {
185
  "epoch": 0.33,
186
- "learning_rate": 0.00013634431837894983,
187
- "loss": 0.7687,
188
  "step": 150
189
  },
190
  {
191
  "epoch": 0.34,
192
- "learning_rate": 0.0001359331177704834,
193
- "loss": 0.8484,
194
  "step": 155
195
  },
196
  {
197
  "epoch": 0.35,
198
- "learning_rate": 0.00013544237574555944,
199
- "loss": 0.9342,
200
  "step": 160
201
  },
202
  {
203
  "epoch": 0.36,
204
- "learning_rate": 0.00013487267202359548,
205
- "loss": 0.887,
206
  "step": 165
207
  },
208
  {
209
  "epoch": 0.37,
210
- "learning_rate": 0.00013422467960240306,
211
- "loss": 0.9549,
212
  "step": 170
213
  },
214
  {
215
  "epoch": 0.38,
216
- "learning_rate": 0.00013349916396317312,
217
- "loss": 1.0232,
218
  "step": 175
219
  },
220
  {
221
  "epoch": 0.39,
222
- "learning_rate": 0.00013269698216619567,
223
- "loss": 0.856,
224
  "step": 180
225
  },
226
  {
227
- "epoch": 0.4,
228
- "learning_rate": 0.00013181908183841387,
229
- "loss": 0.8467,
230
  "step": 185
231
  },
232
  {
233
  "epoch": 0.42,
234
- "learning_rate": 0.0001308665000539712,
235
- "loss": 0.8928,
236
  "step": 190
237
  },
238
  {
239
  "epoch": 0.43,
240
- "learning_rate": 0.00012984036210911092,
241
- "loss": 0.9901,
242
  "step": 195
243
  },
244
  {
245
  "epoch": 0.44,
246
- "learning_rate": 0.00012874188019284667,
247
- "loss": 0.79,
248
  "step": 200
249
  },
250
  {
251
  "epoch": 0.45,
252
- "learning_rate": 0.0001275723519549898,
253
- "loss": 0.8893,
254
  "step": 205
255
  },
256
  {
257
  "epoch": 0.46,
258
- "learning_rate": 0.00012633315897322012,
259
- "loss": 0.9893,
260
  "step": 210
261
  },
262
  {
263
  "epoch": 0.47,
264
- "learning_rate": 0.0001250257651210129,
265
- "loss": 0.8199,
266
  "step": 215
267
  },
268
  {
269
  "epoch": 0.48,
270
- "learning_rate": 0.00012365171483834196,
271
- "loss": 0.8304,
272
  "step": 220
273
  },
274
  {
275
  "epoch": 0.49,
276
- "learning_rate": 0.00012221263130722672,
277
- "loss": 0.924,
278
  "step": 225
279
  },
280
  {
281
  "epoch": 0.5,
282
- "learning_rate": 0.00012071021453423055,
283
- "loss": 0.8617,
284
  "step": 230
285
  },
286
  {
287
- "epoch": 0.51,
288
- "learning_rate": 0.00011914623934224188,
289
- "loss": 0.8712,
290
  "step": 235
291
  },
292
  {
293
  "epoch": 0.53,
294
- "learning_rate": 0.00011752255327383619,
295
- "loss": 1.0037,
296
  "step": 240
297
  },
298
  {
299
  "epoch": 0.54,
300
- "learning_rate": 0.00011584107440876567,
301
- "loss": 0.8259,
302
  "step": 245
303
  },
304
  {
305
  "epoch": 0.55,
306
- "learning_rate": 0.0001141037890981024,
307
- "loss": 0.8382,
308
  "step": 250
309
  },
310
  {
311
  "epoch": 0.56,
312
- "learning_rate": 0.00011231274961773941,
313
- "loss": 1.0107,
314
  "step": 255
315
  },
316
  {
317
  "epoch": 0.57,
318
- "learning_rate": 0.00011047007174401259,
319
- "loss": 0.9122,
320
  "step": 260
321
  },
322
  {
323
  "epoch": 0.58,
324
- "learning_rate": 0.00010857793225431036,
325
- "loss": 0.9962,
326
  "step": 265
327
  },
328
  {
329
  "epoch": 0.59,
330
- "learning_rate": 0.00010663856635561276,
331
- "loss": 0.7623,
332
  "step": 270
333
  },
334
  {
335
  "epoch": 0.6,
336
- "learning_rate": 0.0001046542650440321,
337
- "loss": 0.9404,
338
  "step": 275
339
  },
340
  {
341
  "epoch": 0.61,
342
- "learning_rate": 0.00010262737239840785,
343
- "loss": 0.9339,
344
  "step": 280
345
  },
346
  {
347
  "epoch": 0.62,
348
- "learning_rate": 0.00010056028281124445,
349
- "loss": 0.9049,
350
  "step": 285
351
  },
352
  {
353
- "epoch": 0.63,
354
- "learning_rate": 9.845543816016187e-05,
355
- "loss": 1.0607,
356
  "step": 290
357
  },
358
  {
359
  "epoch": 0.65,
360
- "learning_rate": 9.63153249232949e-05,
361
- "loss": 0.8714,
362
  "step": 295
363
  },
364
  {
365
  "epoch": 0.66,
366
- "learning_rate": 9.414247124197839e-05,
367
- "loss": 0.9058,
368
  "step": 300
369
  },
370
  {
371
  "epoch": 0.67,
372
- "learning_rate": 9.19394439342254e-05,
373
- "loss": 0.9844,
374
  "step": 305
375
  },
376
  {
377
  "epoch": 0.68,
378
- "learning_rate": 8.970884546251429e-05,
379
- "loss": 0.8605,
380
  "step": 310
381
  },
382
  {
383
  "epoch": 0.69,
384
- "learning_rate": 8.74533108594708e-05,
385
- "loss": 1.0247,
386
  "step": 315
387
  },
388
  {
389
  "epoch": 0.7,
390
- "learning_rate": 8.517550461506486e-05,
391
- "loss": 0.8763,
392
  "step": 320
393
  },
394
  {
395
  "epoch": 0.71,
396
- "learning_rate": 8.287811752902797e-05,
397
- "loss": 0.9301,
398
  "step": 325
399
  },
400
  {
401
  "epoch": 0.72,
402
- "learning_rate": 8.056386353217202e-05,
403
- "loss": 0.8877,
404
  "step": 330
405
  },
406
  {
407
  "epoch": 0.73,
408
- "learning_rate": 7.823547648039207e-05,
409
- "loss": 0.7894,
410
  "step": 335
411
  },
412
  {
413
- "epoch": 0.74,
414
- "learning_rate": 7.589570692512989e-05,
415
- "loss": 1.0017,
416
  "step": 340
417
  },
418
  {
419
- "epoch": 0.75,
420
- "learning_rate": 7.354731886411453e-05,
421
- "loss": 0.9258,
422
  "step": 345
423
  },
424
  {
425
  "epoch": 0.77,
426
- "learning_rate": 7.119308647622086e-05,
427
- "loss": 0.8299,
428
  "step": 350
429
  },
430
  {
431
  "epoch": 0.78,
432
- "learning_rate": 6.883579084428906e-05,
433
- "loss": 0.936,
434
  "step": 355
435
  },
436
  {
437
  "epoch": 0.79,
438
- "learning_rate": 6.64782166698198e-05,
439
- "loss": 0.8412,
440
  "step": 360
441
  },
442
  {
443
  "epoch": 0.8,
444
- "learning_rate": 6.412314898334569e-05,
445
- "loss": 0.9055,
446
  "step": 365
447
  },
448
  {
449
  "epoch": 0.81,
450
- "learning_rate": 6.17733698544629e-05,
451
- "loss": 0.8395,
452
  "step": 370
453
  },
454
  {
455
  "epoch": 0.82,
456
- "learning_rate": 5.943165510532945e-05,
457
- "loss": 0.8795,
458
  "step": 375
459
  },
460
  {
461
  "epoch": 0.83,
462
- "learning_rate": 5.710077103155516e-05,
463
- "loss": 0.9003,
464
  "step": 380
465
  },
466
  {
467
  "epoch": 0.84,
468
- "learning_rate": 5.4783471134343704e-05,
469
- "loss": 0.7688,
470
  "step": 385
471
  },
472
  {
473
- "epoch": 0.85,
474
- "learning_rate": 5.248249286775053e-05,
475
- "loss": 0.8768,
476
  "step": 390
477
  },
478
  {
479
- "epoch": 0.86,
480
- "learning_rate": 5.02005544048863e-05,
481
- "loss": 0.8473,
482
  "step": 395
483
  },
484
  {
485
  "epoch": 0.88,
486
- "learning_rate": 4.7940351426927284e-05,
487
- "loss": 0.9321,
488
  "step": 400
489
  },
490
  {
491
  "epoch": 0.89,
492
- "learning_rate": 4.5704553938646724e-05,
493
- "loss": 0.7829,
494
  "step": 405
495
  },
496
  {
497
  "epoch": 0.9,
498
- "learning_rate": 4.349580311433621e-05,
499
- "loss": 0.9309,
500
  "step": 410
501
  },
502
  {
503
  "epoch": 0.91,
504
- "learning_rate": 4.1316708177727135e-05,
505
- "loss": 0.8959,
506
  "step": 415
507
  },
508
  {
509
  "epoch": 0.92,
510
- "learning_rate": 3.9169843319703834e-05,
511
- "loss": 1.0492,
512
  "step": 420
513
  },
514
  {
515
  "epoch": 0.93,
516
- "learning_rate": 3.7057744657371865e-05,
517
- "loss": 0.8489,
518
  "step": 425
519
  },
520
  {
521
  "epoch": 0.94,
522
- "learning_rate": 3.4982907238114506e-05,
523
- "loss": 0.8533,
524
  "step": 430
525
  },
526
  {
527
  "epoch": 0.95,
528
- "learning_rate": 3.294778209216339e-05,
529
- "loss": 0.8134,
530
  "step": 435
531
  },
532
  {
533
  "epoch": 0.96,
534
- "learning_rate": 3.0954773337168915e-05,
535
- "loss": 0.715,
536
  "step": 440
537
  },
538
  {
539
- "epoch": 0.97,
540
- "learning_rate": 2.9006235338179113e-05,
541
- "loss": 0.8985,
542
  "step": 445
543
  },
544
  {
545
- "epoch": 0.98,
546
- "learning_rate": 2.7104469926418013e-05,
547
- "loss": 0.7858,
548
  "step": 450
549
  },
550
  {
551
  "epoch": 1.0,
552
- "learning_rate": 2.5251723680080823e-05,
553
- "loss": 0.8067,
554
  "step": 455
555
  },
556
  {
557
  "epoch": 1.0,
558
- "eval_loss": 0.5359192490577698,
559
- "eval_runtime": 35.4879,
560
- "eval_samples_per_second": 17.527,
561
- "eval_steps_per_second": 2.198,
562
- "step": 457
563
- },
564
- {
565
- "epoch": 1.02,
566
- "learning_rate": 7.411387289257788e-05,
567
- "loss": 0.7428,
568
- "step": 460
569
- },
570
- {
571
- "epoch": 1.04,
572
- "learning_rate": 7.650217107773121e-05,
573
- "loss": 0.7294,
574
- "step": 465
575
- },
576
- {
577
- "epoch": 1.05,
578
- "learning_rate": 7.888079875674826e-05,
579
- "loss": 0.8298,
580
- "step": 470
581
- },
582
- {
583
- "epoch": 1.06,
584
- "learning_rate": 8.1246845016499e-05,
585
- "loss": 0.7537,
586
- "step": 475
587
- },
588
- {
589
- "epoch": 1.07,
590
- "learning_rate": 8.359741434072031e-05,
591
- "loss": 0.6783,
592
- "step": 480
593
- },
594
- {
595
- "epoch": 1.08,
596
- "learning_rate": 8.59296301534902e-05,
597
- "loss": 0.7367,
598
- "step": 485
599
- },
600
- {
601
- "epoch": 1.09,
602
- "learning_rate": 8.824063833951144e-05,
603
- "loss": 0.8382,
604
- "step": 490
605
- },
606
- {
607
- "epoch": 1.1,
608
- "learning_rate": 9.052761073692769e-05,
609
- "loss": 0.6501,
610
- "step": 495
611
- },
612
- {
613
- "epoch": 1.11,
614
- "learning_rate": 9.2787748598361e-05,
615
- "loss": 0.7697,
616
- "step": 500
617
- },
618
- {
619
- "epoch": 1.12,
620
- "learning_rate": 9.501828601595296e-05,
621
- "loss": 0.7842,
622
- "step": 505
623
- },
624
- {
625
- "epoch": 1.14,
626
- "learning_rate": 9.72164933062357e-05,
627
- "loss": 0.8064,
628
- "step": 510
629
- },
630
- {
631
- "epoch": 1.15,
632
- "learning_rate": 9.937968035064501e-05,
633
- "loss": 0.8361,
634
- "step": 515
635
- },
636
- {
637
- "epoch": 1.16,
638
- "learning_rate": 0.00010150519988764197,
639
- "loss": 0.7419,
640
- "step": 520
641
- },
642
- {
643
- "epoch": 1.17,
644
- "learning_rate": 0.00010359045075236705,
645
- "loss": 0.8181,
646
- "step": 525
647
- },
648
- {
649
- "epoch": 1.18,
650
- "learning_rate": 0.00010563288105989297,
651
- "loss": 0.7946,
652
- "step": 530
653
- },
654
- {
655
- "epoch": 1.19,
656
- "learning_rate": 0.00010762999132815968,
657
- "loss": 0.7522,
658
- "step": 535
659
- },
660
- {
661
- "epoch": 1.2,
662
- "learning_rate": 0.00010957933753679646,
663
- "loss": 0.721,
664
- "step": 540
665
- },
666
- {
667
- "epoch": 1.21,
668
- "learning_rate": 0.00011147853411805452,
669
- "loss": 0.7965,
670
- "step": 545
671
- },
672
- {
673
- "epoch": 1.22,
674
- "learning_rate": 0.00011332525687621633,
675
- "loss": 0.8731,
676
- "step": 550
677
- },
678
- {
679
- "epoch": 1.24,
680
- "learning_rate": 0.00011511724583189029,
681
- "loss": 0.7461,
682
- "step": 555
683
- },
684
- {
685
- "epoch": 1.25,
686
- "learning_rate": 0.00011685230798773347,
687
- "loss": 0.8536,
688
- "step": 560
689
- },
690
- {
691
- "epoch": 1.26,
692
- "learning_rate": 0.00011852832001218846,
693
- "loss": 0.8069,
694
- "step": 565
695
- },
696
- {
697
- "epoch": 1.27,
698
- "learning_rate": 0.00012014323083797337,
699
- "loss": 0.7363,
700
- "step": 570
701
- },
702
- {
703
- "epoch": 1.28,
704
- "learning_rate": 0.00012169506417212839,
705
- "loss": 0.8555,
706
- "step": 575
707
- },
708
- {
709
- "epoch": 1.29,
710
- "learning_rate": 0.0001231819209145695,
711
- "loss": 0.8844,
712
- "step": 580
713
- },
714
- {
715
- "epoch": 1.3,
716
- "learning_rate": 0.00012460198148215605,
717
- "loss": 0.8991,
718
- "step": 585
719
- },
720
- {
721
- "epoch": 1.31,
722
- "learning_rate": 0.00012595350803546582,
723
- "loss": 0.7508,
724
- "step": 590
725
- },
726
- {
727
- "epoch": 1.33,
728
- "learning_rate": 0.0001272348466055214,
729
- "loss": 0.8352,
730
- "step": 595
731
- },
732
- {
733
- "epoch": 1.34,
734
- "learning_rate": 0.00012844442911787884,
735
- "loss": 0.8305,
736
- "step": 600
737
- },
738
- {
739
- "epoch": 1.35,
740
- "learning_rate": 0.00012958077531160963,
741
- "loss": 0.8743,
742
- "step": 605
743
- },
744
- {
745
- "epoch": 1.36,
746
- "learning_rate": 0.0001306424945508076,
747
- "loss": 0.9164,
748
- "step": 610
749
- },
750
- {
751
- "epoch": 1.37,
752
- "learning_rate": 0.00013162828752642103,
753
- "loss": 0.8642,
754
- "step": 615
755
- },
756
- {
757
- "epoch": 1.38,
758
- "learning_rate": 0.0001325369478463158,
759
- "loss": 0.845,
760
- "step": 620
761
- },
762
- {
763
- "epoch": 1.39,
764
- "learning_rate": 0.00013336736351163685,
765
- "loss": 0.8869,
766
- "step": 625
767
- },
768
- {
769
- "epoch": 1.4,
770
- "learning_rate": 0.00013411851827764502,
771
- "loss": 0.8367,
772
- "step": 630
773
- },
774
- {
775
- "epoch": 1.41,
776
- "learning_rate": 0.00013478949289737682,
777
- "loss": 0.8495,
778
- "step": 635
779
- },
780
- {
781
- "epoch": 1.43,
782
- "learning_rate": 0.00013537946624659676,
783
- "loss": 0.9375,
784
- "step": 640
785
- },
786
- {
787
- "epoch": 1.44,
788
- "learning_rate": 0.00013588771632867518,
789
- "loss": 0.9639,
790
- "step": 645
791
- },
792
- {
793
- "epoch": 1.45,
794
- "learning_rate": 0.00013631362115814792,
795
- "loss": 0.8622,
796
- "step": 650
797
- },
798
- {
799
- "epoch": 1.46,
800
- "learning_rate": 0.00013665665952189337,
801
- "loss": 0.7462,
802
- "step": 655
803
- },
804
- {
805
- "epoch": 1.47,
806
- "learning_rate": 0.0001369164116169769,
807
- "loss": 0.8065,
808
- "step": 660
809
- },
810
- {
811
- "epoch": 1.48,
812
- "learning_rate": 0.0001370925595644003,
813
- "loss": 0.7944,
814
- "step": 665
815
- },
816
- {
817
- "epoch": 1.49,
818
- "learning_rate": 0.00013718488779811326,
819
- "loss": 0.9673,
820
- "step": 670
821
- },
822
- {
823
- "epoch": 1.5,
824
- "learning_rate": 0.00013719328332881955,
825
- "loss": 0.8319,
826
- "step": 675
827
- },
828
- {
829
- "epoch": 1.51,
830
- "learning_rate": 0.00013711773588225001,
831
- "loss": 0.7709,
832
- "step": 680
833
- },
834
- {
835
- "epoch": 1.53,
836
- "learning_rate": 0.00013695833791173674,
837
- "loss": 0.8833,
838
- "step": 685
839
- },
840
- {
841
- "epoch": 1.54,
842
- "learning_rate": 0.00013671528448507009,
843
- "loss": 0.8851,
844
- "step": 690
845
- },
846
- {
847
- "epoch": 1.55,
848
- "learning_rate": 0.00013638887304577923,
849
- "loss": 0.8425,
850
- "step": 695
851
- },
852
- {
853
- "epoch": 1.56,
854
- "learning_rate": 0.00013597950304912684,
855
- "loss": 0.9205,
856
- "step": 700
857
- },
858
- {
859
- "epoch": 1.57,
860
- "learning_rate": 0.0001354876754732656,
861
- "loss": 0.9954,
862
- "step": 705
863
- },
864
- {
865
- "epoch": 1.58,
866
- "learning_rate": 0.00013491399220615015,
867
- "loss": 0.8622,
868
- "step": 710
869
- },
870
- {
871
- "epoch": 1.59,
872
- "learning_rate": 0.00013425915530896078,
873
- "loss": 0.8284,
874
- "step": 715
875
- },
876
- {
877
- "epoch": 1.6,
878
- "learning_rate": 0.00013352396615693527,
879
- "loss": 0.8501,
880
- "step": 720
881
- },
882
- {
883
- "epoch": 1.61,
884
- "learning_rate": 0.00013270932445866562,
885
- "loss": 0.7884,
886
- "step": 725
887
- },
888
- {
889
- "epoch": 1.63,
890
- "learning_rate": 0.00013181622715504977,
891
- "loss": 0.9186,
892
- "step": 730
893
- },
894
- {
895
- "epoch": 1.64,
896
- "learning_rate": 0.000130845767199263,
897
- "loss": 0.8647,
898
- "step": 735
899
- },
900
- {
901
- "epoch": 1.65,
902
- "learning_rate": 0.0001297991322192165,
903
- "loss": 0.8587,
904
- "step": 740
905
- },
906
- {
907
- "epoch": 1.66,
908
- "learning_rate": 0.00012867760306417496,
909
- "loss": 0.8789,
910
- "step": 745
911
- },
912
- {
913
- "epoch": 1.67,
914
- "learning_rate": 0.00012748255223727105,
915
- "loss": 0.7334,
916
- "step": 750
917
- },
918
- {
919
- "epoch": 1.68,
920
- "learning_rate": 0.00012621544221587383,
921
- "loss": 0.854,
922
- "step": 755
923
- },
924
- {
925
- "epoch": 1.69,
926
- "learning_rate": 0.000124877823661834,
927
- "loss": 0.8584,
928
- "step": 760
929
- },
930
- {
931
- "epoch": 1.7,
932
- "learning_rate": 0.0001234713335238199,
933
- "loss": 0.8898,
934
- "step": 765
935
- },
936
- {
937
- "epoch": 1.71,
938
- "learning_rate": 0.00012199769303405894,
939
- "loss": 0.842,
940
- "step": 770
941
- },
942
- {
943
- "epoch": 1.73,
944
- "learning_rate": 0.00012045870560192401,
945
- "loss": 0.8657,
946
- "step": 775
947
- },
948
- {
949
- "epoch": 1.74,
950
- "learning_rate": 0.00011885625460696435,
951
- "loss": 0.8646,
952
- "step": 780
953
- },
954
- {
955
- "epoch": 1.75,
956
- "learning_rate": 0.00011719230109406423,
957
- "loss": 0.974,
958
- "step": 785
959
- },
960
- {
961
- "epoch": 1.76,
962
- "learning_rate": 0.00011546888137356373,
963
- "loss": 0.892,
964
- "step": 790
965
- },
966
- {
967
- "epoch": 1.77,
968
- "learning_rate": 0.00011368810452925682,
969
- "loss": 0.8279,
970
- "step": 795
971
- },
972
- {
973
- "epoch": 1.78,
974
- "learning_rate": 0.00011185214983735281,
975
- "loss": 0.9267,
976
- "step": 800
977
- },
978
- {
979
- "epoch": 1.79,
980
- "learning_rate": 0.0001099632640995077,
981
- "loss": 0.8753,
982
- "step": 805
983
- },
984
- {
985
- "epoch": 1.8,
986
- "learning_rate": 0.00010802375889325772,
987
- "loss": 0.8341,
988
- "step": 810
989
- },
990
- {
991
- "epoch": 1.82,
992
- "learning_rate": 0.00010603600774314254,
993
- "loss": 0.9575,
994
- "step": 815
995
- },
996
- {
997
- "epoch": 1.83,
998
- "learning_rate": 0.00010400244321605133,
999
- "loss": 0.851,
1000
- "step": 820
1001
- },
1002
- {
1003
- "epoch": 1.84,
1004
- "learning_rate": 0.0001019255539442888,
1005
- "loss": 0.7728,
1006
- "step": 825
1007
- },
1008
- {
1009
- "epoch": 1.85,
1010
- "learning_rate": 9.980788158004715e-05,
1011
- "loss": 0.8267,
1012
- "step": 830
1013
- },
1014
- {
1015
- "epoch": 1.86,
1016
- "learning_rate": 9.765201768498375e-05,
1017
- "loss": 0.8464,
1018
- "step": 835
1019
- },
1020
- {
1021
- "epoch": 1.87,
1022
- "learning_rate": 9.546060055873103e-05,
1023
- "loss": 0.8036,
1024
- "step": 840
1025
- },
1026
- {
1027
- "epoch": 1.88,
1028
- "learning_rate": 9.32363120101919e-05,
1029
- "loss": 0.9536,
1030
- "step": 845
1031
- },
1032
- {
1033
- "epoch": 1.89,
1034
- "learning_rate": 9.098187407560697e-05,
1035
- "loss": 0.8601,
1036
- "step": 850
1037
- },
1038
- {
1039
- "epoch": 1.9,
1040
- "learning_rate": 8.870004568738188e-05,
1041
- "loss": 0.78,
1042
- "step": 855
1043
- },
1044
- {
1045
- "epoch": 1.92,
1046
- "learning_rate": 8.639361929777101e-05,
1047
- "loss": 0.8761,
1048
- "step": 860
1049
- },
1050
- {
1051
- "epoch": 1.93,
1052
- "learning_rate": 8.406541746153297e-05,
1053
- "loss": 0.8593,
1054
- "step": 865
1055
- },
1056
- {
1057
- "epoch": 1.94,
1058
- "learning_rate": 8.171828938173595e-05,
1059
- "loss": 0.7247,
1060
- "step": 870
1061
- },
1062
- {
1063
- "epoch": 1.95,
1064
- "learning_rate": 7.935510742298902e-05,
1065
- "loss": 0.8669,
1066
- "step": 875
1067
- },
1068
- {
1069
- "epoch": 1.96,
1070
- "learning_rate": 7.697876359628178e-05,
1071
- "loss": 0.9187,
1072
- "step": 880
1073
- },
1074
- {
1075
- "epoch": 1.97,
1076
- "learning_rate": 7.459216601982097e-05,
1077
- "loss": 0.8965,
1078
- "step": 885
1079
- },
1080
- {
1081
- "epoch": 1.98,
1082
- "learning_rate": 7.219823536012484e-05,
1083
- "loss": 0.9159,
1084
- "step": 890
1085
- },
1086
- {
1087
- "epoch": 1.99,
1088
- "learning_rate": 6.97999012577823e-05,
1089
- "loss": 1.0147,
1090
- "step": 895
1091
- },
1092
- {
1093
- "epoch": 2.0,
1094
- "eval_loss": 0.5308113694190979,
1095
- "eval_runtime": 33.6128,
1096
- "eval_samples_per_second": 20.587,
1097
- "eval_steps_per_second": 2.588,
1098
- "step": 898
1099
- },
1100
- {
1101
- "epoch": 1.98,
1102
- "learning_rate": 0.00013331119038948074,
1103
- "loss": 0.7724,
1104
- "step": 900
1105
- },
1106
- {
1107
- "epoch": 1.99,
1108
- "learning_rate": 0.0001340585253974795,
1109
- "loss": 0.7496,
1110
- "step": 905
1111
- },
1112
- {
1113
- "epoch": 2.0,
1114
- "learning_rate": 0.0001347278522437355,
1115
- "loss": 0.8119,
1116
- "step": 910
1117
- },
1118
- {
1119
- "epoch": 2.0,
1120
- "eval_loss": 0.39583703875541687,
1121
- "eval_runtime": 13.6672,
1122
- "eval_samples_per_second": 47.047,
1123
- "eval_steps_per_second": 5.927,
1124
- "step": 910
1125
- },
1126
- {
1127
- "epoch": 1.98,
1128
- "learning_rate": 1.7062254198703994e-05,
1129
- "loss": 0.8052,
1130
- "step": 915
1131
- },
1132
- {
1133
- "epoch": 1.99,
1134
- "learning_rate": 1.8631088017841603e-05,
1135
- "loss": 0.7911,
1136
- "step": 920
1137
- },
1138
- {
1139
- "epoch": 2.0,
1140
- "eval_loss": 0.37163448333740234,
1141
- "eval_runtime": 28.6365,
1142
- "eval_samples_per_second": 20.498,
1143
- "eval_steps_per_second": 2.584,
1144
- "step": 924
1145
  }
1146
  ],
1147
- "max_steps": 1386,
1148
  "num_train_epochs": 3,
1149
- "total_flos": 963383721984000.0,
1150
  "trial_name": null,
1151
  "trial_params": null
1152
  }
 
1
  {
2
+ "best_metric": 0.41102197766304016,
3
+ "best_model_checkpoint": "output/eminem/checkpoint-456",
4
+ "epoch": 1.0,
5
+ "global_step": 456,
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": 9.668898730225426e-06,
13
+ "loss": 0.6451,
14
  "step": 5
15
  },
16
  {
17
  "epoch": 0.02,
18
+ "learning_rate": 8.494471426523322e-06,
19
+ "loss": 0.6405,
20
  "step": 10
21
  },
22
  {
23
  "epoch": 0.03,
24
+ "learning_rate": 7.391359230778123e-06,
25
+ "loss": 0.6683,
26
  "step": 15
27
  },
28
  {
29
  "epoch": 0.04,
30
+ "learning_rate": 6.360870983745512e-06,
31
+ "loss": 0.6156,
32
  "step": 20
33
  },
34
  {
35
  "epoch": 0.05,
36
+ "learning_rate": 5.404229357984623e-06,
37
+ "loss": 0.7032,
38
  "step": 25
39
  },
40
  {
41
  "epoch": 0.07,
42
+ "learning_rate": 4.522569407168996e-06,
43
+ "loss": 0.6814,
44
  "step": 30
45
  },
46
  {
47
  "epoch": 0.08,
48
+ "learning_rate": 3.7169372193369285e-06,
49
+ "loss": 0.7428,
50
  "step": 35
51
  },
52
  {
53
  "epoch": 0.09,
54
+ "learning_rate": 2.988288675715626e-06,
55
+ "loss": 0.6695,
56
  "step": 40
57
  },
58
  {
59
  "epoch": 0.1,
60
+ "learning_rate": 2.3374883165699825e-06,
61
+ "loss": 0.6716,
62
  "step": 45
63
  },
64
  {
65
  "epoch": 0.11,
66
+ "learning_rate": 1.7653083154249965e-06,
67
+ "loss": 0.6442,
68
  "step": 50
69
  },
70
  {
71
  "epoch": 0.12,
72
+ "learning_rate": 1.2724275628843243e-06,
73
+ "loss": 0.7111,
74
  "step": 55
75
  },
76
  {
77
  "epoch": 0.13,
78
+ "learning_rate": 8.594308611261225e-07,
79
+ "loss": 0.717,
80
  "step": 60
81
  },
82
  {
83
  "epoch": 0.14,
84
+ "learning_rate": 5.26808230038689e-07,
85
+ "loss": 0.6728,
86
  "step": 65
87
  },
88
  {
89
  "epoch": 0.15,
90
+ "learning_rate": 2.7495432581104424e-07,
91
+ "loss": 0.5893,
92
  "step": 70
93
  },
94
  {
95
  "epoch": 0.16,
96
+ "learning_rate": 1.0416797267454092e-07,
97
+ "loss": 0.6088,
98
  "step": 75
99
  },
100
  {
101
  "epoch": 0.18,
102
+ "learning_rate": 1.465180834793276e-08,
103
+ "loss": 0.6029,
104
  "step": 80
105
  },
106
  {
107
  "epoch": 0.19,
108
+ "learning_rate": 6.5120436089167734e-09,
109
+ "loss": 0.6099,
110
  "step": 85
111
  },
112
  {
113
  "epoch": 0.2,
114
+ "learning_rate": 7.975833627447873e-08,
115
+ "loss": 0.5843,
116
  "step": 90
117
  },
118
  {
119
  "epoch": 0.21,
120
+ "learning_rate": 2.3430377974243327e-07,
121
+ "loss": 0.6587,
122
  "step": 95
123
  },
124
  {
125
  "epoch": 0.22,
126
+ "learning_rate": 4.699650061053238e-07,
127
+ "loss": 0.5792,
128
  "step": 100
129
  },
130
  {
131
  "epoch": 0.23,
132
+ "learning_rate": 7.864624037171852e-07,
133
+ "loss": 0.6884,
134
  "step": 105
135
  },
136
  {
137
  "epoch": 0.24,
138
+ "learning_rate": 1.1834204489510753e-06,
139
+ "loss": 0.6179,
140
  "step": 110
141
  },
142
  {
143
  "epoch": 0.25,
144
+ "learning_rate": 1.6603681517579621e-06,
145
+ "loss": 0.6588,
146
  "step": 115
147
  },
148
  {
149
  "epoch": 0.26,
150
+ "learning_rate": 2.2167396144949356e-06,
151
+ "loss": 0.5756,
152
  "step": 120
153
  },
154
  {
155
  "epoch": 0.27,
156
+ "learning_rate": 2.851874703363704e-06,
157
+ "loss": 0.7218,
158
  "step": 125
159
  },
160
  {
161
+ "epoch": 0.29,
162
+ "learning_rate": 3.565019831655522e-06,
163
+ "loss": 0.6423,
164
  "step": 130
165
  },
166
  {
167
  "epoch": 0.3,
168
+ "learning_rate": 4.3553288538822644e-06,
169
+ "loss": 0.7181,
170
  "step": 135
171
  },
172
  {
173
  "epoch": 0.31,
174
+ "learning_rate": 5.221864069725448e-06,
175
+ "loss": 0.7147,
176
  "step": 140
177
  },
178
  {
179
  "epoch": 0.32,
180
+ "learning_rate": 6.163597336619036e-06,
181
+ "loss": 0.6411,
182
  "step": 145
183
  },
184
  {
185
  "epoch": 0.33,
186
+ "learning_rate": 7.179411289634524e-06,
187
+ "loss": 0.6499,
188
  "step": 150
189
  },
190
  {
191
  "epoch": 0.34,
192
+ "learning_rate": 8.268100667234415e-06,
193
+ "loss": 0.6962,
194
  "step": 155
195
  },
196
  {
197
  "epoch": 0.35,
198
+ "learning_rate": 9.428373741315982e-06,
199
+ "loss": 0.6569,
200
  "step": 160
201
  },
202
  {
203
  "epoch": 0.36,
204
+ "learning_rate": 1.0658853849838009e-05,
205
+ "loss": 0.6364,
206
  "step": 165
207
  },
208
  {
209
  "epoch": 0.37,
210
+ "learning_rate": 1.1958081030236758e-05,
211
+ "loss": 0.6697,
212
  "step": 170
213
  },
214
  {
215
  "epoch": 0.38,
216
+ "learning_rate": 1.3324513751656602e-05,
217
+ "loss": 0.6964,
218
  "step": 175
219
  },
220
  {
221
  "epoch": 0.39,
222
+ "learning_rate": 1.47565307439888e-05,
223
+ "loss": 0.6235,
224
  "step": 180
225
  },
226
  {
227
+ "epoch": 0.41,
228
+ "learning_rate": 1.6252432921492454e-05,
229
+ "loss": 0.6185,
230
  "step": 185
231
  },
232
  {
233
  "epoch": 0.42,
234
+ "learning_rate": 1.7810445398767395e-05,
235
+ "loss": 0.7024,
236
  "step": 190
237
  },
238
  {
239
  "epoch": 0.43,
240
+ "learning_rate": 1.9428719596644185e-05,
241
+ "loss": 0.6126,
242
  "step": 195
243
  },
244
  {
245
  "epoch": 0.44,
246
+ "learning_rate": 2.1105335435523642e-05,
247
+ "loss": 0.6189,
248
  "step": 200
249
  },
250
  {
251
  "epoch": 0.45,
252
+ "learning_rate": 2.283830361355242e-05,
253
+ "loss": 0.5719,
254
  "step": 205
255
  },
256
  {
257
  "epoch": 0.46,
258
+ "learning_rate": 2.4625567966918e-05,
259
+ "loss": 0.7211,
260
  "step": 210
261
  },
262
  {
263
  "epoch": 0.47,
264
+ "learning_rate": 2.6465007909489048e-05,
265
+ "loss": 0.6735,
266
  "step": 215
267
  },
268
  {
269
  "epoch": 0.48,
270
+ "learning_rate": 2.8354440948884326e-05,
271
+ "loss": 0.5916,
272
  "step": 220
273
  },
274
  {
275
  "epoch": 0.49,
276
+ "learning_rate": 3.0291625276002506e-05,
277
+ "loss": 0.5753,
278
  "step": 225
279
  },
280
  {
281
  "epoch": 0.5,
282
+ "learning_rate": 3.2274262424915514e-05,
283
+ "loss": 0.6077,
284
  "step": 230
285
  },
286
  {
287
+ "epoch": 0.52,
288
+ "learning_rate": 3.430000000000064e-05,
289
+ "loss": 0.7398,
290
  "step": 235
291
  },
292
  {
293
  "epoch": 0.53,
294
+ "learning_rate": 3.636643446705075e-05,
295
+ "loss": 0.7589,
296
  "step": 240
297
  },
298
  {
299
  "epoch": 0.54,
300
+ "learning_rate": 3.847111400507036e-05,
301
+ "loss": 0.7287,
302
  "step": 245
303
  },
304
  {
305
  "epoch": 0.55,
306
+ "learning_rate": 4.0611541415345134e-05,
307
+ "loss": 0.7015,
308
  "step": 250
309
  },
310
  {
311
  "epoch": 0.56,
312
+ "learning_rate": 4.2785177084377745e-05,
313
+ "loss": 0.6185,
314
  "step": 255
315
  },
316
  {
317
  "epoch": 0.57,
318
+ "learning_rate": 4.49894419971228e-05,
319
+ "loss": 0.6638,
320
  "step": 260
321
  },
322
  {
323
  "epoch": 0.58,
324
+ "learning_rate": 4.7221720796988937e-05,
325
+ "loss": 0.6769,
326
  "step": 265
327
  },
328
  {
329
  "epoch": 0.59,
330
+ "learning_rate": 4.947936488896066e-05,
331
+ "loss": 0.7237,
332
  "step": 270
333
  },
334
  {
335
  "epoch": 0.6,
336
+ "learning_rate": 5.1759695582140806e-05,
337
+ "loss": 0.7138,
338
  "step": 275
339
  },
340
  {
341
  "epoch": 0.61,
342
+ "learning_rate": 5.4060007268020076e-05,
343
+ "loss": 0.655,
344
  "step": 280
345
  },
346
  {
347
  "epoch": 0.62,
348
+ "learning_rate": 5.637757063067322e-05,
349
+ "loss": 0.7932,
350
  "step": 285
351
  },
352
  {
353
+ "epoch": 0.64,
354
+ "learning_rate": 5.870963588509545e-05,
355
+ "loss": 0.6143,
356
  "step": 290
357
  },
358
  {
359
  "epoch": 0.65,
360
+ "learning_rate": 6.105343603980619e-05,
361
+ "loss": 0.6686,
362
  "step": 295
363
  },
364
  {
365
  "epoch": 0.66,
366
+ "learning_rate": 6.34061901798874e-05,
367
+ "loss": 0.7298,
368
  "step": 300
369
  },
370
  {
371
  "epoch": 0.67,
372
+ "learning_rate": 6.576510676653068e-05,
373
+ "loss": 0.6932,
374
  "step": 305
375
  },
376
  {
377
  "epoch": 0.68,
378
+ "learning_rate": 6.812738694920095e-05,
379
+ "loss": 0.7121,
380
  "step": 310
381
  },
382
  {
383
  "epoch": 0.69,
384
+ "learning_rate": 7.049022788645635e-05,
385
+ "loss": 0.668,
386
  "step": 315
387
  },
388
  {
389
  "epoch": 0.7,
390
+ "learning_rate": 7.285082607152067e-05,
391
+ "loss": 0.6626,
392
  "step": 320
393
  },
394
  {
395
  "epoch": 0.71,
396
+ "learning_rate": 7.52063806586452e-05,
397
+ "loss": 0.7163,
398
  "step": 325
399
  },
400
  {
401
  "epoch": 0.72,
402
+ "learning_rate": 7.755409678629386e-05,
403
+ "loss": 0.753,
404
  "step": 330
405
  },
406
  {
407
  "epoch": 0.73,
408
+ "learning_rate": 7.989118889325919e-05,
409
+ "loss": 0.6771,
410
  "step": 335
411
  },
412
  {
413
+ "epoch": 0.75,
414
+ "learning_rate": 8.221488402370275e-05,
415
+ "loss": 0.6505,
416
  "step": 340
417
  },
418
  {
419
+ "epoch": 0.76,
420
+ "learning_rate": 8.452242511728957e-05,
421
+ "loss": 0.6098,
422
  "step": 345
423
  },
424
  {
425
  "epoch": 0.77,
426
+ "learning_rate": 8.681107428041253e-05,
427
+ "loss": 0.6978,
428
  "step": 350
429
  },
430
  {
431
  "epoch": 0.78,
432
+ "learning_rate": 8.907811603471558e-05,
433
+ "loss": 0.6606,
434
  "step": 355
435
  },
436
  {
437
  "epoch": 0.79,
438
+ "learning_rate": 9.132086053899106e-05,
439
+ "loss": 0.788,
440
  "step": 360
441
  },
442
  {
443
  "epoch": 0.8,
444
+ "learning_rate": 9.353664678067919e-05,
445
+ "loss": 0.7524,
446
  "step": 365
447
  },
448
  {
449
  "epoch": 0.81,
450
+ "learning_rate": 9.572284573315139e-05,
451
+ "loss": 0.6861,
452
  "step": 370
453
  },
454
  {
455
  "epoch": 0.82,
456
+ "learning_rate": 9.787686347505305e-05,
457
+ "loss": 0.6812,
458
  "step": 375
459
  },
460
  {
461
  "epoch": 0.83,
462
+ "learning_rate": 9.999614426797605e-05,
463
+ "loss": 0.6453,
464
  "step": 380
465
  },
466
  {
467
  "epoch": 0.84,
468
+ "learning_rate": 0.00010207817358884356,
469
+ "loss": 0.7003,
470
  "step": 385
471
  },
472
  {
473
+ "epoch": 0.86,
474
+ "learning_rate": 0.00010412048111339246,
475
+ "loss": 0.714,
476
  "step": 390
477
  },
478
  {
479
+ "epoch": 0.87,
480
+ "learning_rate": 0.00010612064364719859,
481
+ "loss": 0.7008,
482
  "step": 395
483
  },
484
  {
485
  "epoch": 0.88,
486
+ "learning_rate": 0.00010807628800079975,
487
+ "loss": 0.6857,
488
  "step": 400
489
  },
490
  {
491
  "epoch": 0.89,
492
+ "learning_rate": 0.00010998509380547761,
493
+ "loss": 0.7609,
494
  "step": 405
495
  },
496
  {
497
  "epoch": 0.9,
498
+ "learning_rate": 0.00011184479626637769,
499
+ "loss": 0.7942,
500
  "step": 410
501
  },
502
  {
503
  "epoch": 0.91,
504
+ "learning_rate": 0.00011365318884967352,
505
+ "loss": 0.6518,
506
  "step": 415
507
  },
508
  {
509
  "epoch": 0.92,
510
+ "learning_rate": 0.00011540812590062915,
511
+ "loss": 0.7253,
512
  "step": 420
513
  },
514
  {
515
  "epoch": 0.93,
516
+ "learning_rate": 0.000117107525189397,
517
+ "loss": 0.7335,
518
  "step": 425
519
  },
520
  {
521
  "epoch": 0.94,
522
+ "learning_rate": 0.00011874937038159925,
523
+ "loss": 0.7592,
524
  "step": 430
525
  },
526
  {
527
  "epoch": 0.95,
528
+ "learning_rate": 0.00012033171343069444,
529
+ "loss": 0.7893,
530
  "step": 435
531
  },
532
  {
533
  "epoch": 0.96,
534
+ "learning_rate": 0.00012185267688933072,
535
+ "loss": 0.7095,
536
  "step": 440
537
  },
538
  {
539
+ "epoch": 0.98,
540
+ "learning_rate": 0.00012331045613693775,
541
+ "loss": 0.7761,
542
  "step": 445
543
  },
544
  {
545
+ "epoch": 0.99,
546
+ "learning_rate": 0.0001247033215209036,
547
+ "loss": 0.6766,
548
  "step": 450
549
  },
550
  {
551
  "epoch": 1.0,
552
+ "learning_rate": 0.00012602962040880915,
553
+ "loss": 0.7222,
554
  "step": 455
555
  },
556
  {
557
  "epoch": 1.0,
558
+ "eval_loss": 0.41102197766304016,
559
+ "eval_runtime": 15.2365,
560
+ "eval_samples_per_second": 41.742,
561
+ "eval_steps_per_second": 5.251,
562
+ "step": 456
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
563
  }
564
  ],
565
+ "max_steps": 1368,
566
  "num_train_epochs": 3,
567
+ "total_flos": 475812790272000.0,
568
  "trial_name": null,
569
  "trial_params": null
570
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:805b5eda2fc68bf70f2494e9471c8381559074bf1249ef0daf4c166fcf463efc
3
  size 3055
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d9d4e84835a869c4c9d855147718c055221c1c706451dc866b663d99f0469f6
3
  size 3055