AlekseyKorshuk commited on
Commit
be1285a
1 Parent(s): bb97fc0

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/3q37f1zd/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/3u62y0tb) for full transparency and reproducibility.
55
 
56
- At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/3u62y0tb/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/3q82f3by/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/21pupjqd) for full transparency and reproducibility.
55
 
56
+ At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/21pupjqd/artifacts) is logged and versioned.
57
 
58
  ## How to use
59
 
config.json CHANGED
@@ -35,7 +35,7 @@
35
  }
36
  },
37
  "torch_dtype": "float32",
38
- "transformers_version": "4.10.2",
39
  "use_cache": true,
40
  "vocab_size": 50257
41
  }
 
35
  }
36
  },
37
  "torch_dtype": "float32",
38
+ "transformers_version": "4.10.3",
39
  "use_cache": true,
40
  "vocab_size": 50257
41
  }
evaluation.txt CHANGED
@@ -1 +1 @@
1
- {"eval_loss": 1.000430703163147, "eval_runtime": 28.8376, "eval_samples_per_second": 22.644, "eval_steps_per_second": 2.844, "epoch": 2.0}
 
1
+ {"eval_loss": 1.0850846767425537, "eval_runtime": 31.9854, "eval_samples_per_second": 20.666, "eval_steps_per_second": 2.595, "epoch": 2.0}
flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:905d47515ed8012c1522a97cf3c5ec368e9c0b5de761fdd5a1981d8c91e3ebbd
3
  size 497764120
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f5277b428ad69317c468bbc220cef6229159db8e741a69d9a9511b002bcc8bd
3
  size 497764120
optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e308675c9f383331794ee7f24a23e7346f1bc4b6db0befd8a253356df5ccc833
3
  size 995604017
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:553799fcd9709b41c5742b8d21b8a6d378c74829ba270197872aa7d424eca183
3
  size 995604017
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e88b6b107844dd300f000a65c9516549403da254ae63d3af928eee0ddd2ed31f
3
  size 510403817
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f644f9500ffb69c2f5a8973f42e17d2431ae3161f1423c38674ffd177b8ebd89
3
  size 510403817
rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:59e2e6dfadd1068d6503a67af1b4b3c98a7328859872f6c2dd34af82e8e434ce
3
  size 14503
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f74cebe65c91ab590f337b232a799c58364fbd9a30704cba6b25c380d7f9e724
3
  size 14503
scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dcaff93017b04d3e8a9b68caa2cafbfa18e37b27621754efa991e70ea4396de0
3
  size 623
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f39f036bb76f470586f077b34f42cae2b8baa0ccc2883838439fd9b58bd619e
3
  size 623
trainer_state.json CHANGED
@@ -1,572 +1,564 @@
1
  {
2
- "best_metric": 1.000430703163147,
3
- "best_model_checkpoint": "output/eminem/checkpoint-454",
4
  "epoch": 1.0,
5
- "global_step": 454,
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.0001347061084611844,
13
- "loss": 1.5687,
14
  "step": 5
15
  },
16
  {
17
  "epoch": 0.02,
18
- "learning_rate": 0.00013403094091447146,
19
- "loss": 1.4406,
20
  "step": 10
21
  },
22
  {
23
  "epoch": 0.03,
24
- "learning_rate": 0.00013327710810466772,
25
- "loss": 1.4875,
26
  "step": 15
27
  },
28
  {
29
  "epoch": 0.04,
30
- "learning_rate": 0.00013244551633780427,
31
- "loss": 1.3898,
32
  "step": 20
33
  },
34
  {
35
  "epoch": 0.06,
36
- "learning_rate": 0.0001315371654067031,
37
- "loss": 1.6245,
38
  "step": 25
39
  },
40
  {
41
  "epoch": 0.07,
42
- "learning_rate": 0.00013055314738896493,
43
- "loss": 1.4266,
44
  "step": 30
45
  },
46
  {
47
  "epoch": 0.08,
48
- "learning_rate": 0.00012949464533400422,
49
- "loss": 1.5298,
50
  "step": 35
51
  },
52
  {
53
  "epoch": 0.09,
54
- "learning_rate": 0.00012836293184070698,
55
- "loss": 1.4604,
56
  "step": 40
57
  },
58
  {
59
  "epoch": 0.1,
60
- "learning_rate": 0.0001271593675274372,
61
- "loss": 1.4618,
62
  "step": 45
63
  },
64
  {
65
  "epoch": 0.11,
66
- "learning_rate": 0.00012588539939621004,
67
- "loss": 1.2952,
68
  "step": 50
69
  },
