chansung commited on
Commit
43cf298
1 Parent(s): 8b3b565

Model save

Browse files
README.md CHANGED
@@ -2,13 +2,12 @@
2
  license: gemma
3
  library_name: peft
4
  tags:
5
- - alignment-handbook
6
  - trl
7
  - sft
8
  - generated_from_trainer
9
  base_model: google/gemma-2b
10
  datasets:
11
- - llama-duo/synth_summarize_dataset_dedup
12
  model-index:
13
  - name: gemma2b-summarize-gemini1_5flash-64k
14
  results: []
@@ -19,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  # gemma2b-summarize-gemini1_5flash-64k
21
 
22
- This model is a fine-tuned version of [google/gemma-2b](https://huggingface.co/google/gemma-2b) on the llama-duo/synth_summarize_dataset_dedup dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 2.7177
25
 
26
  ## Model description
27
 
@@ -41,14 +40,14 @@ More information needed
41
 
42
  The following hyperparameters were used during training:
43
  - learning_rate: 0.0002
44
- - train_batch_size: 16
45
- - eval_batch_size: 16
46
  - seed: 42
47
  - distributed_type: multi-GPU
48
  - num_devices: 8
49
  - gradient_accumulation_steps: 2
50
- - total_train_batch_size: 256
51
- - total_eval_batch_size: 128
52
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
  - lr_scheduler_type: cosine
54
  - lr_scheduler_warmup_ratio: 0.1
@@ -56,21 +55,18 @@ The following hyperparameters were used during training:
56
 
57
  ### Training results
58
 
59
- | Training Loss | Epoch | Step | Validation Loss |
60
- |:-------------:|:-------:|:----:|:---------------:|
61
- | 1.518 | 0.9905 | 52 | 2.7709 |
62
- | 1.1423 | 2.0 | 105 | 2.6595 |
63
- | 1.0681 | 2.9905 | 157 | 2.6406 |
64
- | 1.0335 | 4.0 | 210 | 2.6427 |
65
- | 1.0079 | 4.9905 | 262 | 2.6459 |
66
- | 0.9837 | 6.0 | 315 | 2.6574 |
67
- | 0.966 | 6.9905 | 367 | 2.6700 |
68
- | 0.9474 | 8.0 | 420 | 2.6799 |
69
- | 0.9406 | 8.9905 | 472 | 2.6883 |
70
- | 0.9245 | 10.0 | 525 | 2.6975 |
71
- | 0.9208 | 10.9905 | 577 | 2.7079 |
72
- | 0.9195 | 12.0 | 630 | 2.7148 |
73
- | 0.9212 | 12.9905 | 682 | 2.7154 |
74
 
75
 
76
  ### Framework versions
 
2
  license: gemma
3
  library_name: peft
4
  tags:
 
5
  - trl
6
  - sft
7
  - generated_from_trainer
8
  base_model: google/gemma-2b
9
  datasets:
10
+ - generator
11
  model-index:
12
  - name: gemma2b-summarize-gemini1_5flash-64k
13
  results: []
 
18
 
19
  # gemma2b-summarize-gemini1_5flash-64k
20
 
21
+ This model is a fine-tuned version of [google/gemma-2b](https://huggingface.co/google/gemma-2b) on the generator dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 2.5156
24
 
25
  ## Model description
26
 
 
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 0.0002
43
+ - train_batch_size: 8
44
+ - eval_batch_size: 8
45
  - seed: 42
46
  - distributed_type: multi-GPU
47
  - num_devices: 8
48
  - gradient_accumulation_steps: 2
49
+ - total_train_batch_size: 128
50
+ - total_eval_batch_size: 64
51
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
  - lr_scheduler_type: cosine
53
  - lr_scheduler_warmup_ratio: 0.1
 
55
 
56
  ### Training results
57
 
58
+ | Training Loss | Epoch | Step | Validation Loss |
59
+ |:-------------:|:------:|:----:|:---------------:|
60
+ | 1.3387 | 0.9905 | 52 | 2.5711 |
61
+ | 1.0869 | 2.0 | 105 | 2.4883 |
62
+ | 1.0363 | 2.9905 | 157 | 2.4850 |
63
+ | 0.9967 | 4.0 | 210 | 2.4809 |
64
+ | 0.98 | 4.9905 | 262 | 2.4904 |
65
+ | 0.9574 | 6.0 | 315 | 2.4995 |
66
+ | 0.9392 | 6.9905 | 367 | 2.5062 |
67
+ | 0.9415 | 8.0 | 420 | 2.5143 |
68
+ | 0.9367 | 8.9905 | 472 | 2.5155 |
69
+ | 0.9348 | 9.9048 | 520 | 2.5156 |
 
 
 
70
 
71
 
72
  ### Framework versions
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:657ae35d44261dc89a4505f5d6be177b857fa49056fd5b2851b51f46fb1d3134
3
  size 39256960
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4711620e971ab843c174762480b72259ca41cd9ba5e1123d1b7c569ba006191
3
  size 39256960
all_results.json CHANGED
@@ -1,14 +1,9 @@
1
  {
2
- "epoch": 13.333333333333334,
3
- "eval_loss": 2.7176907062530518,
4
- "eval_runtime": 0.8578,
5
- "eval_samples": 25,
6
- "eval_samples_per_second": 24.482,
7
- "eval_steps_per_second": 1.166,
8
- "total_flos": 1.1018283868225536e+18,
9
- "train_loss": 0.0,
10
- "train_runtime": 3.5395,
11
  "train_samples": 63353,
12
- "train_samples_per_second": 37654.826,
13
- "train_steps_per_second": 146.913
14
  }
 
1
  {
2
+ "epoch": 9.904761904761905,
3
+ "total_flos": 8.18501087353897e+17,
4
+ "train_loss": 1.098491345002101,
5
+ "train_runtime": 2780.3493,
 
 
 
 
 
6
  "train_samples": 63353,
7
+ "train_samples_per_second": 23.972,
8
+ "train_steps_per_second": 0.187
9
  }
runs/Jun10_05-12-11_48ddfe8e991f/events.out.tfevents.1717996367.48ddfe8e991f.142677.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:eb58893d2412081b9b73761b8e7ae1ca1047670dbf636a8bc4cc76ffe6f25b50
3
- size 29240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed27c036d85716b87752449c452c5de6cd3beb6bedfcc4b976658c6ff3aea995
3
+ size 30709
train_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "epoch": 13.333333333333334,
3
- "total_flos": 1.1018283868225536e+18,
4
- "train_loss": 0.0,
5
- "train_runtime": 3.5395,
6
  "train_samples": 63353,
7
- "train_samples_per_second": 37654.826,
8
- "train_steps_per_second": 146.913
9
  }
 
1
  {
2
+ "epoch": 9.904761904761905,
3
+ "total_flos": 8.18501087353897e+17,
4
+ "train_loss": 1.098491345002101,
5
+ "train_runtime": 2780.3493,
6
  "train_samples": 63353,
7
+ "train_samples_per_second": 23.972,
8
+ "train_steps_per_second": 0.187
9
  }
trainer_state.json CHANGED
@@ -1,1112 +1,836 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 13.333333333333334,
5
  "eval_steps": 500,
6
- "global_step": 700,
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.01904761904761905,
13
- "grad_norm": 3.15625,
14
- "learning_rate": 2.564102564102564e-06,
15
- "loss": 3.3205,
16
  "step": 1
17
  },
18
  {
19
  "epoch": 0.09523809523809523,
20
- "grad_norm": 4.34375,
21
- "learning_rate": 1.282051282051282e-05,
22
- "loss": 3.3126,
23
  "step": 5
24
  },
25
  {
26
  "epoch": 0.19047619047619047,
27
- "grad_norm": 2.609375,
28
- "learning_rate": 2.564102564102564e-05,
29
- "loss": 3.2719,
30
  "step": 10
31
  },
32
  {
33
  "epoch": 0.2857142857142857,
34
- "grad_norm": 2.828125,
35
- "learning_rate": 3.846153846153846e-05,
36
- "loss": 3.1205,
37
  "step": 15
38
  },
39
  {
40
  "epoch": 0.38095238095238093,
41
- "grad_norm": 2.03125,
42
- "learning_rate": 5.128205128205128e-05,
43
- "loss": 2.8114,
44
  "step": 20
45
  },
46
  {
47
  "epoch": 0.47619047619047616,
48
- "grad_norm": 2.125,
49
- "learning_rate": 6.410256410256412e-05,
50
- "loss": 2.5384,
51
  "step": 25
52
  },
53
  {
54
  "epoch": 0.5714285714285714,
55
- "grad_norm": 6.03125,
56
- "learning_rate": 7.692307692307693e-05,
57
- "loss": 2.3201,
58
  "step": 30
59
  },
60
  {
61
  "epoch": 0.6666666666666666,
62
- "grad_norm": 2.0625,
63
- "learning_rate": 8.974358974358975e-05,
64
- "loss": 2.1053,
65
  "step": 35
66
  },
