jazzson commited on
Commit
ffb9413
1 Parent(s): 3c94ce7

End of training

Browse files
Files changed (2) hide show
  1. README.md +3 -1
  2. trainer_state.json +1092 -0
README.md CHANGED
@@ -5,6 +5,8 @@ datasets:
5
  library_name: peft
6
  license: cc-by-nc-4.0
7
  tags:
 
 
8
  - generated_from_trainer
9
  model-index:
10
  - name: musicgen-melody-lora-classic-test4-colab
@@ -16,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # musicgen-melody-lora-classic-test4-colab
18
 
19
- This model is a fine-tuned version of [facebook/musicgen-melody](https://huggingface.co/facebook/musicgen-melody) on the audiofolder dataset.
20
 
21
  ## Model description
22
 
 
5
  library_name: peft
6
  license: cc-by-nc-4.0
7
  tags:
8
+ - text-to-audio
9
+ - classic melody
10
  - generated_from_trainer
11
  model-index:
12
  - name: musicgen-melody-lora-classic-test4-colab
 
18
 
19
  # musicgen-melody-lora-classic-test4-colab
20
 
21
+ This model is a fine-tuned version of [facebook/musicgen-melody](https://huggingface.co/facebook/musicgen-melody) on the jazzson/classic_dataset dataset.
22
 
23
  ## Model description
24
 
trainer_state.json ADDED
@@ -0,0 +1,1092 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 80.0,
5
+ "eval_steps": 500,
6
+ "global_step": 300,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.5333333333333333,
13
+ "grad_norm": 4.345211029052734,
14
+ "learning_rate": 0.00019866666666666668,
15
+ "loss": 5.2267,
16
+ "step": 2
17
+ },
18
+ {
19
+ "epoch": 1.0666666666666667,
20
+ "grad_norm": 2.278362274169922,
21
+ "learning_rate": 0.00019733333333333335,
22
+ "loss": 5.2119,
23
+ "step": 4
24
+ },
25
+ {
26
+ "epoch": 1.6,
27
+ "grad_norm": 2.3086485862731934,
28
+ "learning_rate": 0.000196,
29
+ "loss": 5.0838,
30
+ "step": 6
31
+ },
32
+ {
33
+ "epoch": 2.1333333333333333,
34
+ "grad_norm": 2.830240249633789,
35
+ "learning_rate": 0.0001946666666666667,
36
+ "loss": 5.169,
37
+ "step": 8
38
+ },
39
+ {
40
+ "epoch": 2.6666666666666665,
41
+ "grad_norm": 1.9018843173980713,
42
+ "learning_rate": 0.00019333333333333333,
43
+ "loss": 5.0305,
44
+ "step": 10
45
+ },
46
+ {
47
+ "epoch": 3.2,
48
+ "grad_norm": 1.686331033706665,
49
+ "learning_rate": 0.000192,
50
+ "loss": 4.8707,
51
+ "step": 12
52
+ },
53
+ {
54
+ "epoch": 3.7333333333333334,
55
+ "grad_norm": 1.7581056356430054,
56
+ "learning_rate": 0.00019066666666666668,
57
+ "loss": 4.8496,
58
+ "step": 14
59
+ },
60
+ {
61
+ "epoch": 4.266666666666667,
62
+ "grad_norm": 2.22395920753479,
63
+ "learning_rate": 0.00018933333333333335,
64
+ "loss": 5.0846,
65
+ "step": 16
66
+ },
67
+ {
68
+ "epoch": 4.8,
69
+ "grad_norm": 1.7067149877548218,
70
+ "learning_rate": 0.000188,
71
+ "loss": 4.753,
72
+ "step": 18
73
+ },
74
+ {
75
+ "epoch": 5.333333333333333,
76
+ "grad_norm": 2.107501268386841,
77
+ "learning_rate": 0.0001866666666666667,
78
+ "loss": 4.7453,
79
+ "step": 20
80
+ },
81
+ {
82
+ "epoch": 5.866666666666667,
83
+ "grad_norm": 1.7721563577651978,
84
+ "learning_rate": 0.00018533333333333333,
85
+ "loss": 4.8327,
86
+ "step": 22
87
+ },
88
+ {
89
+ "epoch": 6.4,
90
+ "grad_norm": 1.7392892837524414,
91
+ "learning_rate": 0.00018400000000000003,
92
+ "loss": 4.5603,
93
+ "step": 24
94
+ },
95
+ {
96
+ "epoch": 6.933333333333334,
97
+ "grad_norm": 2.0787057876586914,
98
+ "learning_rate": 0.00018266666666666667,
99
+ "loss": 4.696,
100
+ "step": 26
101
+ },
102
+ {
103
+ "epoch": 7.466666666666667,
104
+ "grad_norm": 1.3806192874908447,
105
+ "learning_rate": 0.00018133333333333334,
106
+ "loss": 4.6356,
107
+ "step": 28
108
+ },
109
+ {
110
+ "epoch": 8.0,
111
+ "grad_norm": 1.6404259204864502,
112
+ "learning_rate": 0.00018,
113
+ "loss": 4.6332,
114
+ "step": 30
115
+ },
116
+ {
117
+ "epoch": 8.533333333333333,
118
+ "grad_norm": 1.2937849760055542,
119
+ "learning_rate": 0.00017866666666666668,
120
+ "loss": 4.6328,
121
+ "step": 32
122
+ },
123
+ {
124
+ "epoch": 9.066666666666666,
125
+ "grad_norm": 1.6056703329086304,
126
+ "learning_rate": 0.00017733333333333335,
127
+ "loss": 4.4918,
128
+ "step": 34
129
+ },
130
+ {
131
+ "epoch": 9.6,
132
+ "grad_norm": 1.609789490699768,
133
+ "learning_rate": 0.00017600000000000002,
134
+ "loss": 4.5626,
135
+ "step": 36
136
+ },
137
+ {
138
+ "epoch": 10.133333333333333,
139
+ "grad_norm": 1.7872248888015747,
140
+ "learning_rate": 0.00017466666666666667,
141
+ "loss": 4.3717,
142
+ "step": 38
143
+ },
144
+ {
145
+ "epoch": 10.666666666666666,
146
+ "grad_norm": 1.3249127864837646,
147
+ "learning_rate": 0.00017333333333333334,
148
+ "loss": 4.461,
149
+ "step": 40
150
+ },
151
+ {
152
+ "epoch": 11.2,
153
+ "grad_norm": 1.613872766494751,
154
+ "learning_rate": 0.000172,
155
+ "loss": 4.4685,
156
+ "step": 42
157
+ },
158
+ {
159
+ "epoch": 11.733333333333333,
160
+ "grad_norm": 1.3189940452575684,
161
+ "learning_rate": 0.00017066666666666668,
162
+ "loss": 4.4823,
163
+ "step": 44
164
+ },
165
+ {
166
+ "epoch": 12.266666666666667,
167
+ "grad_norm": 1.381230354309082,
168
+ "learning_rate": 0.00016933333333333335,
169
+ "loss": 4.3209,
170
+ "step": 46
171
+ },
172
+ {
173
+ "epoch": 12.8,
174
+ "grad_norm": 1.753049612045288,
175
+ "learning_rate": 0.000168,
176
+ "loss": 4.3607,
177
+ "step": 48
178
+ },
179
+ {
180
+ "epoch": 13.333333333333334,
181
+ "grad_norm": 1.791304111480713,
182
+ "learning_rate": 0.0001666666666666667,
183
+ "loss": 4.3454,
184
+ "step": 50
185
+ },
186
+ {
187
+ "epoch": 13.866666666666667,
188
+ "grad_norm": 1.5237058401107788,
189
+ "learning_rate": 0.00016533333333333333,
190
+ "loss": 4.337,
191
+ "step": 52
192
+ },
193
+ {
194
+ "epoch": 14.4,
195
+ "grad_norm": 1.508701205253601,
196
+ "learning_rate": 0.000164,
197
+ "loss": 4.3174,
198
+ "step": 54
199
+ },
200
+ {
201
+ "epoch": 14.933333333333334,
202
+ "grad_norm": 1.3160889148712158,
203
+ "learning_rate": 0.00016266666666666667,
204
+ "loss": 4.3048,
205
+ "step": 56
206
+ },
207
+ {
208
+ "epoch": 15.466666666666667,
209
+ "grad_norm": 2.0741655826568604,
210
+ "learning_rate": 0.00016133333333333334,
211
+ "loss": 4.2962,
212
+ "step": 58
213
+ },
214
+ {
215
+ "epoch": 16.0,
216
+ "grad_norm": 1.9378279447555542,
217
+ "learning_rate": 0.00016,
218
+ "loss": 4.2517,
219
+ "step": 60
220
+ },
221
+ {
222
+ "epoch": 16.533333333333335,
223
+ "grad_norm": 2.626952648162842,
224
+ "learning_rate": 0.00015866666666666668,
225
+ "loss": 4.3802,
226
+ "step": 62
227
+ },
228
+ {
229
+ "epoch": 17.066666666666666,
230
+ "grad_norm": 2.282458782196045,
231
+ "learning_rate": 0.00015733333333333333,
232
+ "loss": 4.0704,
233
+ "step": 64
234
+ },
235
+ {
236
+ "epoch": 17.6,
237
+ "grad_norm": 1.3620326519012451,
238
+ "learning_rate": 0.00015600000000000002,
239
+ "loss": 4.3915,
240
+ "step": 66
241
+ },
242
+ {
243
+ "epoch": 18.133333333333333,
244
+ "grad_norm": 1.2274025678634644,
245
+ "learning_rate": 0.00015466666666666667,
246
+ "loss": 4.0713,
247
+ "step": 68
248
+ },
249
+ {
250
+ "epoch": 18.666666666666668,
251
+ "grad_norm": 1.383685827255249,
252
+ "learning_rate": 0.00015333333333333334,
253
+ "loss": 4.1094,
254
+ "step": 70
255
+ },
256
+ {
257
+ "epoch": 19.2,
258
+ "grad_norm": 1.4509886503219604,
259
+ "learning_rate": 0.000152,
260
+ "loss": 4.1963,
261
+ "step": 72
262
+ },
263
+ {
264
+ "epoch": 19.733333333333334,
265
+ "grad_norm": 1.327390432357788,
266
+ "learning_rate": 0.00015066666666666668,
267
+ "loss": 4.0569,
268
+ "step": 74
269
+ },
270
+ {
271
+ "epoch": 20.266666666666666,
272
+ "grad_norm": 1.6511682271957397,
273
+ "learning_rate": 0.00014933333333333335,
274
+ "loss": 4.2426,
275
+ "step": 76
276
+ },
277
+ {
278
+ "epoch": 20.8,
279
+ "grad_norm": 1.379820466041565,
280
+ "learning_rate": 0.000148,
281
+ "loss": 4.0468,
282
+ "step": 78
283
+ },
284
+ {
285
+ "epoch": 21.333333333333332,
286
+ "grad_norm": 1.4640957117080688,
287
+ "learning_rate": 0.00014666666666666666,
288
+ "loss": 4.0385,
289
+ "step": 80
290
+ },
291
+ {
292
+ "epoch": 21.866666666666667,
293
+ "grad_norm": 1.3565938472747803,
294
+ "learning_rate": 0.00014533333333333333,
295
+ "loss": 4.1265,
296
+ "step": 82
297
+ },
298
+ {
299
+ "epoch": 22.4,
300
+ "grad_norm": 1.7414348125457764,
301
+ "learning_rate": 0.000144,
302
+ "loss": 3.9923,
303
+ "step": 84
304
+ },
305
+ {
306
+ "epoch": 22.933333333333334,
307
+ "grad_norm": 1.5553879737854004,
308
+ "learning_rate": 0.00014266666666666667,
309
+ "loss": 4.1569,
310
+ "step": 86
311
+ },
312
+ {
313
+ "epoch": 23.466666666666665,
314
+ "grad_norm": 1.8400905132293701,
315
+ "learning_rate": 0.00014133333333333334,
316
+ "loss": 4.0139,
317
+ "step": 88
318
+ },
319
+ {
320
+ "epoch": 24.0,
321
+ "grad_norm": 1.6468483209609985,
322
+ "learning_rate": 0.00014,
323
+ "loss": 4.0731,
324
+ "step": 90
325
+ },
326
+ {
327
+ "epoch": 24.533333333333335,
328
+ "grad_norm": 1.2724766731262207,
329
+ "learning_rate": 0.00013866666666666669,
330
+ "loss": 3.963,
331
+ "step": 92
332
+ },
333
+ {
334
+ "epoch": 25.066666666666666,
335
+ "grad_norm": 1.5315759181976318,
336
+ "learning_rate": 0.00013733333333333333,
337
+ "loss": 4.0767,
338
+ "step": 94
339
+ },
340
+ {
341
+ "epoch": 25.6,
342
+ "grad_norm": 1.4627431631088257,
343
+ "learning_rate": 0.00013600000000000003,
344
+ "loss": 4.0264,
345
+ "step": 96
346
+ },
347
+ {
348
+ "epoch": 26.133333333333333,
349
+ "grad_norm": 1.1797367334365845,
350
+ "learning_rate": 0.00013466666666666667,
351
+ "loss": 4.002,
352
+ "step": 98
353
+ },
354
+ {
355
+ "epoch": 26.666666666666668,
356
+ "grad_norm": 1.1659247875213623,
357
+ "learning_rate": 0.00013333333333333334,
358
+ "loss": 3.9323,
359
+ "step": 100
360
+ },
361
+ {
362
+ "epoch": 27.2,
363
+ "grad_norm": 1.1997700929641724,
364
+ "learning_rate": 0.000132,
365
+ "loss": 3.9828,
366
+ "step": 102
367
+ },
368
+ {
369
+ "epoch": 27.733333333333334,
370
+ "grad_norm": 1.3612560033798218,
371
+ "learning_rate": 0.00013066666666666668,
372
+ "loss": 3.9506,
373
+ "step": 104
374
+ },
375
+ {
376
+ "epoch": 28.266666666666666,
377
+ "grad_norm": 1.3943043947219849,
378
+ "learning_rate": 0.00012933333333333332,
379
+ "loss": 3.9771,
380
+ "step": 106
381
+ },
382
+ {
383
+ "epoch": 28.8,
384
+ "grad_norm": 1.4121628999710083,
385
+ "learning_rate": 0.00012800000000000002,
386
+ "loss": 3.9147,
387
+ "step": 108
388
+ },
389
+ {
390
+ "epoch": 29.333333333333332,
391
+ "grad_norm": 1.3936829566955566,
392
+ "learning_rate": 0.00012666666666666666,
393
+ "loss": 3.8693,
394
+ "step": 110
395
+ },
396
+ {
397
+ "epoch": 29.866666666666667,
398
+ "grad_norm": 1.2019048929214478,
399
+ "learning_rate": 0.00012533333333333334,
400
+ "loss": 3.9673,
401
+ "step": 112
402
+ },
403
+ {
404
+ "epoch": 30.4,
405
+ "grad_norm": 1.4029520750045776,
406
+ "learning_rate": 0.000124,
407
+ "loss": 3.9296,
408
+ "step": 114
409
+ },
410
+ {
411
+ "epoch": 30.933333333333334,
412
+ "grad_norm": 1.3513153791427612,
413
+ "learning_rate": 0.00012266666666666668,
414
+ "loss": 3.8406,
415
+ "step": 116
416
+ },
417
+ {
418
+ "epoch": 31.466666666666665,
419
+ "grad_norm": 1.2894293069839478,
420
+ "learning_rate": 0.00012133333333333335,
421
+ "loss": 3.9678,
422
+ "step": 118
423
+ },
424
+ {
425
+ "epoch": 32.0,
426
+ "grad_norm": 1.1996159553527832,
427
+ "learning_rate": 0.00012,
428
+ "loss": 3.824,
429
+ "step": 120
430
+ },
431
+ {
432
+ "epoch": 32.53333333333333,
433
+ "grad_norm": 1.2499544620513916,
434
+ "learning_rate": 0.00011866666666666669,
435
+ "loss": 3.9474,
436
+ "step": 122
437
+ },
438
+ {
439
+ "epoch": 33.06666666666667,
440
+ "grad_norm": 1.5120335817337036,
441
+ "learning_rate": 0.00011733333333333334,
442
+ "loss": 3.8489,
443
+ "step": 124
444
+ },
445
+ {
446
+ "epoch": 33.6,
447
+ "grad_norm": 1.3104084730148315,
448
+ "learning_rate": 0.000116,
449
+ "loss": 3.8567,
450
+ "step": 126
451
+ },
452
+ {
453
+ "epoch": 34.13333333333333,
454
+ "grad_norm": 1.3807789087295532,
455
+ "learning_rate": 0.00011466666666666667,
456
+ "loss": 3.7975,
457
+ "step": 128
458
+ },
459
+ {
460
+ "epoch": 34.666666666666664,
461
+ "grad_norm": 1.4722609519958496,
462
+ "learning_rate": 0.00011333333333333334,
463
+ "loss": 3.9113,
464
+ "step": 130
465
+ },
466
+ {
467
+ "epoch": 35.2,
468
+ "grad_norm": 1.6289302110671997,
469
+ "learning_rate": 0.00011200000000000001,
470
+ "loss": 3.791,
471
+ "step": 132
472
+ },
473
+ {
474
+ "epoch": 35.733333333333334,
475
+ "grad_norm": 1.8681869506835938,
476
+ "learning_rate": 0.00011066666666666667,
477
+ "loss": 3.754,
478
+ "step": 134
479
+ },
480
+ {
481
+ "epoch": 36.266666666666666,
482
+ "grad_norm": 1.5681208372116089,
483
+ "learning_rate": 0.00010933333333333333,
484
+ "loss": 3.6688,
485
+ "step": 136
486
+ },
487
+ {
488
+ "epoch": 36.8,
489
+ "grad_norm": 1.2196656465530396,
490
+ "learning_rate": 0.00010800000000000001,
491
+ "loss": 3.9256,
492
+ "step": 138
493
+ },
494
+ {
495
+ "epoch": 37.333333333333336,
496
+ "grad_norm": 1.4438775777816772,
497
+ "learning_rate": 0.00010666666666666667,
498
+ "loss": 3.7748,
499
+ "step": 140
500
+ },
501
+ {
502
+ "epoch": 37.86666666666667,
503
+ "grad_norm": 1.2203658819198608,
504
+ "learning_rate": 0.00010533333333333332,
505
+ "loss": 3.839,
506
+ "step": 142
507
+ },
508
+ {
509
+ "epoch": 38.4,
510
+ "grad_norm": 1.1427325010299683,
511
+ "learning_rate": 0.00010400000000000001,
512
+ "loss": 3.7725,
513
+ "step": 144
514
+ },
515
+ {
516
+ "epoch": 38.93333333333333,
517
+ "grad_norm": 1.0757108926773071,
518
+ "learning_rate": 0.00010266666666666666,
519
+ "loss": 3.8459,
520
+ "step": 146
521
+ },
522
+ {
523
+ "epoch": 39.46666666666667,
524
+ "grad_norm": 1.3734352588653564,
525
+ "learning_rate": 0.00010133333333333335,
526
+ "loss": 3.64,
527
+ "step": 148
528
+ },
529
+ {
530
+ "epoch": 40.0,
531
+ "grad_norm": 1.3007618188858032,
532
+ "learning_rate": 0.0001,
533
+ "loss": 3.8641,
534
+ "step": 150
535
+ },
536
+ {
537
+ "epoch": 40.53333333333333,
538
+ "grad_norm": 1.1311804056167603,
539
+ "learning_rate": 9.866666666666668e-05,
540
+ "loss": 3.7593,
541
+ "step": 152
542
+ },
543
+ {
544
+ "epoch": 41.06666666666667,
545
+ "grad_norm": 1.4323419332504272,
546
+ "learning_rate": 9.733333333333335e-05,
547
+ "loss": 3.7318,
548
+ "step": 154
549
+ },
550
+ {
551
+ "epoch": 41.6,
552
+ "grad_norm": 1.298574686050415,
553
+ "learning_rate": 9.6e-05,
554
+ "loss": 3.7567,
555
+ "step": 156
556
+ },
557
+ {
558
+ "epoch": 42.13333333333333,
559
+ "grad_norm": 1.4180625677108765,
560
+ "learning_rate": 9.466666666666667e-05,
561
+ "loss": 3.7507,
562
+ "step": 158
563
+ },
564
+ {
565
+ "epoch": 42.666666666666664,
566
+ "grad_norm": 1.311776876449585,
567
+ "learning_rate": 9.333333333333334e-05,
568
+ "loss": 3.7975,
569
+ "step": 160
570
+ },
571
+ {
572
+ "epoch": 43.2,
573
+ "grad_norm": 1.2341830730438232,
574
+ "learning_rate": 9.200000000000001e-05,
575
+ "loss": 3.6367,
576
+ "step": 162
577
+ },
578
+ {
579
+ "epoch": 43.733333333333334,
580
+ "grad_norm": 1.242844820022583,
581
+ "learning_rate": 9.066666666666667e-05,
582
+ "loss": 3.7197,
583
+ "step": 164
584
+ },
585
+ {
586
+ "epoch": 44.266666666666666,
587
+ "grad_norm": 1.2008047103881836,
588
+ "learning_rate": 8.933333333333334e-05,
589
+ "loss": 3.7872,
590
+ "step": 166
591
+ },
592
+ {
593
+ "epoch": 44.8,
594
+ "grad_norm": 1.3553142547607422,
595
+ "learning_rate": 8.800000000000001e-05,
596
+ "loss": 3.5827,
597
+ "step": 168
598
+ },
599
+ {
600
+ "epoch": 45.333333333333336,
601
+ "grad_norm": 1.3207334280014038,
602
+ "learning_rate": 8.666666666666667e-05,
603
+ "loss": 3.8034,
604
+ "step": 170
605
+ },
606
+ {
607
+ "epoch": 45.86666666666667,
608
+ "grad_norm": 1.3472046852111816,
609
+ "learning_rate": 8.533333333333334e-05,
610
+ "loss": 3.7445,
611
+ "step": 172
612
+ },
613
+ {
614
+ "epoch": 46.4,
615
+ "grad_norm": 1.4812192916870117,
616
+ "learning_rate": 8.4e-05,
617
+ "loss": 3.6336,
618
+ "step": 174
619
+ },
620
+ {
621
+ "epoch": 46.93333333333333,
622
+ "grad_norm": 2.008852005004883,
623
+ "learning_rate": 8.266666666666667e-05,
624
+ "loss": 3.7016,
625
+ "step": 176
626
+ },
627
+ {
628
+ "epoch": 47.46666666666667,
629
+ "grad_norm": 1.6691290140151978,
630
+ "learning_rate": 8.133333333333334e-05,
631
+ "loss": 3.6785,
632
+ "step": 178
633
+ },
634
+ {
635
+ "epoch": 48.0,
636
+ "grad_norm": 1.4747165441513062,
637
+ "learning_rate": 8e-05,
638
+ "loss": 3.668,
639
+ "step": 180
640
+ },
641
+ {
642
+ "epoch": 48.53333333333333,
643
+ "grad_norm": 1.3272978067398071,
644
+ "learning_rate": 7.866666666666666e-05,
645
+ "loss": 3.7995,
646
+ "step": 182
647
+ },
648
+ {
649
+ "epoch": 49.06666666666667,
650
+ "grad_norm": 1.1977958679199219,
651
+ "learning_rate": 7.733333333333333e-05,
652
+ "loss": 3.5753,
653
+ "step": 184
654
+ },
655
+ {
656
+ "epoch": 49.6,
657
+ "grad_norm": 1.4208600521087646,
658
+ "learning_rate": 7.6e-05,
659
+ "loss": 3.6183,
660
+ "step": 186
661
+ },
662
+ {
663
+ "epoch": 50.13333333333333,
664
+ "grad_norm": 1.2266496419906616,
665
+ "learning_rate": 7.466666666666667e-05,
666
+ "loss": 3.7525,
667
+ "step": 188
668
+ },
669
+ {
670
+ "epoch": 50.666666666666664,
671
+ "grad_norm": 1.1905524730682373,
672
+ "learning_rate": 7.333333333333333e-05,
673
+ "loss": 3.5681,
674
+ "step": 190
675
+ },
676
+ {
677
+ "epoch": 51.2,
678
+ "grad_norm": 1.1554759740829468,
679
+ "learning_rate": 7.2e-05,
680
+ "loss": 3.5788,
681
+ "step": 192
682
+ },
683
+ {
684
+ "epoch": 51.733333333333334,
685
+ "grad_norm": 1.1419849395751953,
686
+ "learning_rate": 7.066666666666667e-05,
687
+ "loss": 3.7141,
688
+ "step": 194
689
+ },
690
+ {
691
+ "epoch": 52.266666666666666,
692
+ "grad_norm": 1.2043012380599976,
693
+ "learning_rate": 6.933333333333334e-05,
694
+ "loss": 3.6421,
695
+ "step": 196
696
+ },
697
+ {
698
+ "epoch": 52.8,
699
+ "grad_norm": 1.0960720777511597,
700
+ "learning_rate": 6.800000000000001e-05,
701
+ "loss": 3.6484,
702
+ "step": 198
703
+ },
704
+ {
705
+ "epoch": 53.333333333333336,
706
+ "grad_norm": 1.1179126501083374,
707
+ "learning_rate": 6.666666666666667e-05,
708
+ "loss": 3.4974,
709
+ "step": 200
710
+ },
711
+ {
712
+ "epoch": 53.86666666666667,
713
+ "grad_norm": 1.3456348180770874,
714
+ "learning_rate": 6.533333333333334e-05,
715
+ "loss": 3.7002,
716
+ "step": 202
717
+ },
718
+ {
719
+ "epoch": 54.4,
720
+ "grad_norm": 1.1445540189743042,
721
+ "learning_rate": 6.400000000000001e-05,
722
+ "loss": 3.5583,
723
+ "step": 204
724
+ },
725
+ {
726
+ "epoch": 54.93333333333333,
727
+ "grad_norm": 1.247029423713684,
728
+ "learning_rate": 6.266666666666667e-05,
729
+ "loss": 3.6645,
730
+ "step": 206
731
+ },
732
+ {
733
+ "epoch": 55.46666666666667,
734
+ "grad_norm": 1.210913062095642,
735
+ "learning_rate": 6.133333333333334e-05,
736
+ "loss": 3.5581,
737
+ "step": 208
738
+ },
739
+ {
740
+ "epoch": 56.0,
741
+ "grad_norm": 1.1455459594726562,
742
+ "learning_rate": 6e-05,
743
+ "loss": 3.679,
744
+ "step": 210
745
+ },
746
+ {
747
+ "epoch": 56.53333333333333,
748
+ "grad_norm": 1.1794884204864502,
749
+ "learning_rate": 5.866666666666667e-05,
750
+ "loss": 3.6196,
751
+ "step": 212
752
+ },
753
+ {
754
+ "epoch": 57.06666666666667,
755
+ "grad_norm": 1.2104743719100952,
756
+ "learning_rate": 5.7333333333333336e-05,
757
+ "loss": 3.566,
758
+ "step": 214
759
+ },
760
+ {
761
+ "epoch": 57.6,
762
+ "grad_norm": 1.3371809720993042,
763
+ "learning_rate": 5.6000000000000006e-05,
764
+ "loss": 3.7143,
765
+ "step": 216
766
+ },
767
+ {
768
+ "epoch": 58.13333333333333,
769
+ "grad_norm": 1.430527925491333,
770
+ "learning_rate": 5.466666666666666e-05,
771
+ "loss": 3.4442,
772
+ "step": 218
773
+ },
774
+ {
775
+ "epoch": 58.666666666666664,
776
+ "grad_norm": 1.1653505563735962,
777
+ "learning_rate": 5.333333333333333e-05,
778
+ "loss": 3.5261,
779
+ "step": 220
780
+ },
781
+ {
782
+ "epoch": 59.2,
783
+ "grad_norm": 1.17708158493042,
784
+ "learning_rate": 5.2000000000000004e-05,
785
+ "loss": 3.6556,
786
+ "step": 222
787
+ },
788
+ {
789
+ "epoch": 59.733333333333334,
790
+ "grad_norm": 1.0106998682022095,
791
+ "learning_rate": 5.0666666666666674e-05,
792
+ "loss": 3.7191,
793
+ "step": 224
794
+ },
795
+ {
796
+ "epoch": 60.266666666666666,
797
+ "grad_norm": 1.1047751903533936,
798
+ "learning_rate": 4.933333333333334e-05,
799
+ "loss": 3.4595,
800
+ "step": 226
801
+ },
802
+ {
803
+ "epoch": 60.8,
804
+ "grad_norm": 1.0759046077728271,
805
+ "learning_rate": 4.8e-05,
806
+ "loss": 3.5661,
807
+ "step": 228
808
+ },
809
+ {
810
+ "epoch": 61.333333333333336,
811
+ "grad_norm": 1.1332894563674927,
812
+ "learning_rate": 4.666666666666667e-05,
813
+ "loss": 3.5103,
814
+ "step": 230
815
+ },
816
+ {
817
+ "epoch": 61.86666666666667,
818
+ "grad_norm": 1.0236306190490723,
819
+ "learning_rate": 4.5333333333333335e-05,
820
+ "loss": 3.5468,
821
+ "step": 232
822
+ },
823
+ {
824
+ "epoch": 62.4,
825
+ "grad_norm": 0.97084641456604,
826
+ "learning_rate": 4.4000000000000006e-05,
827
+ "loss": 3.6628,
828
+ "step": 234
829
+ },
830
+ {
831
+ "epoch": 62.93333333333333,
832
+ "grad_norm": 1.025275707244873,
833
+ "learning_rate": 4.266666666666667e-05,
834
+ "loss": 3.4807,
835
+ "step": 236
836
+ },
837
+ {
838
+ "epoch": 63.46666666666667,
839
+ "grad_norm": 1.0704567432403564,
840
+ "learning_rate": 4.133333333333333e-05,
841
+ "loss": 3.6431,
842
+ "step": 238
843
+ },
844
+ {
845
+ "epoch": 64.0,
846
+ "grad_norm": 1.0402157306671143,
847
+ "learning_rate": 4e-05,
848
+ "loss": 3.4682,
849
+ "step": 240
850
+ },
851
+ {
852
+ "epoch": 64.53333333333333,
853
+ "grad_norm": 0.9403516054153442,
854
+ "learning_rate": 3.866666666666667e-05,
855
+ "loss": 3.5516,
856
+ "step": 242
857
+ },
858
+ {
859
+ "epoch": 65.06666666666666,
860
+ "grad_norm": 1.120253086090088,
861
+ "learning_rate": 3.733333333333334e-05,
862
+ "loss": 3.4593,
863
+ "step": 244
864
+ },
865
+ {
866
+ "epoch": 65.6,
867
+ "grad_norm": 1.1206554174423218,
868
+ "learning_rate": 3.6e-05,
869
+ "loss": 3.4822,
870
+ "step": 246
871
+ },
872
+ {
873
+ "epoch": 66.13333333333334,
874
+ "grad_norm": 1.181155800819397,
875
+ "learning_rate": 3.466666666666667e-05,
876
+ "loss": 3.6441,
877
+ "step": 248
878
+ },
879
+ {
880
+ "epoch": 66.66666666666667,
881
+ "grad_norm": 1.0042577981948853,
882
+ "learning_rate": 3.3333333333333335e-05,
883
+ "loss": 3.437,
884
+ "step": 250
885
+ },
886
+ {
887
+ "epoch": 67.2,
888
+ "grad_norm": 1.0655186176300049,
889
+ "learning_rate": 3.2000000000000005e-05,
890
+ "loss": 3.6071,
891
+ "step": 252
892
+ },
893
+ {
894
+ "epoch": 67.73333333333333,
895
+ "grad_norm": 1.0911762714385986,
896
+ "learning_rate": 3.066666666666667e-05,
897
+ "loss": 3.6135,
898
+ "step": 254
899
+ },
900
+ {
901
+ "epoch": 68.26666666666667,
902
+ "grad_norm": 0.947772204875946,
903
+ "learning_rate": 2.9333333333333336e-05,
904
+ "loss": 3.4867,
905
+ "step": 256
906
+ },
907
+ {
908
+ "epoch": 68.8,
909
+ "grad_norm": 1.002395749092102,
910
+ "learning_rate": 2.8000000000000003e-05,
911
+ "loss": 3.501,
912
+ "step": 258
913
+ },
914
+ {
915
+ "epoch": 69.33333333333333,
916
+ "grad_norm": 1.0234873294830322,
917
+ "learning_rate": 2.6666666666666667e-05,
918
+ "loss": 3.4908,
919
+ "step": 260
920
+ },
921
+ {
922
+ "epoch": 69.86666666666666,
923
+ "grad_norm": 1.1004796028137207,
924
+ "learning_rate": 2.5333333333333337e-05,
925
+ "loss": 3.6095,
926
+ "step": 262
927
+ },
928
+ {
929
+ "epoch": 70.4,
930
+ "grad_norm": 1.1492009162902832,
931
+ "learning_rate": 2.4e-05,
932
+ "loss": 3.4027,
933
+ "step": 264
934
+ },
935
+ {
936
+ "epoch": 70.93333333333334,
937
+ "grad_norm": 0.9846398830413818,
938
+ "learning_rate": 2.2666666666666668e-05,
939
+ "loss": 3.5456,
940
+ "step": 266
941
+ },
942
+ {
943
+ "epoch": 71.46666666666667,
944
+ "grad_norm": 1.1093006134033203,
945
+ "learning_rate": 2.1333333333333335e-05,
946
+ "loss": 3.457,
947
+ "step": 268
948
+ },
949
+ {
950
+ "epoch": 72.0,
951
+ "grad_norm": 0.9900301694869995,
952
+ "learning_rate": 2e-05,
953
+ "loss": 3.5602,
954
+ "step": 270
955
+ },
956
+ {
957
+ "epoch": 72.53333333333333,
958
+ "grad_norm": 0.9894844889640808,
959
+ "learning_rate": 1.866666666666667e-05,
960
+ "loss": 3.4681,
961
+ "step": 272
962
+ },
963
+ {
964
+ "epoch": 73.06666666666666,
965
+ "grad_norm": 1.0027010440826416,
966
+ "learning_rate": 1.7333333333333336e-05,
967
+ "loss": 3.5727,
968
+ "step": 274
969
+ },
970
+ {
971
+ "epoch": 73.6,
972
+ "grad_norm": 0.9067515730857849,
973
+ "learning_rate": 1.6000000000000003e-05,
974
+ "loss": 3.4379,
975
+ "step": 276
976
+ },
977
+ {
978
+ "epoch": 74.13333333333334,
979
+ "grad_norm": 0.9846177697181702,
980
+ "learning_rate": 1.4666666666666668e-05,
981
+ "loss": 3.49,
982
+ "step": 278
983
+ },
984
+ {
985
+ "epoch": 74.66666666666667,
986
+ "grad_norm": 0.9469838738441467,
987
+ "learning_rate": 1.3333333333333333e-05,
988
+ "loss": 3.5143,
989
+ "step": 280
990
+ },
991
+ {
992
+ "epoch": 75.2,
993
+ "grad_norm": 0.9074487686157227,
994
+ "learning_rate": 1.2e-05,
995
+ "loss": 3.4596,
996
+ "step": 282
997
+ },
998
+ {
999
+ "epoch": 75.73333333333333,
1000
+ "grad_norm": 0.8845421671867371,
1001
+ "learning_rate": 1.0666666666666667e-05,
1002
+ "loss": 3.4648,
1003
+ "step": 284
1004
+ },
1005
+ {
1006
+ "epoch": 76.26666666666667,
1007
+ "grad_norm": 0.9070376753807068,
1008
+ "learning_rate": 9.333333333333334e-06,
1009
+ "loss": 3.6296,
1010
+ "step": 286
1011
+ },
1012
+ {
1013
+ "epoch": 76.8,
1014
+ "grad_norm": 0.9411069750785828,
1015
+ "learning_rate": 8.000000000000001e-06,
1016
+ "loss": 3.4424,
1017
+ "step": 288
1018
+ },
1019
+ {
1020
+ "epoch": 77.33333333333333,
1021
+ "grad_norm": 0.8896852135658264,
1022
+ "learning_rate": 6.666666666666667e-06,
1023
+ "loss": 3.5589,
1024
+ "step": 290
1025
+ },
1026
+ {
1027
+ "epoch": 77.86666666666666,
1028
+ "grad_norm": 0.8958707451820374,
1029
+ "learning_rate": 5.333333333333334e-06,
1030
+ "loss": 3.3978,
1031
+ "step": 292
1032
+ },
1033
+ {
1034
+ "epoch": 78.4,
1035
+ "grad_norm": 0.8889992237091064,
1036
+ "learning_rate": 4.000000000000001e-06,
1037
+ "loss": 3.4738,
1038
+ "step": 294
1039
+ },
1040
+ {
1041
+ "epoch": 78.93333333333334,
1042
+ "grad_norm": 0.9154471755027771,
1043
+ "learning_rate": 2.666666666666667e-06,
1044
+ "loss": 3.5366,
1045
+ "step": 296
1046
+ },
1047
+ {
1048
+ "epoch": 79.46666666666667,
1049
+ "grad_norm": 0.8812463879585266,
1050
+ "learning_rate": 1.3333333333333334e-06,
1051
+ "loss": 3.4683,
1052
+ "step": 298
1053
+ },
1054
+ {
1055
+ "epoch": 80.0,
1056
+ "grad_norm": 0.8584585785865784,
1057
+ "learning_rate": 0.0,
1058
+ "loss": 3.5199,
1059
+ "step": 300
1060
+ },
1061
+ {
1062
+ "epoch": 80.0,
1063
+ "step": 300,
1064
+ "total_flos": 715681220927544.0,
1065
+ "train_loss": 3.9132511806488037,
1066
+ "train_runtime": 7847.3305,
1067
+ "train_samples_per_second": 0.765,
1068
+ "train_steps_per_second": 0.038
1069
+ }
1070
+ ],
1071
+ "logging_steps": 2,
1072
+ "max_steps": 300,
1073
+ "num_input_tokens_seen": 0,
1074
+ "num_train_epochs": 100,
1075
+ "save_steps": 500,
1076
+ "stateful_callbacks": {
1077
+ "TrainerControl": {
1078
+ "args": {
1079
+ "should_epoch_stop": false,
1080
+ "should_evaluate": false,
1081
+ "should_log": false,
1082
+ "should_save": false,
1083
+ "should_training_stop": false
1084
+ },
1085
+ "attributes": {}
1086
+ }
1087
+ },
1088
+ "total_flos": 715681220927544.0,
1089
+ "train_batch_size": 2,
1090
+ "trial_name": null,
1091
+ "trial_params": null
1092
+ }