70
  {
71
  "epoch": 0.12,
72
- "learning_rate": 0.00012454255909301294,
73
- "loss": 1.5173,
74
  "step": 55
75
  },
76
  {
77
  "epoch": 0.13,
78
- "learning_rate": 0.00012313246106636778,
79
- "loss": 1.4386,
80
  "step": 60
81
  },
82
  {
83
  "epoch": 0.14,
84
- "learning_rate": 0.00012165680062633223,
85
- "loss": 1.5353,
86
  "step": 65
87
  },
88
  {
89
  "epoch": 0.15,
90
- "learning_rate": 0.00012011735190629218,
91
- "loss": 1.5712,
92
  "step": 70
93
  },
94
  {
95
  "epoch": 0.17,
96
- "learning_rate": 0.0001185159657299846,
97
- "loss": 1.5336,
98
  "step": 75
99
  },
100
  {
101
  "epoch": 0.18,
102
- "learning_rate": 0.0001168545673863188,
103
- "loss": 1.4126,
104
  "step": 80
105
  },
106
  {
107
  "epoch": 0.19,
108
- "learning_rate": 0.00011513515431467095,
109
- "loss": 1.4119,
110
  "step": 85
111
  },
112
  {
113
  "epoch": 0.2,
114
- "learning_rate": 0.00011335979370343469,
115
- "loss": 1.4324,
116
  "step": 90
117
  },
118
  {
119
  "epoch": 0.21,
120
- "learning_rate": 0.00011153062000471494,
121
- "loss": 1.4149,
122
  "step": 95
123
  },
124
  {
125
  "epoch": 0.22,
126
- "learning_rate": 0.00010964983236815233,
127
- "loss": 1.6004,
128
  "step": 100
129
  },
130
  {
131
  "epoch": 0.23,
132
- "learning_rate": 0.00010771969199696865,
133
- "loss": 1.4738,
134
  "step": 105
135
  },
136
  {
137
  "epoch": 0.24,
138
- "learning_rate": 0.0001057425194293951,
139
- "loss": 1.5572,
140
  "step": 110
141
  },
142
  {
143
  "epoch": 0.25,
144
- "learning_rate": 0.00010372069174878459,
145
- "loss": 1.5809,
146
  "step": 115
147
  },
148
  {
149
- "epoch": 0.26,
150
- "learning_rate": 0.00010165663972571805,
151
- "loss": 1.3915,
152
  "step": 120
153
  },
154
  {
155
  "epoch": 0.28,
156
- "learning_rate": 9.955284489558465e-05,
157
- "loss": 1.5034,
158
  "step": 125
159
  },
160
  {
161
  "epoch": 0.29,
162
- "learning_rate": 9.741183657512136e-05,
163
- "loss": 1.5812,
164
  "step": 130
165
  },
166
  {
167
  "epoch": 0.3,
168
- "learning_rate": 9.523618882149785e-05,
169
- "loss": 1.5666,
170
  "step": 135
171
  },
172
  {
173
  "epoch": 0.31,
174
- "learning_rate": 9.30285173376325e-05,
175
- "loss": 1.2922,
176
  "step": 140
177
  },
178
  {
179
  "epoch": 0.32,
180
- "learning_rate": 9.079147632741836e-05,
181
- "loss": 1.5392,
182
  "step": 145
183
  },
184
  {
185
  "epoch": 0.33,
186
- "learning_rate": 8.85277553046691e-05,
187
- "loss": 1.5071,
188
  "step": 150
189
  },
190
  {
191
  "epoch": 0.34,
192
- "learning_rate": 8.624007585962182e-05,
193
- "loss": 1.5317,
194
  "step": 155
195
  },
196
  {
197
  "epoch": 0.35,
198
- "learning_rate": 8.393118838685801e-05,
199
- "loss": 1.5537,
200
  "step": 160
201
  },
202
  {
203
- "epoch": 0.36,
204
- "learning_rate": 8.160386877860775e-05,
205
- "loss": 1.5394,
206
  "step": 165
207
  },
208
  {
209
  "epoch": 0.38,
210
- "learning_rate": 7.926091508739358e-05,
211
- "loss": 1.4322,
212
  "step": 170
213
  },
214
  {
215
  "epoch": 0.39,
216
- "learning_rate": 7.690514416203279e-05,
217
- "loss": 1.2707,
218
  "step": 175
219
  },
220
  {
221
  "epoch": 0.4,
222
- "learning_rate": 7.45393882610415e-05,
223
- "loss": 1.6366,
224
  "step": 180
225
  },