67
  {
68
  "epoch": 0.7619047619047619,
69
- "grad_norm": 1.359375,
70
- "learning_rate": 0.00010256410256410256,
71
- "loss": 1.8717,
72
  "step": 40
73
  },
74
  {
75
  "epoch": 0.8571428571428571,
76
- "grad_norm": 0.54296875,
77
- "learning_rate": 0.00011538461538461538,
78
- "loss": 1.6592,
79
  "step": 45
80
  },
81
  {
82
  "epoch": 0.9523809523809523,
83
- "grad_norm": 0.427734375,
84
- "learning_rate": 0.00012820512820512823,
85
- "loss": 1.518,
86
  "step": 50
87
  },
88
  {
89
  "epoch": 0.9904761904761905,
90
- "eval_loss": 2.7708818912506104,
91
- "eval_runtime": 0.4987,
92
- "eval_samples_per_second": 42.112,
93
- "eval_steps_per_second": 2.005,
94
  "step": 52
95
  },
96
  {
97
  "epoch": 1.0476190476190477,
98
- "grad_norm": 0.396484375,
99
- "learning_rate": 0.00014102564102564104,
100
- "loss": 1.426,
101
  "step": 55
102
  },
103
  {
104
  "epoch": 1.1428571428571428,
105
- "grad_norm": 0.51953125,
106
- "learning_rate": 0.00015384615384615385,
107
- "loss": 1.3513,
108
  "step": 60
109
  },
110
  {
111
  "epoch": 1.2380952380952381,
112
- "grad_norm": 0.75390625,
113
- "learning_rate": 0.0001666666666666667,
114
- "loss": 1.3036,
115
  "step": 65
116
  },
117
  {
118
  "epoch": 1.3333333333333333,
119
- "grad_norm": 0.703125,
120
- "learning_rate": 0.0001794871794871795,
121
- "loss": 1.2627,
122
  "step": 70
123
  },
124
  {
125
  "epoch": 1.4285714285714286,
126
- "grad_norm": 0.6015625,
127
- "learning_rate": 0.00019230769230769233,
128
- "loss": 1.2335,
129
  "step": 75
130
  },
131
  {
132
  "epoch": 1.5238095238095237,
133
- "grad_norm": 0.31640625,
134
- "learning_rate": 0.00019999599453798524,
135
- "loss": 1.2094,
136
  "step": 80
137
  },
138
  {
139
  "epoch": 1.619047619047619,
140
- "grad_norm": 0.359375,
141
- "learning_rate": 0.0001999509367752813,
142
- "loss": 1.1957,
143
  "step": 85
144
  },
145
  {
146
  "epoch": 1.7142857142857144,
147
- "grad_norm": 0.80859375,
148
- "learning_rate": 0.00019985583705641418,
149
- "loss": 1.1772,
150
  "step": 90
151
  },
152
  {
153
  "epoch": 1.8095238095238095,
154
- "grad_norm": 0.5,
155
- "learning_rate": 0.0001997107429945041,
156
- "loss": 1.1595,
157
  "step": 95
158
  },
159
  {
160
  "epoch": 1.9047619047619047,
161
- "grad_norm": 0.48046875,
162
- "learning_rate": 0.0001995157272330992,
163
- "loss": 1.1451,
164
  "step": 100
165
  },
166
  {
167
  "epoch": 2.0,
168
- "grad_norm": 0.408203125,
169
- "learning_rate": 0.0001992708874098054,
170
- "loss": 1.1423,
171
  "step": 105
172
  },
173
  {
174
  "epoch": 2.0,
175
- "eval_loss": 2.659493923187256,
176
- "eval_runtime": 0.4877,
177
- "eval_samples_per_second": 43.055,
178
- "eval_steps_per_second": 2.05,
179
  "step": 105
180
  },
181
  {
182
  "epoch": 2.0952380952380953,
183
- "grad_norm": 0.419921875,
184
- "learning_rate": 0.00019897634610740287,
185
- "loss": 1.1131,
186
  "step": 110
187
  },
188
  {
189
  "epoch": 2.1904761904761907,
190
- "grad_norm": 0.50390625,
191
- "learning_rate": 0.00019863225079247285,
192
- "loss": 1.1121,
193
  "step": 115
194
  },
195
  {
196
  "epoch": 2.2857142857142856,
197
- "grad_norm": 0.333984375,
198
- "learning_rate": 0.00019823877374156647,
199
- "loss": 1.103,
200
  "step": 120
201
  },
202
  {
203
  "epoch": 2.380952380952381,
204
- "grad_norm": 0.5859375,
205
- "learning_rate": 0.00019779611195495177,
206
- "loss": 1.1023,
207
  "step": 125
208
  },
209
  {
210
  "epoch": 2.4761904761904763,
211
- "grad_norm": 0.95703125,
212
- "learning_rate": 0.00019730448705798239,
213
- "loss": 1.086,
214
  "step": 130
215
  },
216
  {
217
  "epoch": 2.571428571428571,
218
- "grad_norm": 0.392578125,
219
- "learning_rate": 0.00019676414519013781,
220
- "loss": 1.0886,
221
  "step": 135
222
  },
223
  {
224
  "epoch": 2.6666666666666665,
225
- "grad_norm": 0.70703125,
226
- "learning_rate": 0.0001961753568817896,
227
- "loss": 1.0818,
228
  "step": 140
229
  },
230
  {
231
  "epoch": 2.761904761904762,
232
- "grad_norm": 0.49609375,
233
- "learning_rate": 0.0001955384169187563,
234
- "loss": 1.0799,
235
  "step": 145
236
  },
237
  {
238
  "epoch": 2.857142857142857,
239
- "grad_norm": 0.71484375,
240
- "learning_rate": 0.00019485364419471454,
241
- "loss": 1.0792,
242
  "step": 150
243
  },
244
  {
245
  "epoch": 2.9523809523809526,
246
- "grad_norm": 0.267578125,
247
- "learning_rate": 0.00019412138155154002,
248
- "loss": 1.0681,
249
  "step": 155
250
  },
251
  {
252
  "epoch": 2.9904761904761905,
253
- "eval_loss": 2.6406373977661133,
254
- "eval_runtime": 0.4936,
255
- "eval_samples_per_second": 42.54,
256
- "eval_steps_per_second": 2.026,
257
  "step": 157
258
  },
259
  {
260
  "epoch": 3.0476190476190474,
261
- "grad_norm": 0.291015625,
262
- "learning_rate": 0.0001933419956076584,
263
- "loss": 1.0607,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 3.142857142857143,
268
- "grad_norm": 0.384765625,
269
- "learning_rate": 0.00019251587657449236,
270
- "loss": 1.0467,
271
  "step": 165
272
  },
273
  {
274
  "epoch": 3.238095238095238,
275
- "grad_norm": 0.2578125,
276
- "learning_rate": 0.00019164343806109632,
277
- "loss": 1.0429,
278
  "step": 170
279
  },
280
  {
281
  "epoch": 3.3333333333333335,
282
- "grad_norm": 0.28125,
283
- "learning_rate": 0.00019072511686707663,
284
- "loss": 1.0477,
285
  "step": 175
286
  },
287
  {
288
  "epoch": 3.4285714285714284,
289
- "grad_norm": 0.23046875,
290
- "learning_rate": 0.0001897613727639014,
291
- "loss": 1.0407,
292
  "step": 180
293
  },
294
  {
295
  "epoch": 3.5238095238095237,
296
- "grad_norm": 0.390625,
297
- "learning_rate": 0.00018875268826470872,
298
- "loss": 1.0408,
299
  "step": 185
300
  },
301
  {
302
  "epoch": 3.619047619047619,
303
- "grad_norm": 0.3046875,
304
- "learning_rate": 0.00018769956838272936,
305
- "loss": 1.0392,
306
  "step": 190
307
  },
308
  {
309
  "epoch": 3.7142857142857144,
310
- "grad_norm": 0.3203125,
311
- "learning_rate": 0.00018660254037844388,
312
- "loss": 1.0393,
313
  "step": 195
314
  },
315
  {
316
  "epoch": 3.8095238095238093,
317
- "grad_norm": 0.32421875,
318
- "learning_rate": 0.00018546215349560203,
319
- "loss": 1.0385,
320
  "step": 200
321
  },
322
  {
323
  "epoch": 3.9047619047619047,
324
- "grad_norm": 0.828125,
325
- "learning_rate": 0.00018427897868623534,
326
- "loss": 1.0383,
327
  "step": 205
328
  },
329
  {
330
  "epoch": 4.0,
331
- "grad_norm": 0.4375,
332
- "learning_rate": 0.00018305360832480117,
333
- "loss": 1.0335,
334
  "step": 210
335
  },
336
  {
337
  "epoch": 4.0,
338
- "eval_loss": 2.642651319503784,
339
- "eval_runtime": 0.4862,
340
- "eval_samples_per_second": 43.193,
341
- "eval_steps_per_second": 2.057,
342
  "step": 210
343
  },