226
  {
227
  "epoch": 0.41,
228
- "learning_rate": 7.216649164751235e-05,
229
- "loss": 1.2109,
230
  "step": 185
231
  },
232
  {
233
  "epoch": 0.42,
234
- "learning_rate": 6.978930716955943e-05,
235
- "loss": 1.5598,
236
  "step": 190
237
  },
238
  {
239
  "epoch": 0.43,
240
- "learning_rate": 6.741069283044077e-05,
241
- "loss": 1.6169,
242
  "step": 195
243
  },
244
  {
245
  "epoch": 0.44,
246
- "learning_rate": 6.503350835248882e-05,
247
- "loss": 1.4735,
248
  "step": 200
249
  },
250
  {
251
  "epoch": 0.45,
252
- "learning_rate": 6.266061173895871e-05,
253
- "loss": 1.5091,
254
  "step": 205
255
  },
256
  {
257
  "epoch": 0.46,
258
- "learning_rate": 6.029485583796741e-05,
259
- "loss": 1.3971,
260
  "step": 210
261
  },
262
  {
263
- "epoch": 0.47,
264
- "learning_rate": 5.793908491260758e-05,
265
- "loss": 1.4943,
266
  "step": 215
267
  },
268
  {
269
  "epoch": 0.49,
270
- "learning_rate": 5.559613122139245e-05,
271
- "loss": 1.5372,
272
  "step": 220
273
  },
274
  {
275
  "epoch": 0.5,
276
- "learning_rate": 5.326881161314218e-05,
277
- "loss": 1.5643,
278
  "step": 225
279
  },
280
  {
281
  "epoch": 0.51,
282
- "learning_rate": 5.0959924140378376e-05,
283
- "loss": 1.4382,
284
  "step": 230
285
  },
286
  {
287
  "epoch": 0.52,
288
- "learning_rate": 4.86722446953311e-05,
289
- "loss": 1.5737,
290
  "step": 235
291
  },
292
  {
293
  "epoch": 0.53,
294
- "learning_rate": 4.6408523672582755e-05,
295
- "loss": 1.6657,
296
  "step": 240
297
  },
298
  {
299
  "epoch": 0.54,
300
- "learning_rate": 4.417148266236768e-05,
301
- "loss": 1.2837,
302
  "step": 245
303
  },
304
  {
305
  "epoch": 0.55,
306
- "learning_rate": 4.1963811178502344e-05,
307
- "loss": 1.4313,
308
  "step": 250
309
  },
310
  {
311
  "epoch": 0.56,
312
- "learning_rate": 3.978816342487971e-05,
313
- "loss": 1.4629,
314
  "step": 255
315
  },
316
  {
317
- "epoch": 0.57,
318
- "learning_rate": 3.7647155104415525e-05,
319
- "loss": 1.3243,
320
  "step": 260
321
  },
322
  {
323
- "epoch": 0.58,
324
- "learning_rate": 3.554336027428212e-05,
325
- "loss": 1.3828,
326
  "step": 265
327
  },
328
  {
329
  "epoch": 0.6,
330
- "learning_rate": 3.347930825121643e-05,
331
- "loss": 1.4753,
332
  "step": 270
333
  },
334
  {
335
  "epoch": 0.61,
336
- "learning_rate": 3.145748057060507e-05,
337
- "loss": 1.7174,
338
  "step": 275
339
  },
340
  {
341
  "epoch": 0.62,
342
- "learning_rate": 2.9480308003031502e-05,
343
- "loss": 1.4543,
344
  "step": 280
345
  },
346
  {
347
  "epoch": 0.63,
348
- "learning_rate": 2.755016763184782e-05,
349
- "loss": 1.4277,
350
  "step": 285
351
  },
352
  {
353
  "epoch": 0.64,
354
- "learning_rate": 2.5669379995285212e-05,
355
- "loss": 1.4237,
356
  "step": 290
357
  },
358
  {
359
  "epoch": 0.65,
360
- "learning_rate": 2.3840206296566214e-05,
361
- "loss": 1.4508,
362
  "step": 295
363
  },
364
  {
365
  "epoch": 0.66,
366
- "learning_rate": 2.2064845685329196e-05,
367
- "loss": 1.4977,
368
  "step": 300
369
  },
370
  {
371
  "epoch": 0.67,
372
- "learning_rate": 2.034543261368135e-05,
373
- "loss": 1.5943,
374
  "step": 305
375
  },
376
  {
377
- "epoch": 0.68,
378
- "learning_rate": 1.8684034270016213e-05,
379
- "loss": 1.3977,
380
  "step": 310
381
  },