344
  {
345
  "epoch": 4.095238095238095,
346
- "grad_norm": 0.291015625,
347
- "learning_rate": 0.00018178665591160172,
348
- "loss": 1.0188,
349
  "step": 215
350
  },
351
  {
352
  "epoch": 4.190476190476191,
353
- "grad_norm": 0.27734375,
354
- "learning_rate": 0.00018047875576562557,
355
- "loss": 1.0134,
356
  "step": 220
357
  },
358
  {
359
  "epoch": 4.285714285714286,
360
- "grad_norm": 0.255859375,
361
- "learning_rate": 0.0001791305627069662,
362
- "loss": 1.0109,
363
  "step": 225
364
  },
365
  {
366
  "epoch": 4.380952380952381,
367
- "grad_norm": 0.265625,
368
- "learning_rate": 0.0001777427517289766,
369
- "loss": 1.009,
370
  "step": 230
371
  },
372
  {
373
  "epoch": 4.476190476190476,
374
- "grad_norm": 0.2890625,
375
- "learning_rate": 0.00017631601766032336,
376
- "loss": 1.0116,
377
  "step": 235
378
  },
379
  {
380
  "epoch": 4.571428571428571,
381
- "grad_norm": 0.5,
382
- "learning_rate": 0.00017485107481711012,
383
- "loss": 1.0067,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 4.666666666666667,
388
- "grad_norm": 0.58203125,
389
- "learning_rate": 0.0001733486566452446,
390
- "loss": 1.0112,
391
  "step": 245
392
  },
393
  {
394
  "epoch": 4.761904761904762,
395
- "grad_norm": 0.40234375,
396
- "learning_rate": 0.0001718095153532274,
397
- "loss": 1.0096,
398
  "step": 250
399
  },
400
  {
401
  "epoch": 4.857142857142857,
402
- "grad_norm": 0.67578125,
403
- "learning_rate": 0.00017023442153554777,
404
- "loss": 1.0046,
405
  "step": 255
406
  },
407
  {
408
  "epoch": 4.9523809523809526,
409
- "grad_norm": 0.76953125,
410
- "learning_rate": 0.0001686241637868734,
411
- "loss": 1.0079,
412
  "step": 260
413
  },
414
  {
415
  "epoch": 4.9904761904761905,
416
- "eval_loss": 2.645949363708496,
417
- "eval_runtime": 0.4992,
418
- "eval_samples_per_second": 42.069,
419
- "eval_steps_per_second": 2.003,
420
  "step": 262
421
  },
422
  {
423
  "epoch": 5.0476190476190474,
424
- "grad_norm": 0.310546875,
425
- "learning_rate": 0.00016697954830722868,
426
- "loss": 0.9948,
427
  "step": 265
428
  },
429
  {
430
  "epoch": 5.142857142857143,
431
- "grad_norm": 0.34375,
432
- "learning_rate": 0.0001653013984983585,
433
- "loss": 0.9868,
434
  "step": 270
435
  },
436
  {
437
  "epoch": 5.238095238095238,
438
- "grad_norm": 0.5625,
439
- "learning_rate": 0.0001635905545514795,
440
- "loss": 0.9845,
441
  "step": 275
442
  },
443
  {
444
  "epoch": 5.333333333333333,
445
- "grad_norm": 0.5859375,
446
- "learning_rate": 0.0001618478730266255,
447
- "loss": 0.9812,
448
  "step": 280
449
  },
450
  {
451
  "epoch": 5.428571428571429,
452
- "grad_norm": 0.65625,
453
- "learning_rate": 0.0001600742264237979,
454
- "loss": 0.9887,
455
  "step": 285
456
  },
457
  {
458
  "epoch": 5.523809523809524,
459
- "grad_norm": 0.357421875,
460
- "learning_rate": 0.00015827050274613513,
461
- "loss": 0.9863,
462
  "step": 290
463
  },
464
  {
465
  "epoch": 5.619047619047619,
466
- "grad_norm": 0.30078125,
467
- "learning_rate": 0.0001564376050553205,
468
- "loss": 0.9862,
469
  "step": 295
470
  },
471
  {
472
  "epoch": 5.714285714285714,
473
- "grad_norm": 0.275390625,
474
- "learning_rate": 0.00015457645101945046,
475
- "loss": 0.9865,
476
  "step": 300
477
  },
478
  {
479
  "epoch": 5.809523809523809,
480
- "grad_norm": 0.337890625,
481
- "learning_rate": 0.00015268797245359035,
482
- "loss": 0.9929,
483
  "step": 305
484
  },
485
  {
486
  "epoch": 5.904761904761905,
487
- "grad_norm": 0.294921875,
488
- "learning_rate": 0.0001507731148532468,
489
- "loss": 0.9868,
490
  "step": 310
491
  },
492
  {
493
  "epoch": 6.0,
494
- "grad_norm": 0.32421875,
495
- "learning_rate": 0.00014883283692099112,
496
- "loss": 0.9837,
497
  "step": 315
498
  },
499
  {
500
  "epoch": 6.0,
501
- "eval_loss": 2.657383441925049,
502
- "eval_runtime": 0.4893,
503
- "eval_samples_per_second": 42.915,
504
- "eval_steps_per_second": 2.044,
505
  "step": 315
506
  },
507
  {
508
  "epoch": 6.095238095238095,
509
- "grad_norm": 0.294921875,
510
- "learning_rate": 0.00014686811008647038,
511
- "loss": 0.9669,
512
  "step": 320
513
  },
514
  {
515
  "epoch": 6.190476190476191,
516
- "grad_norm": 0.287109375,
517
- "learning_rate": 0.00014487991802004623,
518
- "loss": 0.9674,
519
  "step": 325
520
  },
521
  {
522
  "epoch": 6.285714285714286,
523
- "grad_norm": 0.3046875,
524
- "learning_rate": 0.00014286925614030542,
525
- "loss": 0.9712,
526
  "step": 330
527
  },
528
  {
529
  "epoch": 6.380952380952381,
530
- "grad_norm": 0.28515625,
531
- "learning_rate": 0.00014083713111568842,
532
- "loss": 0.9663,
533
  "step": 335
534
  },
535
  {
536
  "epoch": 6.476190476190476,
537
- "grad_norm": 0.462890625,
538
- "learning_rate": 0.0001387845603604855,
539
- "loss": 0.9674,
540
  "step": 340
541
  },
542
  {
543
  "epoch": 6.571428571428571,
544
- "grad_norm": 0.54296875,
545
- "learning_rate": 0.00013671257152545277,
546
- "loss": 0.9732,
547
  "step": 345
548
  },
549
  {
550
  "epoch": 6.666666666666667,
551
- "grad_norm": 0.490234375,
552
- "learning_rate": 0.00013462220198330328,
553
- "loss": 0.9743,
554
  "step": 350
555
  },
556
  {
557
  "epoch": 6.761904761904762,
558
- "grad_norm": 0.296875,
559
- "learning_rate": 0.0001325144983093305,
560
- "loss": 0.9704,
561
  "step": 355
562
  },
563
  {
564
  "epoch": 6.857142857142857,
565
- "grad_norm": 0.326171875,
566
- "learning_rate": 0.0001303905157574247,
567
- "loss": 0.9617,
568
  "step": 360
569
  },
570
  {
571
  "epoch": 6.9523809523809526,
572
- "grad_norm": 0.3984375,
573
- "learning_rate": 0.0001282513177317437,
574
- "loss": 0.966,
575
  "step": 365
576
  },
577
  {
578
  "epoch": 6.9904761904761905,
579
- "eval_loss": 2.6700327396392822,
580
- "eval_runtime": 0.5365,
581
- "eval_samples_per_second": 39.146,
582
- "eval_steps_per_second": 1.864,
583
  "step": 367
584
  },
585
  {
586
  "epoch": 7.0476190476190474,
587
- "grad_norm": 0.294921875,
588
- "learning_rate": 0.00012609797525430373,
589
- "loss": 0.9592,
590
  "step": 370
591
  },
592
  {
593
  "epoch": 7.142857142857143,
594
- "grad_norm": 0.3203125,
595
- "learning_rate": 0.0001239315664287558,
596
- "loss": 0.9535,
597
  "step": 375
598
  },
599
  {
600
  "epoch": 7.238095238095238,
601
- "grad_norm": 0.26953125,
602
- "learning_rate": 0.00012175317590061674,
603
- "loss": 0.9528,
604
  "step": 380
605
  },
606
  {
607
  "epoch": 7.333333333333333,
608
- "grad_norm": 0.416015625,
609
- "learning_rate": 0.00011956389431422507,
610
- "loss": 0.9572,
611
  "step": 385
612
  },
613
  {
614
  "epoch": 7.428571428571429,
615
- "grad_norm": 0.341796875,
616
- "learning_rate": 0.00011736481776669306,
617
- "loss": 0.9461,
618
  "step": 390
619
  },