382
  {
383
  "epoch": 0.7,
384
- "learning_rate": 1.7082648093707956e-05,
385
- "loss": 1.6013,
386
  "step": 315
387
  },
388
  {
389
  "epoch": 0.71,
390
- "learning_rate": 1.5543199373667898e-05,
391
- "loss": 1.6127,
392
  "step": 320
393
  },
394
  {
395
  "epoch": 0.72,
396
- "learning_rate": 1.4067538933632335e-05,
397
- "loss": 1.3648,
398
  "step": 325
399
  },
400
  {
401
  "epoch": 0.73,
402
- "learning_rate": 1.2657440906987175e-05,
403
- "loss": 1.3725,
404
  "step": 330
405
  },
406
  {
407
  "epoch": 0.74,
408
- "learning_rate": 1.1314600603790608e-05,
409
- "loss": 1.5536,
410
  "step": 335
411
  },
412
  {
413
  "epoch": 0.75,
414
- "learning_rate": 1.0040632472562908e-05,
415
- "loss": 1.3899,
416
  "step": 340
417
  },
418
  {
419
  "epoch": 0.76,
420
- "learning_rate": 8.83706815929313e-06,
421
- "loss": 1.6119,
422
  "step": 345
423
  },
424
  {
425
  "epoch": 0.77,
426
- "learning_rate": 7.705354665996331e-06,
427
- "loss": 1.4506,
428
  "step": 350
429
  },
430
  {
431
- "epoch": 0.78,
432
- "learning_rate": 6.64685261103514e-06,
433
- "loss": 1.4292,
434
  "step": 355
435
  },
436
  {
437
- "epoch": 0.79,
438
- "learning_rate": 5.662834593296984e-06,
439
- "loss": 1.3922,
440
  "step": 360
441
  },
442
  {
443
  "epoch": 0.81,
444
- "learning_rate": 4.754483662196166e-06,
445
- "loss": 1.6076,
446
  "step": 365
447
  },
448
  {
449
  "epoch": 0.82,
450
- "learning_rate": 3.9228918953323614e-06,
451
- "loss": 1.4014,
452
  "step": 370
453
  },
454
  {
455
  "epoch": 0.83,
456
- "learning_rate": 3.1690590855285985e-06,
457
- "loss": 1.4242,
458
  "step": 375
459
  },
460
  {
461
  "epoch": 0.84,
462
- "learning_rate": 2.493891538815897e-06,
463
- "loss": 1.4092,
464
  "step": 380
465
  },
466
  {
467
  "epoch": 0.85,
468
- "learning_rate": 1.8982009848114428e-06,
469
- "loss": 1.4806,
470
  "step": 385
471
  },
472
  {
473
  "epoch": 0.86,
474
- "learning_rate": 1.382703600809147e-06,
475
- "loss": 1.4482,
476
  "step": 390
477
  },
478
  {
479
  "epoch": 0.87,
480
- "learning_rate": 9.480191507425794e-07,
481
- "loss": 1.6429,
482
  "step": 395
483
  },
484
  {
485
  "epoch": 0.88,
486
- "learning_rate": 5.946702400664045e-07,
487
- "loss": 1.5814,
488
  "step": 400
489
  },
490
  {
491
- "epoch": 0.89,
492
- "learning_rate": 3.2308168744641287e-07,
493
- "loss": 1.4479,
494
  "step": 405
495
  },
496
  {
497
  "epoch": 0.91,
498
- "learning_rate": 1.3358001401535102e-07,
499
- "loss": 1.3774,
500
  "step": 410
501
  },
502
  {
503
  "epoch": 0.92,
504
- "learning_rate": 2.639305080833756e-08,
505
- "loss": 1.6517,
506
  "step": 415
507
  },
508
  {
509
  "epoch": 0.93,
510
- "learning_rate": 1.6496648497778654e-09,
511
- "loss": 1.3985,
512
  "step": 420
513
  },
514
  {
515
  "epoch": 0.94,
516
- "learning_rate": 5.93796042212353e-08,
517
- "loss": 1.2782,
518
  "step": 425
519
  },
520
  {
521
  "epoch": 0.95,
522
- "learning_rate": 1.9951346229640478e-07,
523
- "loss": 1.516,
524
  "step": 430
525
  },
526
  {
527
  "epoch": 0.96,
528
- "learning_rate": 4.218827611858774e-07,
529
- "loss": 1.3261,
530
  "step": 435
531
  },
532
  {
533
  "epoch": 0.97,
534
- "learning_rate": 7.262201542927391e-07,
535
- "loss": 1.2912,
536
  "step": 440
537
  },