620
  {
621
  "epoch": 7.523809523809524,
622
- "grad_norm": 0.30078125,
623
- "learning_rate": 0.00011515704725912926,
624
- "loss": 0.9513,
625
  "step": 395
626
  },
627
  {
628
  "epoch": 7.619047619047619,
629
- "grad_norm": 0.3125,
630
- "learning_rate": 0.00011294168814540553,
631
- "loss": 0.9566,
632
  "step": 400
633
  },
634
  {
635
  "epoch": 7.714285714285714,
636
- "grad_norm": 0.609375,
637
- "learning_rate": 0.00011071984957874479,
638
- "loss": 0.9594,
639
  "step": 405
640
  },
641
  {
642
  "epoch": 7.809523809523809,
643
- "grad_norm": 0.296875,
644
- "learning_rate": 0.00010849264395640649,
645
- "loss": 0.9554,
646
  "step": 410
647
  },
648
  {
649
  "epoch": 7.904761904761905,
650
- "grad_norm": 0.33203125,
651
- "learning_rate": 0.0001062611863627482,
652
- "loss": 0.9479,
653
  "step": 415
654
  },
655
  {
656
  "epoch": 8.0,
657
- "grad_norm": 0.353515625,
658
- "learning_rate": 0.00010402659401094152,
659
- "loss": 0.9474,
660
  "step": 420
661
  },
662
  {
663
  "epoch": 8.0,
664
- "eval_loss": 2.6798880100250244,
665
- "eval_runtime": 0.4848,
666
- "eval_samples_per_second": 43.314,
667
- "eval_steps_per_second": 2.063,
668
  "step": 420
669
  },
670
  {
671
  "epoch": 8.095238095238095,
672
- "grad_norm": 0.404296875,
673
- "learning_rate": 0.00010178998568362243,
674
- "loss": 0.9432,
675
  "step": 425
676
  },
677
  {
678
  "epoch": 8.19047619047619,
679
- "grad_norm": 0.26953125,
680
- "learning_rate": 9.955248117275566e-05,
681
- "loss": 0.9398,
682
  "step": 430
683
  },
684
  {
685
  "epoch": 8.285714285714286,
686
- "grad_norm": 0.326171875,
687
- "learning_rate": 9.73152007189939e-05,
688
- "loss": 0.9348,
689
  "step": 435
690
  },
691
  {
692
  "epoch": 8.380952380952381,
693
- "grad_norm": 0.30078125,
694
- "learning_rate": 9.507926445081219e-05,
695
- "loss": 0.9416,
696
  "step": 440
697
  },
698
  {
699
  "epoch": 8.476190476190476,
700
- "grad_norm": 0.28515625,
701
- "learning_rate": 9.284579182369867e-05,
702
- "loss": 0.9352,
703
  "step": 445
704
  },
705
  {
706
  "epoch": 8.571428571428571,
707
- "grad_norm": 0.283203125,
708
- "learning_rate": 9.061590105968208e-05,
709
- "loss": 0.9427,
710
  "step": 450
711
  },
712
  {
713
  "epoch": 8.666666666666666,
714
- "grad_norm": 0.283203125,
715
- "learning_rate": 8.839070858747697e-05,
716
- "loss": 0.9413,
717
  "step": 455
718
  },
719
  {
720
  "epoch": 8.761904761904763,
721
- "grad_norm": 0.294921875,
722
- "learning_rate": 8.617132848352671e-05,
723
- "loss": 0.9409,
724
  "step": 460
725
  },
726
  {
727
  "epoch": 8.857142857142858,
728
- "grad_norm": 0.2890625,
729
- "learning_rate": 8.395887191422397e-05,
730
- "loss": 0.942,
731
  "step": 465
732
  },
733
  {
734
  "epoch": 8.952380952380953,
735
- "grad_norm": 0.298828125,
736
- "learning_rate": 8.175444657958876e-05,
737
- "loss": 0.9406,
738
  "step": 470
739
  },
740
  {
741
  "epoch": 8.99047619047619,
742
- "eval_loss": 2.688331365585327,
743
- "eval_runtime": 0.5482,
744
- "eval_samples_per_second": 38.309,
745
- "eval_steps_per_second": 1.824,
746
  "step": 472
747
  },
748
  {
749
  "epoch": 9.047619047619047,
750
- "grad_norm": 0.302734375,
751
- "learning_rate": 7.955915615868111e-05,
752
- "loss": 0.9375,
753
  "step": 475
754
  },
755
  {
756
  "epoch": 9.142857142857142,
757
- "grad_norm": 0.275390625,
758
- "learning_rate": 7.73740997570278e-05,
759
- "loss": 0.9322,
760
  "step": 480
761
  },
762
  {
763
  "epoch": 9.238095238095237,
764
- "grad_norm": 0.26171875,
765
- "learning_rate": 7.520037135633816e-05,
766
- "loss": 0.9292,
767
  "step": 485
768
  },
769
  {
770
  "epoch": 9.333333333333334,
771
- "grad_norm": 0.287109375,
772
- "learning_rate": 7.303905926678564e-05,
773
- "loss": 0.9269,
774
  "step": 490
775
  },
776
  {
777
  "epoch": 9.428571428571429,
778
- "grad_norm": 0.341796875,
779
- "learning_rate": 7.089124558212871e-05,
780
- "loss": 0.9331,
781
  "step": 495
782
  },
783
  {
784
  "epoch": 9.523809523809524,
785
- "grad_norm": 0.28125,
786
- "learning_rate": 6.875800563794425e-05,
787
- "loss": 0.9344,
788
  "step": 500
789
  },
790
  {
791
  "epoch": 9.619047619047619,
792
- "grad_norm": 0.28515625,
793
- "learning_rate": 6.664040747324437e-05,
794
- "loss": 0.9294,
795
  "step": 505
796
  },
797
  {
798
  "epoch": 9.714285714285714,
799
- "grad_norm": 0.267578125,
800
- "learning_rate": 6.453951129574644e-05,
801
- "loss": 0.9347,
802
  "step": 510
803
  },
804
  {
805
  "epoch": 9.80952380952381,
806
- "grad_norm": 0.30078125,
807
- "learning_rate": 6.245636895106402e-05,
808
- "loss": 0.9282,
809
  "step": 515
810
  },
811
  {
812
  "epoch": 9.904761904761905,
813
- "grad_norm": 0.353515625,
814
- "learning_rate": 6.039202339608432e-05,
815
- "loss": 0.9299,
816
  "step": 520
817
  },
818
  {
819
- "epoch": 10.0,
820
- "grad_norm": 0.27734375,
821
- "learning_rate": 5.834750817679606e-05,
822
- "loss": 0.9245,
823
- "step": 525
824
- },
825
- {
826
- "epoch": 10.0,
827
- "eval_loss": 2.6975326538085938,
828
- "eval_runtime": 0.485,
829
- "eval_samples_per_second": 43.303,
830
- "eval_steps_per_second": 2.062,
831
- "step": 525
832
- },
833
- {
834
- "epoch": 10.095238095238095,
835
- "grad_norm": 0.30078125,
836
- "learning_rate": 5.6323846910828735e-05,
837
- "loss": 0.9233,
838
- "step": 530
839
- },
840
- {
841
- "epoch": 10.19047619047619,
842
- "grad_norm": 0.265625,
843
- "learning_rate": 5.432205277496327e-05,
844
- "loss": 0.9235,
845
- "step": 535
846
- },
847
- {
848
- "epoch": 10.285714285714286,
849
- "grad_norm": 0.2734375,
850
- "learning_rate": 5.234312799786921e-05,
851
- "loss": 0.9194,
852
- "step": 540
853
- },
854
- {
855
- "epoch": 10.380952380952381,
856
- "grad_norm": 0.29296875,
857
- "learning_rate": 5.0388063358324134e-05,
858
- "loss": 0.9235,
859
- "step": 545
860
- },
861
- {
862
- "epoch": 10.476190476190476,
863
- "grad_norm": 0.279296875,
864
- "learning_rate": 4.845783768916482e-05,
865
- "loss": 0.9217,
866
- "step": 550
867
- },
868
- {
869
- "epoch": 10.571428571428571,
870
- "grad_norm": 0.265625,
871
- "learning_rate": 4.6553417387219886e-05,
872
- "loss": 0.9243,
873
- "step": 555
874
- },
875
- {
876
- "epoch": 10.666666666666666,
877
- "grad_norm": 0.306640625,
878
- "learning_rate": 4.467575592946864e-05,
879
- "loss": 0.9262,
880
- "step": 560
881
- },
882
- {
883
- "epoch": 10.761904761904763,
884
- "grad_norm": 0.255859375,
885
- "learning_rate": 4.282579339566802e-05,
886
- "loss": 0.9206,
887
- "step": 565
888
- },
889
- {
890
- "epoch": 10.857142857142858,
891
- "grad_norm": 0.2578125,
892
- "learning_rate": 4.100445599768774e-05,
893
- "loss": 0.9256,
894
- "step": 570
895
- },
896
- {
897
- "epoch": 10.952380952380953,
898
- "grad_norm": 0.265625,
899
- "learning_rate": 3.9212655615787804e-05,
900
- "loss": 0.9208,
901
- "step": 575
902
- },
903
- {
904
- "epoch": 10.99047619047619,
905
- "eval_loss": 2.707897663116455,
906
- "eval_runtime": 0.647,
907
- "eval_samples_per_second": 32.456,
908
- "eval_steps_per_second": 1.546,
909
- "step": 577
910
- },
911
- {
912
- "epoch": 11.047619047619047,
913
- "grad_norm": 0.25,
914
- "learning_rate": 3.745128934207225e-05,
915
- "loss": 0.9226,
916
- "step": 580
917
- },
918
- {
919
- "epoch": 11.142857142857142,
920
- "grad_norm": 0.263671875,
921
- "learning_rate": 3.5721239031346066e-05,
922
- "loss": 0.9121,
923
- "step": 585
924
- },
925
- {
926
- "epoch": 11.238095238095237,
927
- "grad_norm": 0.259765625,
928
- "learning_rate": 3.402337085960119e-05,
929
- "loss": 0.9155,
930
- "step": 590
931
- },
932
- {
933
- "epoch": 11.333333333333334,
934
- "grad_norm": 0.2578125,
935
- "learning_rate": 3.235853489035241e-05,
936
- "loss": 0.9194,
937
- "step": 595
938
- },
939
- {
940
- "epoch": 11.428571428571429,
941
- "grad_norm": 0.26953125,
942
- "learning_rate": 3.072756464904006e-05,
943
- "loss": 0.9202,
944
- "step": 600
945
- },
946
- {
947
- "epoch": 11.523809523809524,
948
- "grad_norm": 0.27734375,
949
- "learning_rate": 2.9131276705713006e-05,
950
- "loss": 0.9186,
951
- "step": 605
952
- },
953
- {
954
- "epoch": 11.619047619047619,
955
- "grad_norm": 0.259765625,
956
- "learning_rate": 2.7570470266200176e-05,
957
- "loss": 0.9195,
958
- "step": 610
959
- },
960
- {
961
- "epoch": 11.714285714285714,
962
- "grad_norm": 0.2470703125,
963
- "learning_rate": 2.6045926771976303e-05,
964
- "loss": 0.9223,
965
- "step": 615
966
- },
967
- {
968
- "epoch": 11.80952380952381,
969
- "grad_norm": 0.25,
970
- "learning_rate": 2.4558409508920986e-05,
971
- "loss": 0.9139,
972
- "step": 620
973
- },
974
- {
975
- "epoch": 11.904761904761905,
976
- "grad_norm": 0.248046875,
977
- "learning_rate": 2.3108663225168435e-05,
978
- "loss": 0.9117,
979
- "step": 625
980
- },
981
- {
982
- "epoch": 12.0,
983
- "grad_norm": 0.263671875,
984
- "learning_rate": 2.1697413758237784e-05,
985
- "loss": 0.9195,
986
- "step": 630
987
- },
988
- {
989
- "epoch": 12.0,
990
- "eval_loss": 2.714789867401123,
991
- "eval_runtime": 0.4906,
992
- "eval_samples_per_second": 42.808,
993
- "eval_steps_per_second": 2.038,
994
- "step": 630
995
- },
996
- {
997
- "epoch": 12.095238095238095,
998
- "grad_norm": 0.2392578125,
999
- "learning_rate": 2.032536767163141e-05,
1000
- "loss": 0.9167,
1001
- "step": 635
1002
- },
1003
- {
1004
- "epoch": 12.19047619047619,
1005
- "grad_norm": 0.25,
1006
- "learning_rate": 1.8993211901083353e-05,
1007
- "loss": 0.9144,
1008
- "step": 640
1009
- },
1010
- {
1011
- "epoch": 12.285714285714286,
1012
- "grad_norm": 0.2578125,
1013
- "learning_rate": 1.7701613410634365e-05,
1014
- "loss": 0.915,
1015
- "step": 645
1016
- },
1017
- {
1018
- "epoch": 12.380952380952381,
1019
- "grad_norm": 0.248046875,
1020
- "learning_rate": 1.6451218858706374e-05,
1021
- "loss": 0.9153,
1022
- "step": 650
1023
- },
1024
- {
1025
- "epoch": 12.476190476190476,
1026
- "grad_norm": 0.251953125,
1027
- "learning_rate": 1.5242654274342894e-05,
1028
- "loss": 0.9133,
1029
- "step": 655
1030
- },
1031
- {
1032
- "epoch": 12.571428571428571,
1033
- "grad_norm": 0.25,
1034
- "learning_rate": 1.4076524743778319e-05,
1035
- "loss": 0.9134,
1036
- "step": 660
1037
- },
1038
- {
1039
- "epoch": 12.666666666666666,
1040
- "grad_norm": 0.248046875,
1041
- "learning_rate": 1.295341410749208e-05,
1042
- "loss": 0.9126,
1043
- "step": 665
1044
- },
1045
- {
1046
- "epoch": 12.761904761904763,
1047
- "grad_norm": 0.25,
1048
- "learning_rate": 1.1873884667900125e-05,
1049
- "loss": 0.9094,
1050
- "step": 670
1051
- },
1052
- {
1053
- "epoch": 12.857142857142858,
1054
- "grad_norm": 0.236328125,
1055
- "learning_rate": 1.083847690782972e-05,
1056
- "loss": 0.9183,
1057
- "step": 675
1058
- },
1059
- {
1060
- "epoch": 12.952380952380953,
1061
- "grad_norm": 0.255859375,
1062
- "learning_rate": 9.8477092199184e-06,
1063
- "loss": 0.9212,
1064
- "step": 680
1065
- },
1066
- {
1067
- "epoch": 12.99047619047619,
1068
- "eval_loss": 2.7153878211975098,
1069
- "eval_runtime": 0.5024,
1070
- "eval_samples_per_second": 41.796,
1071
- "eval_steps_per_second": 1.99,
1072
- "step": 682
1073
- },
1074
- {
1075
- "epoch": 13.047619047619047,
1076
- "grad_norm": 0.2431640625,
1077
- "learning_rate": 8.902077647072881e-06,
1078
- "loss": 0.9139,
1079
- "step": 685
1080
- },
1081
- {
1082
- "epoch": 13.142857142857142,
1083
- "grad_norm": 0.2431640625,
1084
- "learning_rate": 8.002055634117578e-06,
1085
- "loss": 0.9123,
1086
- "step": 690
1087
- },
1088
- {
1089
- "epoch": 13.238095238095237,
1090
- "grad_norm": 0.2470703125,
1091
- "learning_rate": 7.148093790757371e-06,
1092
- "loss": 0.9152,
1093
- "step": 695
1094
  },
1095
  {
1096
- "epoch": 13.333333333333334,
1097
- "grad_norm": 0.2578125,
1098
- "learning_rate": 6.3406196659728465e-06,
1099
- "loss": 0.9087,
1100
- "step": 700
1101
- },
1102
- {
1103
- "epoch": 13.333333333333334,
1104
- "step": 700,
1105
- "total_flos": 1.1018283868225536e+18,
1106
- "train_loss": 0.0,
1107
- "train_runtime": 3.5395,
1108
- "train_samples_per_second": 37654.826,
1109
- "train_steps_per_second": 146.913
1110
  }
1111
  ],
1112
  "logging_steps": 5,
@@ -1126,8 +850,8 @@
1126
  "attributes": {}
1127
  }
1128
  },
1129
- "total_flos": 1.1018283868225536e+18,
1130
- "train_batch_size": 16,
1131
  "trial_name": null,
1132
  "trial_params": null
1133
  }
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 9.904761904761905,
5
  "eval_steps": 500,
6
+ "global_step": 520,
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.01904761904761905,
13
+ "grad_norm": 3.328125,
14
+ "learning_rate": 3.846153846153847e-06,
15
+ "loss": 3.0068,
16
  "step": 1
17
  },
18
  {
19
  "epoch": 0.09523809523809523,
20
+ "grad_norm": 3.171875,
21
+ "learning_rate": 1.923076923076923e-05,
22
+ "loss": 3.0102,
23
  "step": 5
24
  },
25
  {
26
  "epoch": 0.19047619047619047,
27
+ "grad_norm": 2.375,
28
+ "learning_rate": 3.846153846153846e-05,
29
+ "loss": 2.9279,
30
  "step": 10
31
  },
32
  {
33
  "epoch": 0.2857142857142857,
34
+ "grad_norm": 1.90625,
35
+ "learning_rate": 5.769230769230769e-05,
36
+ "loss": 2.6349,
37
  "step": 15
38
  },
39
  {
40
  "epoch": 0.38095238095238093,
41
+ "grad_norm": 1.703125,
42
+ "learning_rate": 7.692307692307693e-05,
43
+ "loss": 2.3632,
44
  "step": 20
45
  },