538
  {
539
  "epoch": 0.98,
540
- "learning_rate": 1.112159747733217e-06,
541
- "loss": 1.3938,
542
  "step": 445
543
  },
544
- {
545
- "epoch": 0.99,
546
- "learning_rate": 1.579237540237193e-06,
547
- "loss": 1.4138,
548
- "step": 450
549
- },
550
  {
551
  "epoch": 1.0,
552
- "eval_loss": 1.3114155530929565,
553
- "eval_runtime": 30.7552,
554
- "eval_samples_per_second": 21.46,
555
- "eval_steps_per_second": 2.699,
556
- "step": 453
557
  },
558
  {
559
  "epoch": 1.0,
560
- "eval_loss": 1.000430703163147,
561
- "eval_runtime": 28.2425,
562
- "eval_samples_per_second": 23.121,
563
- "eval_steps_per_second": 2.903,
564
- "step": 454
565
  }
566
  ],
567
- "max_steps": 908,
568
  "num_train_epochs": 2,
569
- "total_flos": 472677285888000.0,
570
  "trial_name": null,
571
  "trial_params": null
572
  }
 
1
  {
2
+ "best_metric": 1.0850846767425537,
3
+ "best_model_checkpoint": "output/eminem/checkpoint-452",
4
  "epoch": 1.0,
5
+ "global_step": 452,
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.0001303175293643722,
13
+ "loss": 1.3814,
14
  "step": 5
15
  },
16
  {
17
  "epoch": 0.02,
18
+ "learning_rate": 0.00012923970346668577,
19
+ "loss": 1.4469,
20
  "step": 10
21
  },
22
  {
23
  "epoch": 0.03,
24
+ "learning_rate": 0.00012808864972555433,
25
+ "loss": 1.3639,
26
  "step": 15
27
  },
28
  {
29
  "epoch": 0.04,
30
+ "learning_rate": 0.00012686575814089952,
31
+ "loss": 1.4233,
32
  "step": 20
33
  },
34
  {
35
  "epoch": 0.06,
36
+ "learning_rate": 0.00012557250546324088,
37
+ "loss": 1.4142,
38
  "step": 25
39
  },
40
  {
41
  "epoch": 0.07,
42
+ "learning_rate": 0.00012421045341038384,
43
+ "loss": 1.5476,
44
  "step": 30
45
  },
46
  {
47
  "epoch": 0.08,
48
+ "learning_rate": 0.00012278124678150593,
49
+ "loss": 1.5554,
50
  "step": 35
51
  },
52
  {
53
  "epoch": 0.09,
54
+ "learning_rate": 0.00012128661147092046,
55
+ "loss": 1.3787,
56
  "step": 40
57
  },
58
  {
59
  "epoch": 0.1,
60
+ "learning_rate": 0.00011972835238390038,
61
+ "loss": 1.504,
62
  "step": 45
63
  },
64
  {
65
  "epoch": 0.11,
66
+ "learning_rate": 0.00011810835125709465,
67
+ "loss": 1.4927,
68
  "step": 50
69
  },
70
  {
71
  "epoch": 0.12,
72
+ "learning_rate": 0.00011642856438616613,
73
+ "loss": 1.543,
74
  "step": 55
75
  },
76
  {
77
  "epoch": 0.13,
78
+ "learning_rate": 0.00011469102026339135,
79
+ "loss": 1.4018,
80
  "step": 60
81
  },
82
  {
83
  "epoch": 0.14,
84
+ "learning_rate": 0.00011289781712806903,
85
+ "loss": 1.5591,
86
  "step": 65
87
  },
88
  {
89
  "epoch": 0.15,
90
+ "learning_rate": 0.0001110511204327154,
91
+ "loss": 1.2862,
92
  "step": 70
93
  },
94
  {
95
  "epoch": 0.17,
96
+ "learning_rate": 0.00010915316022808772,
97
+ "loss": 1.5977,
98
  "step": 75
99
  },
100
  {
101
  "epoch": 0.18,
102
+ "learning_rate": 0.00010720622847019248,
103
+ "loss": 1.2878,
104
  "step": 80
105
  },
106
  {
107
  "epoch": 0.19,
108
+ "learning_rate": 0.00010521267625255193,
109
+ "loss": 1.2948,
110
  "step": 85
111
  },
112
  {
113
  "epoch": 0.2,
114
+ "learning_rate": 0.00010317491096704546,
115
+ "loss": 1.2196,
116
  "step": 90
117
  },
118
  {
119
  "epoch": 0.21,
120
+ "learning_rate": 0.00010109539339677022,
121
+ "loss": 1.4123,
122
  "step": 95
123
  },
124
  {
125
  "epoch": 0.22,
126
+ "learning_rate": 9.897663474442627e-05,
127
+ "loss": 1.3245,
128
  "step": 100
129
  },
130
  {
131
  "epoch": 0.23,
132
+ "learning_rate": 9.682119359981574e-05,
133
+ "loss": 1.3909,
134
  "step": 105
135
  },
136
  {
137
  "epoch": 0.24,
138
+ "learning_rate": 9.463167285011763e-05,
139
+ "loss": 1.458,
140
  "step": 110
141
  },
142
  {
143
  "epoch": 0.25,
144
+ "learning_rate": 9.241071653666979e-05,
145
+ "loss": 1.3268,
146
  "step": 115
147
  },
148
  {
149
+ "epoch": 0.27,
150
+ "learning_rate": 9.016100666205339e-05,
151
+ "loss": 1.3456,
152
  "step": 120
153
  },
154
  {
155
  "epoch": 0.28,
156
+ "learning_rate": 8.788525995133601e-05,
157
+ "loss": 1.2974,
158
  "step": 125
159
  },
160
  {
161
  "epoch": 0.29,
162
+ "learning_rate": 8.558622457138398e-05,
163
+ "loss": 1.3235,
164
  "step": 130
165
  },
166
  {
167
  "epoch": 0.3,
168
+ "learning_rate": 8.326667681220627e-05,
169
+ "loss": 1.416,
170
  "step": 135
171
  },
172
  {
173
  "epoch": 0.31,
174
+ "learning_rate": 8.092941773433681e-05,
175
+ "loss": 1.5582,
176
  "step": 140
177
  },
178
  {
179
  "epoch": 0.32,
180
+ "learning_rate": 7.857726978630461e-05,
181
+ "loss": 1.5253,
182
  "step": 145
183
  },
184
  {
185
  "epoch": 0.33,
186
+ "learning_rate": 7.621307339627555e-05,
187
+ "loss": 1.3431,
188
  "step": 150
189
  },
190
  {
191
  "epoch": 0.34,
192
+ "learning_rate": 7.383968354198248e-05,
193
+ "loss": 1.4096,
194
  "step": 155
195
  },
196
  {
197
  "epoch": 0.35,
198
+ "learning_rate": 7.145996630308523e-05,
199
+ "loss": 1.5607,
200
  "step": 160
201
  },
202
  {
203
+ "epoch": 0.37,
204
+ "learning_rate": 6.907679540012421e-05,
205
+ "loss": 1.3956,
206
  "step": 165
207
  },
208
  {
209
  "epoch": 0.38,
210
+ "learning_rate": 6.669304872424775e-05,
211
+ "loss": 1.4131,
212
  "step": 170
213
  },
214
  {
215
  "epoch": 0.39,
216
+ "learning_rate": 6.431160486189736e-05,
217
+ "loss": 1.5354,
218
  "step": 175
219
  },
220
  {
221
  "epoch": 0.4,
222
+ "learning_rate": 6.193533961866798e-05,
223
+ "loss": 1.4403,
224
  "step": 180
225
  },
226
  {
227
  "epoch": 0.41,
228
+ "learning_rate": 5.9567122546507705e-05,
229
+ "loss": 1.5146,
230
  "step": 185
231
  },
232
  {
233
  "epoch": 0.42,
234
+ "learning_rate": 5.7209813478478305e-05,
235
+ "loss": 1.4112,
236
  "step": 190
237
  },
238
  {
239
  "epoch": 0.43,
240
+ "learning_rate": 5.48662590752607e-05,
241
+ "loss": 1.5145,
242
  "step": 195
243
  },
244
  {
245
  "epoch": 0.44,
246
+ "learning_rate": 5.2539289387549954e-05,
247
+ "loss": 1.4062,
248
  "step": 200
249
  },
250
  {
251
  "epoch": 0.45,
252
+ "learning_rate": 5.023171443851796e-05,
253
+ "loss": 1.2077,
254
  "step": 205
255
  },
256
  {
257
  "epoch": 0.46,
258
+ "learning_rate": 4.7946320830464014e-05,
259
+ "loss": 1.3632,
260
  "step": 210
261
  },
262
  {
263
+ "epoch": 0.48,
264
+ "learning_rate": 4.568586837974509e-05,
265
+ "loss": 1.3578,
266
  "step": 215
267
  },
268
  {
269
  "epoch": 0.49,
270
+ "learning_rate": 4.3453086784043506e-05,
271
+ "loss": 1.2558,
272
  "step": 220
273
  },