46
  {
47
  "epoch": 0.47619047619047616,
48
+ "grad_norm": 1.625,
49
+ "learning_rate": 9.615384615384617e-05,
50
+ "loss": 2.1672,
51
  "step": 25
52
  },
53
  {
54
  "epoch": 0.5714285714285714,
55
+ "grad_norm": 1.171875,
56
+ "learning_rate": 0.00011538461538461538,
57
+ "loss": 1.9589,
58
  "step": 30
59
  },
60
  {
61
  "epoch": 0.6666666666666666,
62
+ "grad_norm": 0.69921875,
63
+ "learning_rate": 0.00013461538461538464,
64
+ "loss": 1.7472,
65
  "step": 35
66
  },
67
  {
68
  "epoch": 0.7619047619047619,
69
+ "grad_norm": 0.84765625,
70
+ "learning_rate": 0.00015384615384615385,
71
+ "loss": 1.5556,
72
  "step": 40
73
  },
74
  {
75
  "epoch": 0.8571428571428571,
76
+ "grad_norm": 0.35546875,
77
+ "learning_rate": 0.0001730769230769231,
78
+ "loss": 1.4254,
79
  "step": 45
80
  },
81
  {
82
  "epoch": 0.9523809523809523,
83
+ "grad_norm": 1.3203125,
84
+ "learning_rate": 0.00019230769230769233,
85
+ "loss": 1.3387,
86
  "step": 50
87
  },
88
  {
89
  "epoch": 0.9904761904761905,
90
+ "eval_loss": 2.571077346801758,
91
+ "eval_runtime": 0.5007,
92
+ "eval_samples_per_second": 19.97,
93
+ "eval_steps_per_second": 1.997,
94
  "step": 52
95
  },
96
  {
97
  "epoch": 1.0476190476190477,
98
+ "grad_norm": 1.546875,
99
+ "learning_rate": 0.00019997972289848503,
100
+ "loss": 1.279,
101
  "step": 55
102
  },
103
  {
104
  "epoch": 1.1428571428571428,
105
+ "grad_norm": 0.466796875,
106
+ "learning_rate": 0.00019985583705641418,
107
+ "loss": 1.2299,
108
  "step": 60
109
  },
110
  {
111
  "epoch": 1.2380952380952381,
112
+ "grad_norm": 0.1943359375,
113
+ "learning_rate": 0.00019961946980917456,
114
+ "loss": 1.1885,
115
  "step": 65
116
  },
117
  {
118
  "epoch": 1.3333333333333333,
119
+ "grad_norm": 0.404296875,
120
+ "learning_rate": 0.0001992708874098054,
121
+ "loss": 1.1735,
122
  "step": 70
123
  },
124
  {
125
  "epoch": 1.4285714285714286,
126
+ "grad_norm": 0.462890625,
127
+ "learning_rate": 0.0001988104825147528,
128
+ "loss": 1.1575,
129
  "step": 75
130
  },
131
  {
132
  "epoch": 1.5238095238095237,
133
+ "grad_norm": 0.443359375,
134
+ "learning_rate": 0.00019823877374156647,
135
+ "loss": 1.1404,
136
  "step": 80
137
  },
138
  {
139
  "epoch": 1.619047619047619,
140
+ "grad_norm": 0.306640625,
141
+ "learning_rate": 0.00019755640508470942,
142
+ "loss": 1.1303,
143
  "step": 85
144
  },
145
  {
146
  "epoch": 1.7142857142857144,
147
+ "grad_norm": 0.21875,
148
+ "learning_rate": 0.00019676414519013781,
149
+ "loss": 1.1185,
150
  "step": 90
151
  },
152
  {
153
  "epoch": 1.8095238095238095,
154
+ "grad_norm": 0.22265625,
155
+ "learning_rate": 0.00019586288648946947,
156
+ "loss": 1.1006,
157
  "step": 95
158
  },
159
  {
160
  "epoch": 1.9047619047619047,
161
+ "grad_norm": 0.30859375,
162
+ "learning_rate": 0.00019485364419471454,
163
+ "loss": 1.1027,
164
  "step": 100
165
  },
166
  {
167
  "epoch": 2.0,
168
+ "grad_norm": 0.2236328125,
169
+ "learning_rate": 0.00019373755515470254,
170
+ "loss": 1.0869,
171
  "step": 105
172
  },
173
  {
174
  "epoch": 2.0,
175
+ "eval_loss": 2.4883005619049072,
176
+ "eval_runtime": 0.4918,
177
+ "eval_samples_per_second": 20.333,
178
+ "eval_steps_per_second": 2.033,
179
  "step": 105
180
  },
181
  {
182
  "epoch": 2.0952380952380953,
183
+ "grad_norm": 0.271484375,
184
+ "learning_rate": 0.00019251587657449236,
185
+ "loss": 1.0706,
186
  "step": 110
187
  },
188
  {
189
  "epoch": 2.1904761904761907,
190
+ "grad_norm": 0.373046875,
191
+ "learning_rate": 0.00019118998459920902,
192
+ "loss": 1.0609,
193
  "step": 115
194
  },
195
  {
196
  "epoch": 2.2857142857142856,
197
+ "grad_norm": 0.68359375,
198
+ "learning_rate": 0.0001897613727639014,
199
+ "loss": 1.057,
200
  "step": 120
201
  },
202
  {
203
  "epoch": 2.380952380952381,
204
+ "grad_norm": 0.50390625,
205
+ "learning_rate": 0.0001882316503111678,
206
+ "loss": 1.0538,
207
  "step": 125
208
  },
209
  {
210
  "epoch": 2.4761904761904763,
211
+ "grad_norm": 0.3046875,
212
+ "learning_rate": 0.00018660254037844388,
213
+ "loss": 1.0531,
214
  "step": 130
215
  },
216
  {
217
  "epoch": 2.571428571428571,
218
+ "grad_norm": 0.5703125,
219
+ "learning_rate": 0.00018487587805699526,
220
+ "loss": 1.0403,
221
  "step": 135
222
  },
223
  {
224
  "epoch": 2.6666666666666665,
225
+ "grad_norm": 0.203125,
226
+ "learning_rate": 0.00018305360832480117,
227
+ "loss": 1.0392,
228
  "step": 140
229
  },
230
  {
231
  "epoch": 2.761904761904762,
232
+ "grad_norm": 0.3359375,
233
+ "learning_rate": 0.00018113778385565733,
234
+ "loss": 1.044,
235
  "step": 145
236
  },
237
  {
238
  "epoch": 2.857142857142857,
239
+ "grad_norm": 0.25390625,
240
+ "learning_rate": 0.0001791305627069662,
241
+ "loss": 1.0329,
242
  "step": 150
243
  },
244
  {
245
  "epoch": 2.9523809523809526,
246
+ "grad_norm": 0.279296875,
247
+ "learning_rate": 0.00017703420588881946,
248
+ "loss": 1.0363,
249
  "step": 155
250
  },
251
  {
252
  "epoch": 2.9904761904761905,
253
+ "eval_loss": 2.4850006103515625,
254
+ "eval_runtime": 0.5452,
255
+ "eval_samples_per_second": 18.343,
256
+ "eval_steps_per_second": 1.834,
257
  "step": 157
258
  },
259
  {
260
  "epoch": 3.0476190476190474,
261
+ "grad_norm": 0.244140625,
262
+ "learning_rate": 0.00017485107481711012,
263
+ "loss": 1.024,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 3.142857142857143,
268
+ "grad_norm": 0.37109375,
269
+ "learning_rate": 0.00017258362865354426,
270
+ "loss": 1.0143,
271
  "step": 165
272
  },
273
  {
274
  "epoch": 3.238095238095238,
275
+ "grad_norm": 0.34375,
276
+ "learning_rate": 0.00017023442153554777,
277
+ "loss": 1.0159,
278
  "step": 170
279
  },
280
  {
281
  "epoch": 3.3333333333333335,
282
+ "grad_norm": 0.259765625,
283
+ "learning_rate": 0.0001678060996991891,
284
+ "loss": 1.01,
285
  "step": 175
286
  },
287
  {
288
  "epoch": 3.4285714285714284,
289
+ "grad_norm": 0.314453125,
290
+ "learning_rate": 0.0001653013984983585,
291
+ "loss": 1.0035,
292
  "step": 180
293
  },
294
  {
295
  "epoch": 3.5238095238095237,
296
+ "grad_norm": 1.265625,
297
+ "learning_rate": 0.00016272313932356162,
298
+ "loss": 1.0042,
299
  "step": 185
300
  },
301
  {
302
  "epoch": 3.619047619047619,
303
+ "grad_norm": 0.81640625,
304
+ "learning_rate": 0.0001600742264237979,
305
+ "loss": 1.0087,
306
  "step": 190
307
  },
308
  {
309
  "epoch": 3.7142857142857144,
310
+ "grad_norm": 0.384765625,
311
+ "learning_rate": 0.0001573576436351046,
312
+ "loss": 1.0057,
313
  "step": 195
314
  },
315
  {
316
  "epoch": 3.8095238095238093,
317
+ "grad_norm": 0.283203125,
318
+ "learning_rate": 0.00015457645101945046,
319
+ "loss": 0.9958,
320
  "step": 200
321
  },
322
  {
323
  "epoch": 3.9047619047619047,
324
+ "grad_norm": 0.2890625,
325
+ "learning_rate": 0.00015173378141776568,
326
+ "loss": 1.0002,
327
  "step": 205
328
  },
329
  {
330
  "epoch": 4.0,
331
+ "grad_norm": 0.298828125,
332
+ "learning_rate": 0.00014883283692099112,
333
+ "loss": 0.9967,
334
  "step": 210
335
  },
336
  {
337
  "epoch": 4.0,
338
+ "eval_loss": 2.480916738510132,
339
+ "eval_runtime": 0.4922,
340
+ "eval_samples_per_second": 20.317,
341
+ "eval_steps_per_second": 2.032,
342
  "step": 210
343
  },
344
  {
345
  "epoch": 4.095238095238095,
346
+ "grad_norm": 0.244140625,
347
+ "learning_rate": 0.00014587688526312143,
348
+ "loss": 0.9743,
349
  "step": 215
350
  },
351
  {
352
  "epoch": 4.190476190476191,
353
+ "grad_norm": 0.3359375,
354
+ "learning_rate": 0.00014286925614030542,
355
+ "loss": 0.9753,
356
  "step": 220
357
  },
358
  {
359
  "epoch": 4.285714285714286,
360
+ "grad_norm": 0.25,
361
+ "learning_rate": 0.0001398133374601501,
362
+ "loss": 0.9836,
363
  "step": 225
364
  },
365
  {
366
  "epoch": 4.380952380952381,
367
+ "grad_norm": 0.267578125,
368
+ "learning_rate": 0.00013671257152545277,
369
+ "loss": 0.9794,
370
  "step": 230
371
  },
372
  {
373
  "epoch": 4.476190476190476,
374
+ "grad_norm": 0.25,
375
+ "learning_rate": 0.0001335704511566605,
376
+ "loss": 0.9753,
377
  "step": 235
378
  },
379
  {
380
  "epoch": 4.571428571428571,
381
+ "grad_norm": 0.26953125,
382
+ "learning_rate": 0.0001303905157574247,
383
+ "loss": 0.9812,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 4.666666666666667,
388
+ "grad_norm": 0.25390625,
389
+ "learning_rate": 0.00012717634732768243,
390
+ "loss": 0.9818,
391
  "step": 245
392
  },
393
  {
394
  "epoch": 4.761904761904762,
395
+ "grad_norm": 0.302734375,
396
+ "learning_rate": 0.0001239315664287558,
397
+ "loss": 0.9834,
398
  "step": 250
399
  },
400
  {
401
  "epoch": 4.857142857142857,
402
+ "grad_norm": 0.302734375,
403
+ "learning_rate": 0.00012065982810501404,
404
+ "loss": 0.9737,
405
  "step": 255
406
  },
407
  {
408
  "epoch": 4.9523809523809526,
409
+ "grad_norm": 0.267578125,
410
+ "learning_rate": 0.00011736481776669306,
411
+ "loss": 0.98,
412
  "step": 260
413
  },
414
  {
415
  "epoch": 4.9904761904761905,
416
+ "eval_loss": 2.490432024002075,
417
+ "eval_runtime": 0.7012,
418
+ "eval_samples_per_second": 14.262,
419
+ "eval_steps_per_second": 1.426,
420
  "step": 262
421
  },
422
  {
423
  "epoch": 5.0476190476190474,
424
+ "grad_norm": 0.546875,
425
+ "learning_rate": 0.00011405024703850929,
426
+ "loss": 0.9722,
427
  "step": 265
428
  },
429
  {
430
  "epoch": 5.142857142857143,
431
+ "grad_norm": 0.314453125,
432
+ "learning_rate": 0.00011071984957874479,
433
+ "loss": 0.9661,
434
  "step": 270
435
  },
436
  {
437
  "epoch": 5.238095238095238,
438
+ "grad_norm": 0.296875,
439
+ "learning_rate": 0.00010737737687351284,
440
+ "loss": 0.9631,
441
  "step": 275
442
  },
443
  {
444
  "epoch": 5.333333333333333,
445
+ "grad_norm": 0.2890625,
446
+ "learning_rate": 0.00010402659401094152,
447
+ "loss": 0.9568,
448
  "step": 280
449
  },
450
  {
451
  "epoch": 5.428571428571429,
452
+ "grad_norm": 0.3359375,
453
+ "learning_rate": 0.00010067127544003563,
454
+ "loss": 0.9619,
455
  "step": 285
456
  },
457
  {
458
  "epoch": 5.523809523809524,
459
+ "grad_norm": 0.255859375,
460
+ "learning_rate": 9.73152007189939e-05,
461
+ "loss": 0.9597,
462
  "step": 290
463
  },
464
  {
465
  "epoch": 5.619047619047619,
466
+ "grad_norm": 0.3984375,
467
+ "learning_rate": 9.396215025777139e-05,
468
+ "loss": 0.9619,
469
  "step": 295
470
  },
471
  {
472
  "epoch": 5.714285714285714,
473
+ "grad_norm": 0.3515625,
474
+ "learning_rate": 9.061590105968208e-05,
475
+ "loss": 0.9614,
476
  "step": 300
477
  },
478
  {
479
  "epoch": 5.809523809523809,
480
+ "grad_norm": 0.435546875,
481
+ "learning_rate": 8.728022246683894e-05,
482
+ "loss": 0.9536,
483
  "step": 305
484
  },
485
  {
486
  "epoch": 5.904761904761905,
487
+ "grad_norm": 0.3359375,
488
+ "learning_rate": 8.395887191422397e-05,
489
+ "loss": 0.9584,
490
  "step": 310
491
  },
492
  {
493
  "epoch": 6.0,
494
+ "grad_norm": 0.2265625,
495
+ "learning_rate": 8.065559069717088e-05,
496
+ "loss": 0.9574,
497
  "step": 315
498
  },
499
  {
500
  "epoch": 6.0,
501
+ "eval_loss": 2.499537706375122,
502
+ "eval_runtime": 0.4857,
503
+ "eval_samples_per_second": 20.591,
504
+ "eval_steps_per_second": 2.059,
505
  "step": 315
506
  },
507
  {
508
  "epoch": 6.095238095238095,
509
+ "grad_norm": 0.24609375,
510
+ "learning_rate": 7.73740997570278e-05,
511
+ "loss": 0.9494,
512
  "step": 320
513
  },
514
  {
515
  "epoch": 6.190476190476191,
516
+ "grad_norm": 0.2490234375,
517
+ "learning_rate": 7.411809548974792e-05,
518
+ "loss": 0.9472,
519
  "step": 325
520
  },
521
  {
522
  "epoch": 6.285714285714286,
523
+ "grad_norm": 0.263671875,
524
+ "learning_rate": 7.089124558212871e-05,
525
+ "loss": 0.9521,
526
  "step": 330
527
  },
528
  {
529
  "epoch": 6.380952380952381,
530
+ "grad_norm": 0.275390625,
531
+ "learning_rate": 6.769718488039023e-05,
532
+ "loss": 0.9466,
533
  "step": 335
534
  },
535
  {
536
  "epoch": 6.476190476190476,
537
+ "grad_norm": 0.240234375,
538
+ "learning_rate": 6.453951129574644e-05,
539
+ "loss": 0.9436,
540
  "step": 340
541
  },
542
  {
543
  "epoch": 6.571428571428571,
544
+ "grad_norm": 0.240234375,
545
+ "learning_rate": 6.142178175158149e-05,
546
+ "loss": 0.9501,
547
  "step": 345
548
  },
549
  {
550
  "epoch": 6.666666666666667,
551
+ "grad_norm": 0.259765625,
552
+ "learning_rate": 5.834750817679606e-05,
553
+ "loss": 0.9484,
554
  "step": 350
555
  },
556
  {
557
  "epoch": 6.761904761904762,
558
+ "grad_norm": 0.234375,
559
+ "learning_rate": 5.5320153549837415e-05,
560
+ "loss": 0.9489,
561
  "step": 355
562
  },
563
  {
564
  "epoch": 6.857142857142857,
565
+ "grad_norm": 0.2236328125,
566
+ "learning_rate": 5.234312799786921e-05,
567
+ "loss": 0.9486,
568
  "step": 360
569
  },
570
  {
571
  "epoch": 6.9523809523809526,
572
+ "grad_norm": 0.2333984375,
573
+ "learning_rate": 4.9419784955474524e-05,
574
+ "loss": 0.9392,
575
  "step": 365
576
  },
577
  {
578
  "epoch": 6.9904761904761905,
579
+ "eval_loss": 2.5062003135681152,
580
+ "eval_runtime": 0.5559,
581
+ "eval_samples_per_second": 17.989,
582
+ "eval_steps_per_second": 1.799,
583
  "step": 367
584
  },