274
  {
275
  "epoch": 0.5,
276
+ "learning_rate": 4.1250672326022146e-05,
277
+ "loss": 1.4552,
278
  "step": 225
279
  },
280
  {
281
  "epoch": 0.51,
282
+ "learning_rate": 3.908128461732282e-05,
283
+ "loss": 1.4219,
284
  "step": 230
285
  },
286
  {
287
  "epoch": 0.52,
288
+ "learning_rate": 3.694754338684033e-05,
289
+ "loss": 1.297,
290
  "step": 235
291
  },
292
  {
293
  "epoch": 0.53,
294
+ "learning_rate": 3.485202531717449e-05,
295
+ "loss": 1.4082,
296
  "step": 240
297
  },
298
  {
299
  "epoch": 0.54,
300
+ "learning_rate": 3.279726093305622e-05,
301
+ "loss": 1.3862,
302
  "step": 245
303
  },
304
  {
305
  "epoch": 0.55,
306
+ "learning_rate": 3.0785731545505864e-05,
307
+ "loss": 1.3033,
308
  "step": 250
309
  },
310
  {
311
  "epoch": 0.56,
312
+ "learning_rate": 2.8819866255437648e-05,
313
+ "loss": 1.5164,
314
  "step": 255
315
  },
316
  {
317
+ "epoch": 0.58,
318
+ "learning_rate": 2.6902039020298912e-05,
319
+ "loss": 1.331,
320
  "step": 260
321
  },
322
  {
323
+ "epoch": 0.59,
324
+ "learning_rate": 2.503456578730975e-05,
325
+ "loss": 1.3233,
326
  "step": 265
327
  },
328
  {
329
  "epoch": 0.6,
330
+ "learning_rate": 2.3219701696742702e-05,
331
+ "loss": 1.3359,
332
  "step": 270
333
  },
334
  {
335
  "epoch": 0.61,
336
+ "learning_rate": 2.1459638358647056e-05,
337
+ "loss": 1.3935,
338
  "step": 275
339
  },
340
  {
341
  "epoch": 0.62,
342
+ "learning_rate": 1.9756501206284136e-05,
343
+ "loss": 1.3479,
344
  "step": 280
345
  },
346
  {
347
  "epoch": 0.63,
348
+ "learning_rate": 1.8112346929471017e-05,
349
+ "loss": 1.2397,
350
  "step": 285
351
  },
352
  {
353
  "epoch": 0.64,
354
+ "learning_rate": 1.652916099095146e-05,
355
+ "loss": 1.2953,
356
  "step": 290
357
  },
358
  {
359
  "epoch": 0.65,
360
+ "learning_rate": 1.5008855228769237e-05,
361
+ "loss": 1.2815,
362
  "step": 295
363
  },
364
  {
365
  "epoch": 0.66,
366
+ "learning_rate": 1.3553265547554126e-05,
367
+ "loss": 1.4089,
368
  "step": 300
369
  },
370
  {
371
  "epoch": 0.67,
372
+ "learning_rate": 1.2164149701503507e-05,
373
+ "loss": 1.4801,
374
  "step": 305
375
  },
376
  {
377
+ "epoch": 0.69,
378
+ "learning_rate": 1.084318517173762e-05,
379
+ "loss": 1.3129,
380
  "step": 310
381
  },
382
  {
383
  "epoch": 0.7,
384
+ "learning_rate": 9.59196714059168e-06,
385
+ "loss": 1.4936,
386
  "step": 315
387
  },
388
  {
389
  "epoch": 0.71,
390
+ "learning_rate": 8.412006565291113e-06,
391
+ "loss": 1.5641,
392
  "step": 320
393
  },
394
  {
395
  "epoch": 0.72,
396
+ "learning_rate": 7.304728353336137e-06,
397
+ "loss": 1.5021,
398
  "step": 325
399
  },
400
  {
401
  "epoch": 0.73,
402
+ "learning_rate": 6.271469641798991e-06,
403
+ "loss": 1.486,
404
  "step": 330
405
  },
406
  {
407
  "epoch": 0.74,
408
+ "learning_rate": 5.3134781826118575e-06,
409
+ "loss": 1.4864,
410
  "step": 335
411
  },
412
  {
413
  "epoch": 0.75,
414
+ "learning_rate": 4.431910835795231e-06,
415
+ "loss": 1.3247,
416
  "step": 340
417
  },
418
  {
419
  "epoch": 0.76,
420
+ "learning_rate": 3.627832172446406e-06,
421
+ "loss": 1.3021,
422
  "step": 345
423
  },