585
  {
586
  "epoch": 7.0476190476190474,
587
+ "grad_norm": 0.2158203125,
588
+ "learning_rate": 4.6553417387219886e-05,
589
+ "loss": 0.9358,
590
  "step": 370
591
  },
592
  {
593
  "epoch": 7.142857142857143,
594
+ "grad_norm": 0.24609375,
595
+ "learning_rate": 4.374725407833532e-05,
596
+ "loss": 0.9313,
597
  "step": 375
598
  },
599
  {
600
  "epoch": 7.238095238095238,
601
+ "grad_norm": 0.2578125,
602
+ "learning_rate": 4.100445599768774e-05,
603
+ "loss": 0.9343,
604
  "step": 380
605
  },
606
  {
607
  "epoch": 7.333333333333333,
608
+ "grad_norm": 0.2236328125,
609
+ "learning_rate": 3.832811273714569e-05,
610
+ "loss": 0.9384,
611
  "step": 385
612
  },
613
  {
614
  "epoch": 7.428571428571429,
615
+ "grad_norm": 0.2490234375,
616
+ "learning_rate": 3.5721239031346066e-05,
617
+ "loss": 0.9387,
618
  "step": 390
619
  },
620
  {
621
  "epoch": 7.523809523809524,
622
+ "grad_norm": 0.2265625,
623
+ "learning_rate": 3.318677136178228e-05,
624
+ "loss": 0.9356,
625
  "step": 395
626
  },
627
  {
628
  "epoch": 7.619047619047619,
629
+ "grad_norm": 0.232421875,
630
+ "learning_rate": 3.072756464904006e-05,
631
+ "loss": 0.9428,
632
  "step": 400
633
  },
634
  {
635
  "epoch": 7.714285714285714,
636
+ "grad_norm": 0.2099609375,
637
+ "learning_rate": 2.8346389036906828e-05,
638
+ "loss": 0.9405,
639
  "step": 405
640
  },
641
  {
642
  "epoch": 7.809523809523809,
643
+ "grad_norm": 0.216796875,
644
+ "learning_rate": 2.6045926771976303e-05,
645
+ "loss": 0.9401,
646
  "step": 410
647
  },
648
  {
649
  "epoch": 7.904761904761905,
650
+ "grad_norm": 0.22265625,
651
+ "learning_rate": 2.382876918226409e-05,
652
+ "loss": 0.9459,
653
  "step": 415
654
  },
655
  {
656
  "epoch": 8.0,
657
+ "grad_norm": 0.2109375,
658
+ "learning_rate": 2.1697413758237784e-05,
659
+ "loss": 0.9415,
660
  "step": 420
661
  },
662
  {
663
  "epoch": 8.0,
664
+ "eval_loss": 2.514326333999634,
665
+ "eval_runtime": 0.4829,
666
+ "eval_samples_per_second": 20.709,
667
+ "eval_steps_per_second": 2.071,
668
  "step": 420
669
  },
670
  {
671
  "epoch": 8.095238095238095,
672
+ "grad_norm": 0.255859375,
673
+ "learning_rate": 1.965426133954854e-05,
674
+ "loss": 0.9302,
675
  "step": 425
676
  },
677
  {
678
  "epoch": 8.19047619047619,
679
+ "grad_norm": 0.2138671875,
680
+ "learning_rate": 1.7701613410634365e-05,
681
+ "loss": 0.941,
682
  "step": 430
683
  },
684
  {
685
  "epoch": 8.285714285714286,
686
+ "grad_norm": 0.21484375,
687
+ "learning_rate": 1.584166950824061e-05,
688
+ "loss": 0.939,
689
  "step": 435
690
  },
691
  {
692
  "epoch": 8.380952380952381,
693
+ "grad_norm": 0.2021484375,
694
+ "learning_rate": 1.4076524743778319e-05,
695
+ "loss": 0.9319,
696
  "step": 440
697
  },
698
  {
699
  "epoch": 8.476190476190476,
700
+ "grad_norm": 0.212890625,
701
+ "learning_rate": 1.2408167443311214e-05,
702
+ "loss": 0.9314,
703
  "step": 445
704
  },
705
  {
706
  "epoch": 8.571428571428571,
707
+ "grad_norm": 0.208984375,
708
+ "learning_rate": 1.083847690782972e-05,
709
+ "loss": 0.9331,
710
  "step": 450
711
  },
712
  {
713
  "epoch": 8.666666666666666,
714
+ "grad_norm": 0.2080078125,
715
+ "learning_rate": 9.369221296335006e-06,
716
+ "loss": 0.9392,
717
  "step": 455
718
  },
719
  {
720
  "epoch": 8.761904761904763,
721
+ "grad_norm": 0.2138671875,
722
+ "learning_rate": 8.002055634117578e-06,
723
+ "loss": 0.9305,
724
  "step": 460
725
  },
726
  {
727
  "epoch": 8.857142857142858,
728
+ "grad_norm": 0.2060546875,
729
+ "learning_rate": 6.738519948473976e-06,
730
+ "loss": 0.9343,
731
  "step": 465
732
  },
733
  {
734
  "epoch": 8.952380952380953,
735
+ "grad_norm": 0.2099609375,
736
+ "learning_rate": 5.580037533961546e-06,
737
+ "loss": 0.9367,
738
  "step": 470
739
  },
740
  {
741
  "epoch": 8.99047619047619,
742
+ "eval_loss": 2.515460252761841,
743
+ "eval_runtime": 0.6608,
744
+ "eval_samples_per_second": 15.132,
745
+ "eval_steps_per_second": 1.513,
746
  "step": 472
747
  },
748
  {
749
  "epoch": 9.047619047619047,
750
+ "grad_norm": 0.2041015625,
751
+ "learning_rate": 4.527913349145441e-06,
752
+ "loss": 0.9329,
753
  "step": 475
754
  },
755
  {
756
  "epoch": 9.142857142857142,
757
+ "grad_norm": 0.2060546875,
758
+ "learning_rate": 3.5833325466437694e-06,
759
+ "loss": 0.9294,
760
  "step": 480
761
  },
762
  {
763
  "epoch": 9.238095238095237,
764
+ "grad_norm": 0.2080078125,
765
+ "learning_rate": 2.7473591381266708e-06,
766
+ "loss": 0.9319,
767
  "step": 485
768
  },
769
  {
770
  "epoch": 9.333333333333334,
771
+ "grad_norm": 0.212890625,
772
+ "learning_rate": 2.0209347957732328e-06,
773
+ "loss": 0.9388,
774
  "step": 490
775
  },
776
  {
777
  "epoch": 9.428571428571429,
778
+ "grad_norm": 0.205078125,
779
+ "learning_rate": 1.404877791536141e-06,
780
+ "loss": 0.9344,
781
  "step": 495
782
  },
783
  {
784
  "epoch": 9.523809523809524,
785
+ "grad_norm": 0.20703125,
786
+ "learning_rate": 8.998820754091531e-07,
787
+ "loss": 0.9345,
788
  "step": 500
789
  },
790
  {
791
  "epoch": 9.619047619047619,
792
+ "grad_norm": 0.203125,
793
+ "learning_rate": 5.065164937354428e-07,
794
+ "loss": 0.9339,
795
  "step": 505
796
  },
797
  {
798
  "epoch": 9.714285714285714,
799
+ "grad_norm": 0.2021484375,
800
+ "learning_rate": 2.2522414843748618e-07,
801
+ "loss": 0.9356,
802
  "step": 510
803
  },
804
  {
805
  "epoch": 9.80952380952381,
806
+ "grad_norm": 0.2021484375,
807
+ "learning_rate": 5.632189789027687e-08,
808
+ "loss": 0.9265,
809
  "step": 515
810
  },
811
  {
812
  "epoch": 9.904761904761905,
813
+ "grad_norm": 0.201171875,
814
+ "learning_rate": 0.0,
815
+ "loss": 0.9348,
816
  "step": 520
817
  },
818
  {
819
+ "epoch": 9.904761904761905,
820
+ "eval_loss": 2.5155773162841797,
821
+ "eval_runtime": 0.4929,
822
+ "eval_samples_per_second": 20.288,
823
+ "eval_steps_per_second": 2.029,
824
+ "step": 520
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
825
  },
826
  {
827
+ "epoch": 9.904761904761905,
828
+ "step": 520,
829
+ "total_flos": 8.18501087353897e+17,
830
+ "train_loss": 1.098491345002101,
831
+ "train_runtime": 2780.3493,
832
+ "train_samples_per_second": 23.972,
833
+ "train_steps_per_second": 0.187
 
 
 
 
 
 
 
834
  }
835
  ],
836
  "logging_steps": 5,
 
850
  "attributes": {}
851
  }
852
  },
853
+ "total_flos": 8.18501087353897e+17,
854
+ "train_batch_size": 8,
855
  "trial_name": null,
856
  "trial_params": null
857
  }