424
  {
425
  "epoch": 0.77,
426
+ "learning_rate": 2.9022131891750565e-06,
427
+ "loss": 1.2565,
428
  "step": 350
429
  },
430
  {
431
+ "epoch": 0.79,
432
+ "learning_rate": 2.255930135538328e-06,
433
+ "loss": 1.3403,
434
  "step": 355
435
  },
436
  {
437
+ "epoch": 0.8,
438
+ "learning_rate": 1.6897634558914518e-06,
439
+ "loss": 1.1603,
440
  "step": 360
441
  },
442
  {
443
  "epoch": 0.81,
444
+ "learning_rate": 1.2043968469316478e-06,
445
+ "loss": 1.3477,
446
  "step": 365
447
  },
448
  {
449
  "epoch": 0.82,
450
+ "learning_rate": 8.004164320734945e-07,
451
+ "loss": 1.4542,
452
  "step": 370
453
  },
454
  {
455
  "epoch": 0.83,
456
+ "learning_rate": 4.783100536527857e-07,
457
+ "loss": 1.171,
458
  "step": 375
459
  },
460
  {
461
  "epoch": 0.84,
462
+ "learning_rate": 2.384666838129012e-07,
463
+ "loss": 1.3221,
464
  "step": 380
465
  },
466
  {
467
  "epoch": 0.85,
468
+ "learning_rate": 8.117595478741029e-08,
469
+ "loss": 1.4165,
470
  "step": 385
471
  },
472
  {
473
  "epoch": 0.86,
474
+ "learning_rate": 6.627809142273766e-09,
475
+ "loss": 1.3171,
476
  "step": 390
477
  },
478
  {
479
  "epoch": 0.87,
480
+ "learning_rate": 1.4912270403714612e-08,
481
+ "loss": 1.278,
482
  "step": 395
483
  },
484
  {
485
  "epoch": 0.88,
486
+ "learning_rate": 1.0601933434685848e-07,
487
+ "loss": 1.1844,
488
  "step": 400
489
  },
490
  {
491
+ "epoch": 0.9,
492
+ "learning_rate": 2.798389810765e-07,
493
+ "loss": 1.4166,
494
  "step": 405
495
  },
496
  {
497
  "epoch": 0.91,
498
+ "learning_rate": 5.361613078862704e-07,
499
+ "loss": 1.347,
500
  "step": 410
501
  },
502
  {
503
  "epoch": 0.92,
504
+ "learning_rate": 8.746767827346377e-07,
505
+ "loss": 1.2687,
506
  "step": 415
507
  },
508
  {
509
  "epoch": 0.93,
510
+ "learning_rate": 1.294976618031831e-06,
511
+ "loss": 1.2153,
512
  "step": 420
513
  },
514
  {
515
  "epoch": 0.94,
516
+ "learning_rate": 1.7965532642887846e-06,
517
+ "loss": 1.4466,
518
  "step": 425
519
  },
520
  {
521
  "epoch": 0.95,
522
+ "learning_rate": 2.3788010230270847e-06,
523
+ "loss": 1.4541,
524
  "step": 430
525
  },
526
  {
527
  "epoch": 0.96,
528
+ "learning_rate": 3.0410167782133005e-06,
529
+ "loss": 1.646,
530
  "step": 435
531
  },
532
  {
533
  "epoch": 0.97,
534
+ "learning_rate": 3.7824008453366295e-06,
535
+ "loss": 1.478,
536
  "step": 440
537
  },
538
  {
539
  "epoch": 0.98,
540
+ "learning_rate": 4.60205793709664e-06,
541
+ "loss": 1.523,
542
  "step": 445
543
  },
 
 
 
 
 
 
544
  {
545
  "epoch": 1.0,
546
+ "learning_rate": 5.498998244543705e-06,
547
+ "loss": 1.3657,
548
+ "step": 450
 
 
549
  },
550
  {
551
  "epoch": 1.0,
552
+ "eval_loss": 1.0850846767425537,
553
+ "eval_runtime": 31.9582,
554
+ "eval_samples_per_second": 20.683,
555
+ "eval_steps_per_second": 2.597,
556
+ "step": 452
557
  }
558
  ],
559
+ "max_steps": 904,
560
  "num_train_epochs": 2,
561
+ "total_flos": 472415993856000.0,
562
  "trial_name": null,
563
  "trial_params": null
564
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ced0124e38eeda92a1a3b5f83f5b32946b5283d3588b341d682a278fd7ad4e94
3
  size 2671
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:096f065f6af008348552f5cbc74729e8d3c0d2e89068dcc66974910fef5f24ce
3
  size 2671