yyx123 commited on
Commit
cfc4e78
1 Parent(s): 5120542

Model save

Browse files
Files changed (5) hide show
  1. README.md +22 -11
  2. all_results.json +8 -8
  3. eval_results.json +4 -4
  4. train_results.json +5 -5
  5. trainer_state.json +1659 -144
README.md CHANGED
@@ -2,13 +2,9 @@
2
  license: other
3
  library_name: peft
4
  tags:
5
- - alignment-handbook
6
- - generated_from_trainer
7
  - trl
8
  - sft
9
  - generated_from_trainer
10
- datasets:
11
- - ruozhiba
12
  base_model: 01-ai/Yi-6B
13
  model-index:
14
  - name: Yi-6B-ruozhiba
@@ -20,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  # Yi-6B-ruozhiba
22
 
23
- This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the ruozhiba dataset.
24
  It achieves the following results on the evaluation set:
25
- - Loss: 1.8389
26
 
27
  ## Model description
28
 
@@ -48,16 +44,31 @@ The following hyperparameters were used during training:
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: cosine
50
  - lr_scheduler_warmup_ratio: 0.1
51
- - num_epochs: 4
52
 
53
  ### Training results
54
 
55
  | Training Loss | Epoch | Step | Validation Loss |
56
  |:-------------:|:-----:|:----:|:---------------:|
57
- | 1.9256 | 1.0 | 55 | 1.8785 |
58
- | 1.6738 | 2.0 | 110 | 1.8229 |
59
- | 1.6181 | 3.0 | 165 | 1.8283 |
60
- | 1.5007 | 4.0 | 220 | 1.8389 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61
 
62
 
63
  ### Framework versions
 
2
  license: other
3
  library_name: peft
4
  tags:
 
 
5
  - trl
6
  - sft
7
  - generated_from_trainer
 
 
8
  base_model: 01-ai/Yi-6B
9
  model-index:
10
  - name: Yi-6B-ruozhiba
 
16
 
17
  # Yi-6B-ruozhiba
18
 
19
+ This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 4.1672
22
 
23
  ## Model description
24
 
 
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
  - lr_scheduler_warmup_ratio: 0.1
47
+ - num_epochs: 20
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:-----:|:----:|:---------------:|
53
+ | 2.0998 | 1.0 | 55 | 2.0215 |
54
+ | 1.7439 | 2.0 | 110 | 1.8371 |
55
+ | 1.7017 | 3.0 | 165 | 1.8182 |
56
+ | 1.3662 | 4.0 | 220 | 1.9196 |
57
+ | 1.0344 | 5.0 | 275 | 2.2265 |
58
+ | 0.6959 | 6.0 | 330 | 2.5187 |
59
+ | 0.4353 | 7.0 | 385 | 2.8303 |
60
+ | 0.2736 | 8.0 | 440 | 3.1702 |
61
+ | 0.1598 | 9.0 | 495 | 3.3739 |
62
+ | 0.117 | 10.0 | 550 | 3.4964 |
63
+ | 0.0856 | 11.0 | 605 | 3.6468 |
64
+ | 0.0681 | 12.0 | 660 | 3.8002 |
65
+ | 0.058 | 13.0 | 715 | 3.9182 |
66
+ | 0.0532 | 14.0 | 770 | 4.0117 |
67
+ | 0.0443 | 15.0 | 825 | 4.0919 |
68
+ | 0.0461 | 17.0 | 935 | 4.1480 |
69
+ | 0.0465 | 18.0 | 990 | 4.1622 |
70
+ | 0.049 | 19.0 | 1045 | 4.1667 |
71
+ | 0.0416 | 20.0 | 1100 | 4.1672 |
72
 
73
 
74
  ### Framework versions
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
- "epoch": 4.0,
3
- "eval_loss": 1.8388779163360596,
4
- "eval_runtime": 4.9283,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 4.667,
7
  "eval_steps_per_second": 1.217,
8
- "train_loss": 1.7501471573656255,
9
- "train_runtime": 1644.9971,
10
  "train_samples": 217,
11
- "train_samples_per_second": 0.528,
12
- "train_steps_per_second": 0.134
13
  }
 
1
  {
2
+ "epoch": 20.0,
3
+ "eval_loss": 4.167160511016846,
4
+ "eval_runtime": 4.9311,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.664,
7
  "eval_steps_per_second": 1.217,
8
+ "train_loss": 0.009085019393400713,
9
+ "train_runtime": 1593.5865,
10
  "train_samples": 217,
11
+ "train_samples_per_second": 2.723,
12
+ "train_steps_per_second": 0.69
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 4.0,
3
- "eval_loss": 1.8388779163360596,
4
- "eval_runtime": 4.9283,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 4.667,
7
  "eval_steps_per_second": 1.217
8
  }
 
1
  {
2
+ "epoch": 20.0,
3
+ "eval_loss": 4.167160511016846,
4
+ "eval_runtime": 4.9311,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.664,
7
  "eval_steps_per_second": 1.217
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 4.0,
3
- "train_loss": 1.7501471573656255,
4
- "train_runtime": 1644.9971,
5
  "train_samples": 217,
6
- "train_samples_per_second": 0.528,
7
- "train_steps_per_second": 0.134
8
  }
 
1
  {
2
+ "epoch": 20.0,
3
+ "train_loss": 0.009085019393400713,
4
+ "train_runtime": 1593.5865,
5
  "train_samples": 217,
6
+ "train_samples_per_second": 2.723,
7
+ "train_steps_per_second": 0.69
8
  }
trainer_state.json CHANGED
@@ -1,417 +1,1932 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 4.0,
5
  "eval_steps": 500,
6
- "global_step": 220,
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.02,
13
- "learning_rate": 2.2727272727272728e-06,
14
  "loss": 2.3833,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.07,
19
- "learning_rate": 9.090909090909091e-06,
20
- "loss": 2.4762,
21
  "step": 4
22
  },
23
  {
24
  "epoch": 0.15,
25
- "learning_rate": 1.8181818181818182e-05,
26
- "loss": 2.2934,
27
  "step": 8
28
  },
29
  {
30
  "epoch": 0.22,
31
- "learning_rate": 2.7272727272727273e-05,
32
- "loss": 2.2509,
33
  "step": 12
34
  },
35
  {
36
  "epoch": 0.29,
37
- "learning_rate": 3.6363636363636364e-05,
38
- "loss": 2.1881,
39
  "step": 16
40
  },
41
  {
42
  "epoch": 0.36,
43
- "learning_rate": 4.545454545454546e-05,
44
- "loss": 2.2269,
45
  "step": 20
46
  },
47
  {
48
  "epoch": 0.44,
49
- "learning_rate": 4.9987413559579636e-05,
50
- "loss": 2.1204,
51
  "step": 24
52
  },
53
  {
54
  "epoch": 0.51,
55
- "learning_rate": 4.988679806432712e-05,
56
- "loss": 2.1803,
57
  "step": 28
58
  },
59
  {
60
  "epoch": 0.58,
61
- "learning_rate": 4.968597221690986e-05,
62
- "loss": 2.0278,
63
  "step": 32
64
  },
65
  {
66
  "epoch": 0.65,
67
- "learning_rate": 4.938574467213518e-05,
68
- "loss": 1.9672,
69
  "step": 36
70
  },
71
  {
72
  "epoch": 0.73,
73
- "learning_rate": 4.898732434036244e-05,
74
- "loss": 1.8425,
75
  "step": 40
76
  },
77
  {
78
  "epoch": 0.8,
79
- "learning_rate": 4.849231551964771e-05,
80
- "loss": 1.9323,
81
  "step": 44
82
  },
83
  {
84
  "epoch": 0.87,
85
- "learning_rate": 4.790271143580174e-05,
86
- "loss": 2.0277,
87
  "step": 48
88
  },
89
  {
90
  "epoch": 0.95,
91
- "learning_rate": 4.722088621637309e-05,
92
- "loss": 1.9256,
93
  "step": 52
94
  },
95
  {
96
  "epoch": 1.0,
97
- "gpt4_scores": 0.6833333333333332,
98
  "step": 55
99
  },
100
  {
101
  "epoch": 1.0,
102
- "eval_loss": 1.878501534461975,
103
- "eval_runtime": 4.9427,
104
- "eval_samples_per_second": 4.653,
105
- "eval_steps_per_second": 1.214,
106
  "step": 55
107
  },
108
  {
109
  "epoch": 1.02,
110
- "learning_rate": 4.644958533087443e-05,
111
- "loss": 1.7957,
112
  "step": 56
113
  },
114
  {
115
  "epoch": 1.09,
116
- "learning_rate": 4.559191453574582e-05,
117
- "loss": 1.8135,
118
  "step": 60
119
  },
120
  {
121
  "epoch": 1.16,
122
- "learning_rate": 4.465132736856969e-05,
123
- "loss": 1.7996,
124
  "step": 64
125
  },
126
  {
127
  "epoch": 1.24,
128
- "learning_rate": 4.3631611241893874e-05,
129
- "loss": 1.7751,
130
  "step": 68
131
  },
132
  {
133
  "epoch": 1.31,
134
- "learning_rate": 4.2536872192658036e-05,
135
- "loss": 1.8436,
136
  "step": 72
137
  },
138
  {
139
  "epoch": 1.38,
140
- "learning_rate": 4.137151834863213e-05,
141
- "loss": 1.7179,
142
  "step": 76
143
  },
144
  {
145
  "epoch": 1.45,
146
- "learning_rate": 4.014024217844167e-05,
147
- "loss": 1.721,
148
  "step": 80
149
  },
150
  {
151
  "epoch": 1.53,
152
- "learning_rate": 3.884800159665276e-05,
153
- "loss": 1.8576,
154
  "step": 84
155
  },
156
  {
157
  "epoch": 1.6,
158
- "learning_rate": 3.7500000000000003e-05,
159
- "loss": 1.8062,
160
  "step": 88
161
  },
162
  {
163
  "epoch": 1.67,
164
- "learning_rate": 3.610166531514436e-05,
165
- "loss": 1.6772,
166
  "step": 92
167
  },
168
  {
169
  "epoch": 1.75,
170
- "learning_rate": 3.465862814232822e-05,
171
- "loss": 1.7685,
172
  "step": 96
173
  },
174
  {
175
  "epoch": 1.82,
176
- "learning_rate": 3.3176699082935545e-05,
177
- "loss": 1.721,
178
  "step": 100
179
  },
180
  {
181
  "epoch": 1.89,
182
- "learning_rate": 3.166184534225087e-05,
183
- "loss": 1.6726,
184
  "step": 104
185
  },
186
  {
187
  "epoch": 1.96,
188
- "learning_rate": 3.012016670162977e-05,
189
- "loss": 1.6738,
190
  "step": 108
191
  },
192
  {
193
  "epoch": 2.0,
194
- "gpt4_scores": 0.6666666666666666,
195
  "step": 110
196
  },
197
  {
198
  "epoch": 2.0,
199
- "eval_loss": 1.8228907585144043,
200
- "eval_runtime": 4.9969,
201
- "eval_samples_per_second": 4.603,
202
- "eval_steps_per_second": 1.201,
203
  "step": 110
204
  },
205
  {
206
  "epoch": 2.04,
207
- "learning_rate": 2.8557870956832132e-05,
208
- "loss": 1.6321,
209
  "step": 112
210
  },
211
  {
212
  "epoch": 2.11,
213
- "learning_rate": 2.698124892141971e-05,
214
- "loss": 1.6618,
215
  "step": 116
216
  },
217
  {
218
  "epoch": 2.18,
219
- "learning_rate": 2.5396649095870202e-05,
220
- "loss": 1.5755,
221
  "step": 120
222
  },
223
  {
224
  "epoch": 2.25,
225
- "learning_rate": 2.3810452104406444e-05,
226
- "loss": 1.6999,
227
  "step": 124
228
  },
229
  {
230
  "epoch": 2.33,
231
- "learning_rate": 2.222904500247473e-05,
232
- "loss": 1.6391,
233
  "step": 128
234
  },
235
  {
236
  "epoch": 2.4,
237
- "learning_rate": 2.0658795558326743e-05,
238
- "loss": 1.583,
239
  "step": 132
240
  },
241
  {
242
  "epoch": 2.47,
243
- "learning_rate": 1.9106026612264316e-05,
244
- "loss": 1.6833,
245
  "step": 136
246
  },
247
  {
248
  "epoch": 2.55,
249
- "learning_rate": 1.7576990616793137e-05,
250
- "loss": 1.6736,
251
  "step": 140
252
  },
253
  {
254
  "epoch": 2.62,
255
- "learning_rate": 1.6077844460203206e-05,
256
- "loss": 1.5683,
257
  "step": 144
258
  },
259
  {
260
  "epoch": 2.69,
261
- "learning_rate": 1.4614624674952842e-05,
262
- "loss": 1.5429,
263
  "step": 148
264
  },
265
  {
266
  "epoch": 2.76,
267
- "learning_rate": 1.3193223130682936e-05,
268
- "loss": 1.6129,
269
  "step": 152
270
  },
271
  {
272
  "epoch": 2.84,
273
- "learning_rate": 1.181936330973744e-05,
274
- "loss": 1.6105,
275
  "step": 156
276
  },
277
  {
278
  "epoch": 2.91,
279
- "learning_rate": 1.049857726072005e-05,
280
- "loss": 1.5663,
281
  "step": 160
282
  },
283
  {
284
  "epoch": 2.98,
285
- "learning_rate": 9.236183322886945e-06,
286
- "loss": 1.6181,
287
  "step": 164
288
  },
289
  {
290
  "epoch": 3.0,
291
- "gpt4_scores": 0.7000000000000001,
292
  "step": 165
293
  },
294
  {
295
  "epoch": 3.0,
296
- "eval_loss": 1.8283250331878662,
297
- "eval_runtime": 4.9725,
298
- "eval_samples_per_second": 4.625,
299
- "eval_steps_per_second": 1.207,
300
  "step": 165
301
  },
302
  {
303
  "epoch": 3.05,
304
- "learning_rate": 8.0372647110717e-06,
305
- "loss": 1.5877,
306
  "step": 168
307
  },
308
  {
309
  "epoch": 3.13,
310
- "learning_rate": 6.906649047373246e-06,
311
- "loss": 1.3838,
312
  "step": 172
313
  },
314
  {
315
  "epoch": 3.2,
316
- "learning_rate": 5.848888922025553e-06,
317
- "loss": 1.5406,
318
  "step": 176
319
  },
320
  {
321
  "epoch": 3.27,
322
- "learning_rate": 4.868243561723535e-06,
323
- "loss": 1.497,
324
  "step": 180
325
  },
326
  {
327
  "epoch": 3.35,
328
- "learning_rate": 3.968661679220468e-06,
329
- "loss": 1.5702,
330
  "step": 184
331
  },
332
  {
333
  "epoch": 3.42,
334
- "learning_rate": 3.1537655732553768e-06,
335
- "loss": 1.6095,
336
  "step": 188
337
  },
338
  {
339
  "epoch": 3.49,
340
- "learning_rate": 2.4268365428344736e-06,
341
- "loss": 1.4712,
342
  "step": 192
343
  },
344
  {
345
  "epoch": 3.56,
346
- "learning_rate": 1.790801674598186e-06,
347
- "loss": 1.5552,
348
  "step": 196
349
  },
350
  {
351
  "epoch": 3.64,
352
- "learning_rate": 1.248222056476367e-06,
353
- "loss": 1.5805,
354
  "step": 200
355
  },
356
  {
357
  "epoch": 3.71,
358
- "learning_rate": 8.012824650910938e-07,
359
- "loss": 1.5588,
360
  "step": 204
361
  },
362
  {
363
  "epoch": 3.78,
364
- "learning_rate": 4.517825684323324e-07,
365
- "loss": 1.64,
366
  "step": 208
367
  },
368
  {
369
  "epoch": 3.85,
370
- "learning_rate": 2.011296792301165e-07,
371
- "loss": 1.5666,
372
  "step": 212
373
  },
374
  {
375
  "epoch": 3.93,
376
- "learning_rate": 5.033308820289184e-08,
377
- "loss": 1.4496,
378
  "step": 216
379
  },
380
  {
381
  "epoch": 4.0,
382
- "learning_rate": 0.0,
383
- "loss": 1.5007,
384
  "step": 220
385
  },
386
  {
387
  "epoch": 4.0,
388
- "gpt4_scores": 0.6666666666666666,
389
  "step": 220
390
  },
391
  {
392
  "epoch": 4.0,
393
- "eval_loss": 1.8388779163360596,
394
- "eval_runtime": 4.9378,
395
- "eval_samples_per_second": 4.658,
396
- "eval_steps_per_second": 1.215,
397
  "step": 220
398
  },
399
  {
400
- "epoch": 4.0,
401
- "step": 220,
402
- "total_flos": 7586259830267904.0,
403
- "train_loss": 1.7501471573656255,
404
- "train_runtime": 1644.9971,
405
- "train_samples_per_second": 0.528,
406
- "train_steps_per_second": 0.134
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
407
  }
408
  ],
409
  "logging_steps": 4,
410
- "max_steps": 220,
411
  "num_input_tokens_seen": 0,
412
- "num_train_epochs": 4,
413
- "save_steps": 20,
414
- "total_flos": 7586259830267904.0,
415
  "train_batch_size": 4,
416
  "trial_name": null,
417
  "trial_params": null
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 20.0,
5
  "eval_steps": 500,
6
+ "global_step": 1100,
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.02,
13
+ "learning_rate": 4.545454545454545e-07,
14
  "loss": 2.3833,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.07,
19
+ "learning_rate": 1.818181818181818e-06,
20
+ "loss": 2.4784,
21
  "step": 4
22
  },
23
  {
24
  "epoch": 0.15,
25
+ "learning_rate": 3.636363636363636e-06,
26
+ "loss": 2.3152,
27
  "step": 8
28
  },
29
  {
30
  "epoch": 0.22,
31
+ "learning_rate": 5.4545454545454545e-06,
32
+ "loss": 2.3229,
33
  "step": 12
34
  },
35
  {
36
  "epoch": 0.29,
37
+ "learning_rate": 7.272727272727272e-06,
38
+ "loss": 2.2926,
39
  "step": 16
40
  },
41
  {
42
  "epoch": 0.36,
43
+ "learning_rate": 9.090909090909091e-06,
44
+ "loss": 2.3494,
45
  "step": 20
46
  },
47
  {
48
  "epoch": 0.44,
49
+ "learning_rate": 1.0909090909090909e-05,
50
+ "loss": 2.3118,
51
  "step": 24
52
  },
53
  {
54
  "epoch": 0.51,
55
+ "learning_rate": 1.2727272727272727e-05,
56
+ "loss": 2.5019,
57
  "step": 28
58
  },
59
  {
60
  "epoch": 0.58,
61
+ "learning_rate": 1.4545454545454545e-05,
62
+ "loss": 2.4156,
63
  "step": 32
64
  },
65
  {
66
  "epoch": 0.65,
67
+ "learning_rate": 1.6363636363636366e-05,
68
+ "loss": 2.2803,
69
  "step": 36
70
  },
71
  {
72
  "epoch": 0.73,
73
+ "learning_rate": 1.8181818181818182e-05,
74
+ "loss": 2.1552,
75
  "step": 40
76
  },
77
  {
78
  "epoch": 0.8,
79
+ "learning_rate": 2e-05,
80
+ "loss": 2.2168,
81
  "step": 44
82
  },
83
  {
84
  "epoch": 0.87,
85
+ "learning_rate": 2.1818181818181818e-05,
86
+ "loss": 2.2977,
87
  "step": 48
88
  },
89
  {
90
  "epoch": 0.95,
91
+ "learning_rate": 2.3636363636363637e-05,
92
+ "loss": 2.0998,
93
  "step": 52
94
  },
95
  {
96
  "epoch": 1.0,
97
+ "gpt4_scores": 0.19999999999999998,
98
  "step": 55
99
  },
100
  {
101
  "epoch": 1.0,
102
+ "eval_loss": 2.0215020179748535,
103
+ "eval_runtime": 4.9485,
104
+ "eval_samples_per_second": 4.648,
105
+ "eval_steps_per_second": 1.212,
106
  "step": 55
107
  },
108
  {
109
  "epoch": 1.02,
110
+ "learning_rate": 2.5454545454545454e-05,
111
+ "loss": 1.9844,
112
  "step": 56
113
  },
114
  {
115
  "epoch": 1.09,
116
+ "learning_rate": 2.7272727272727273e-05,
117
+ "loss": 1.9635,
118
  "step": 60
119
  },
120
  {
121
  "epoch": 1.16,
122
+ "learning_rate": 2.909090909090909e-05,
123
+ "loss": 1.949,
124
  "step": 64
125
  },
126
  {
127
  "epoch": 1.24,
128
+ "learning_rate": 3.090909090909091e-05,
129
+ "loss": 1.9375,
130
  "step": 68
131
  },
132
  {
133
  "epoch": 1.31,
134
+ "learning_rate": 3.272727272727273e-05,
135
+ "loss": 2.0071,
136
  "step": 72
137
  },
138
  {
139
  "epoch": 1.38,
140
+ "learning_rate": 3.454545454545455e-05,
141
+ "loss": 1.8369,
142
  "step": 76
143
  },
144
  {
145
  "epoch": 1.45,
146
+ "learning_rate": 3.6363636363636364e-05,
147
+ "loss": 1.8529,
148
  "step": 80
149
  },
150
  {
151
  "epoch": 1.53,
152
+ "learning_rate": 3.818181818181819e-05,
153
+ "loss": 1.979,
154
  "step": 84
155
  },
156
  {
157
  "epoch": 1.6,
158
+ "learning_rate": 4e-05,
159
+ "loss": 1.9027,
160
  "step": 88
161
  },
162
  {
163
  "epoch": 1.67,
164
+ "learning_rate": 4.181818181818182e-05,
165
+ "loss": 1.7632,
166
  "step": 92
167
  },
168
  {
169
  "epoch": 1.75,
170
+ "learning_rate": 4.3636363636363636e-05,
171
+ "loss": 1.8529,
172
  "step": 96
173
  },
174
  {
175
  "epoch": 1.82,
176
+ "learning_rate": 4.545454545454546e-05,
177
+ "loss": 1.7896,
178
  "step": 100
179
  },
180
  {
181
  "epoch": 1.89,
182
+ "learning_rate": 4.7272727272727275e-05,
183
+ "loss": 1.7356,
184
  "step": 104
185
  },
186
  {
187
  "epoch": 1.96,
188
+ "learning_rate": 4.909090909090909e-05,
189
+ "loss": 1.7439,
190
  "step": 108
191
  },
192
  {
193
  "epoch": 2.0,
194
+ "gpt4_scores": 0.7333333333333334,
195
  "step": 110
196
  },
197
  {
198
  "epoch": 2.0,
199
+ "eval_loss": 1.8371074199676514,
200
+ "eval_runtime": 4.9302,
201
+ "eval_samples_per_second": 4.665,
202
+ "eval_steps_per_second": 1.217,
203
  "step": 110
204
  },
205
  {
206
  "epoch": 2.04,
207
+ "learning_rate": 4.999949650182266e-05,
208
+ "loss": 1.709,
209
  "step": 112
210
  },
211
  {
212
  "epoch": 2.11,
213
+ "learning_rate": 4.999546863808815e-05,
214
+ "loss": 1.7374,
215
  "step": 116
216
  },
217
  {
218
  "epoch": 2.18,
219
+ "learning_rate": 4.9987413559579636e-05,
220
+ "loss": 1.636,
221
  "step": 120
222
  },
223
  {
224
  "epoch": 2.25,
225
+ "learning_rate": 4.99753325641136e-05,
226
+ "loss": 1.7623,
227
  "step": 124
228
  },
229
  {
230
  "epoch": 2.33,
231
+ "learning_rate": 4.995922759815339e-05,
232
+ "loss": 1.7131,
233
  "step": 128
234
  },
235
  {
236
  "epoch": 2.4,
237
+ "learning_rate": 4.993910125649561e-05,
238
+ "loss": 1.65,
239
  "step": 132
240
  },
241
  {
242
  "epoch": 2.47,
243
+ "learning_rate": 4.991495678185202e-05,
244
+ "loss": 1.7748,
245
  "step": 136
246
  },
247
  {
248
  "epoch": 2.55,
249
+ "learning_rate": 4.988679806432712e-05,
250
+ "loss": 1.7411,
251
  "step": 140
252
  },
253
  {
254
  "epoch": 2.62,
255
+ "learning_rate": 4.985462964079137e-05,
256
+ "loss": 1.6517,
257
  "step": 144
258
  },
259
  {
260
  "epoch": 2.69,
261
+ "learning_rate": 4.981845669415022e-05,
262
+ "loss": 1.5844,
263
  "step": 148
264
  },
265
  {
266
  "epoch": 2.76,
267
+ "learning_rate": 4.977828505250903e-05,
268
+ "loss": 1.6811,
269
  "step": 152
270
  },
271
  {
272
  "epoch": 2.84,
273
+ "learning_rate": 4.973412118823412e-05,
274
+ "loss": 1.6825,
275
  "step": 156
276
  },
277
  {
278
  "epoch": 2.91,
279
+ "learning_rate": 4.968597221690986e-05,
280
+ "loss": 1.6176,
281
  "step": 160
282
  },
283
  {
284
  "epoch": 2.98,
285
+ "learning_rate": 4.963384589619233e-05,
286
+ "loss": 1.7017,
287
  "step": 164
288
  },
289
  {
290
  "epoch": 3.0,
291
+ "gpt4_scores": 0.6666666666666666,
292
  "step": 165
293
  },
294
  {
295
  "epoch": 3.0,
296
+ "eval_loss": 1.818211555480957,
297
+ "eval_runtime": 4.9115,
298
+ "eval_samples_per_second": 4.683,
299
+ "eval_steps_per_second": 1.222,
300
  "step": 165
301
  },
302
  {
303
  "epoch": 3.05,
304
+ "learning_rate": 4.957775062455933e-05,
305
+ "loss": 1.59,
306
  "step": 168
307
  },
308
  {
309
  "epoch": 3.13,
310
+ "learning_rate": 4.951769543995731e-05,
311
+ "loss": 1.326,
312
  "step": 172
313
  },
314
  {
315
  "epoch": 3.2,
316
+ "learning_rate": 4.9453690018345144e-05,
317
+ "loss": 1.4622,
318
  "step": 176
319
  },
320
  {
321
  "epoch": 3.27,
322
+ "learning_rate": 4.938574467213518e-05,
323
+ "loss": 1.4188,
324
  "step": 180
325
  },
326
  {
327
  "epoch": 3.35,
328
+ "learning_rate": 4.931387034853173e-05,
329
+ "loss": 1.4488,
330
  "step": 184
331
  },
332
  {
333
  "epoch": 3.42,
334
+ "learning_rate": 4.923807862776728e-05,
335
+ "loss": 1.5268,
336
  "step": 188
337
  },
338
  {
339
  "epoch": 3.49,
340
+ "learning_rate": 4.915838172123671e-05,
341
+ "loss": 1.3807,
342
  "step": 192
343
  },
344
  {
345
  "epoch": 3.56,
346
+ "learning_rate": 4.9074792469529815e-05,
347
+ "loss": 1.4319,
348
  "step": 196
349
  },
350
  {
351
  "epoch": 3.64,
352
+ "learning_rate": 4.898732434036244e-05,
353
+ "loss": 1.4653,
354
  "step": 200
355
  },
356
  {
357
  "epoch": 3.71,
358
+ "learning_rate": 4.889599142640663e-05,
359
+ "loss": 1.417,
360
  "step": 204
361
  },
362
  {
363
  "epoch": 3.78,
364
+ "learning_rate": 4.880080844302004e-05,
365
+ "loss": 1.4848,
366
  "step": 208
367
  },
368
  {
369
  "epoch": 3.85,
370
+ "learning_rate": 4.870179072587499e-05,
371
+ "loss": 1.4331,
372
  "step": 212
373
  },
374
  {
375
  "epoch": 3.93,
376
+ "learning_rate": 4.859895422848767e-05,
377
+ "loss": 1.3679,
378
  "step": 216
379
  },
380
  {
381
  "epoch": 4.0,
382
+ "learning_rate": 4.849231551964771e-05,
383
+ "loss": 1.3662,
384
  "step": 220
385
  },
386
  {
387
  "epoch": 4.0,
388
+ "gpt4_scores": 0.75,
389
  "step": 220
390
  },
391
  {
392
  "epoch": 4.0,
393
+ "eval_loss": 1.9195555448532104,
394
+ "eval_runtime": 4.9614,
395
+ "eval_samples_per_second": 4.636,
396
+ "eval_steps_per_second": 1.209,
397
  "step": 220
398
  },
399
  {
400
+ "epoch": 4.07,
401
+ "learning_rate": 4.838189178074867e-05,
402
+ "loss": 1.1767,
403
+ "step": 224
404
+ },
405
+ {
406
+ "epoch": 4.15,
407
+ "learning_rate": 4.826770080301978e-05,
408
+ "loss": 1.1217,
409
+ "step": 228
410
+ },
411
+ {
412
+ "epoch": 4.22,
413
+ "learning_rate": 4.8149760984659506e-05,
414
+ "loss": 1.1938,
415
+ "step": 232
416
+ },
417
+ {
418
+ "epoch": 4.29,
419
+ "learning_rate": 4.802809132787125e-05,
420
+ "loss": 1.0658,
421
+ "step": 236
422
+ },
423
+ {
424
+ "epoch": 4.36,
425
+ "learning_rate": 4.790271143580174e-05,
426
+ "loss": 0.9068,
427
+ "step": 240
428
+ },
429
+ {
430
+ "epoch": 4.44,
431
+ "learning_rate": 4.777364150938263e-05,
432
+ "loss": 1.0878,
433
+ "step": 244
434
+ },
435
+ {
436
+ "epoch": 4.51,
437
+ "learning_rate": 4.764090234407577e-05,
438
+ "loss": 1.0284,
439
+ "step": 248
440
+ },
441
+ {
442
+ "epoch": 4.58,
443
+ "learning_rate": 4.75045153265227e-05,
444
+ "loss": 1.0695,
445
+ "step": 252
446
+ },
447
+ {
448
+ "epoch": 4.65,
449
+ "learning_rate": 4.7364502431098844e-05,
450
+ "loss": 1.1011,
451
+ "step": 256
452
+ },
453
+ {
454
+ "epoch": 4.73,
455
+ "learning_rate": 4.722088621637309e-05,
456
+ "loss": 1.0877,
457
+ "step": 260
458
+ },
459
+ {
460
+ "epoch": 4.8,
461
+ "learning_rate": 4.707368982147318e-05,
462
+ "loss": 0.9747,
463
+ "step": 264
464
+ },
465
+ {
466
+ "epoch": 4.87,
467
+ "learning_rate": 4.692293696235758e-05,
468
+ "loss": 0.9563,
469
+ "step": 268
470
+ },
471
+ {
472
+ "epoch": 4.95,
473
+ "learning_rate": 4.6768651927994434e-05,
474
+ "loss": 1.0344,
475
+ "step": 272
476
+ },
477
+ {
478
+ "epoch": 5.0,
479
+ "gpt4_scores": 0.6,
480
+ "step": 275
481
+ },
482
+ {
483
+ "epoch": 5.0,
484
+ "eval_loss": 2.226499557495117,
485
+ "eval_runtime": 4.9165,
486
+ "eval_samples_per_second": 4.678,
487
+ "eval_steps_per_second": 1.22,
488
+ "step": 275
489
+ },
490
+ {
491
+ "epoch": 5.02,
492
+ "learning_rate": 4.6610859576448176e-05,
493
+ "loss": 0.9582,
494
+ "step": 276
495
+ },
496
+ {
497
+ "epoch": 5.09,
498
+ "learning_rate": 4.644958533087443e-05,
499
+ "loss": 0.7354,
500
+ "step": 280
501
+ },
502
+ {
503
+ "epoch": 5.16,
504
+ "learning_rate": 4.628485517542392e-05,
505
+ "loss": 0.7346,
506
+ "step": 284
507
+ },
508
+ {
509
+ "epoch": 5.24,
510
+ "learning_rate": 4.611669565105596e-05,
511
+ "loss": 0.6712,
512
+ "step": 288
513
+ },
514
+ {
515
+ "epoch": 5.31,
516
+ "learning_rate": 4.594513385126218e-05,
517
+ "loss": 0.7341,
518
+ "step": 292
519
+ },
520
+ {
521
+ "epoch": 5.38,
522
+ "learning_rate": 4.5770197417701365e-05,
523
+ "loss": 0.6719,
524
+ "step": 296
525
+ },
526
+ {
527
+ "epoch": 5.45,
528
+ "learning_rate": 4.559191453574582e-05,
529
+ "loss": 0.6348,
530
+ "step": 300
531
+ },
532
+ {
533
+ "epoch": 5.53,
534
+ "learning_rate": 4.5410313929940244e-05,
535
+ "loss": 0.7058,
536
+ "step": 304
537
+ },
538
+ {
539
+ "epoch": 5.6,
540
+ "learning_rate": 4.522542485937369e-05,
541
+ "loss": 0.6653,
542
+ "step": 308
543
+ },
544
+ {
545
+ "epoch": 5.67,
546
+ "learning_rate": 4.503727711296538e-05,
547
+ "loss": 0.6575,
548
+ "step": 312
549
+ },
550
+ {
551
+ "epoch": 5.75,
552
+ "learning_rate": 4.4845901004665234e-05,
553
+ "loss": 0.6561,
554
+ "step": 316
555
+ },
556
+ {
557
+ "epoch": 5.82,
558
+ "learning_rate": 4.465132736856969e-05,
559
+ "loss": 0.6728,
560
+ "step": 320
561
+ },
562
+ {
563
+ "epoch": 5.89,
564
+ "learning_rate": 4.445358755395382e-05,
565
+ "loss": 0.703,
566
+ "step": 324
567
+ },
568
+ {
569
+ "epoch": 5.96,
570
+ "learning_rate": 4.425271342022039e-05,
571
+ "loss": 0.6959,
572
+ "step": 328
573
+ },
574
+ {
575
+ "epoch": 6.0,
576
+ "gpt4_scores": 0.7666666666666666,
577
+ "step": 330
578
+ },
579
+ {
580
+ "epoch": 6.0,
581
+ "eval_loss": 2.5187010765075684,
582
+ "eval_runtime": 4.9193,
583
+ "eval_samples_per_second": 4.675,
584
+ "eval_steps_per_second": 1.22,
585
+ "step": 330
586
+ },
587
+ {
588
+ "epoch": 6.04,
589
+ "learning_rate": 4.404873733176678e-05,
590
+ "loss": 0.6332,
591
+ "step": 332
592
+ },
593
+ {
594
+ "epoch": 6.11,
595
+ "learning_rate": 4.384169215277041e-05,
596
+ "loss": 0.3578,
597
+ "step": 336
598
+ },
599
+ {
600
+ "epoch": 6.18,
601
+ "learning_rate": 4.3631611241893874e-05,
602
+ "loss": 0.3878,
603
+ "step": 340
604
+ },
605
+ {
606
+ "epoch": 6.25,
607
+ "learning_rate": 4.341852844691012e-05,
608
+ "loss": 0.3941,
609
+ "step": 344
610
+ },
611
+ {
612
+ "epoch": 6.33,
613
+ "learning_rate": 4.3202478099249105e-05,
614
+ "loss": 0.3927,
615
+ "step": 348
616
+ },
617
+ {
618
+ "epoch": 6.4,
619
+ "learning_rate": 4.2983495008466276e-05,
620
+ "loss": 0.4278,
621
+ "step": 352
622
+ },
623
+ {
624
+ "epoch": 6.47,
625
+ "learning_rate": 4.276161445663423e-05,
626
+ "loss": 0.4638,
627
+ "step": 356
628
+ },
629
+ {
630
+ "epoch": 6.55,
631
+ "learning_rate": 4.2536872192658036e-05,
632
+ "loss": 0.4473,
633
+ "step": 360
634
+ },
635
+ {
636
+ "epoch": 6.62,
637
+ "learning_rate": 4.230930442651557e-05,
638
+ "loss": 0.4563,
639
+ "step": 364
640
+ },
641
+ {
642
+ "epoch": 6.69,
643
+ "learning_rate": 4.2078947823423364e-05,
644
+ "loss": 0.3735,
645
+ "step": 368
646
+ },
647
+ {
648
+ "epoch": 6.76,
649
+ "learning_rate": 4.1845839497929204e-05,
650
+ "loss": 0.4068,
651
+ "step": 372
652
+ },
653
+ {
654
+ "epoch": 6.84,
655
+ "learning_rate": 4.161001700793231e-05,
656
+ "loss": 0.455,
657
+ "step": 376
658
+ },
659
+ {
660
+ "epoch": 6.91,
661
+ "learning_rate": 4.137151834863213e-05,
662
+ "loss": 0.4346,
663
+ "step": 380
664
+ },
665
+ {
666
+ "epoch": 6.98,
667
+ "learning_rate": 4.113038194640658e-05,
668
+ "loss": 0.4353,
669
+ "step": 384
670
+ },
671
+ {
672
+ "epoch": 7.0,
673
+ "gpt4_scores": 0.75,
674
+ "step": 385
675
+ },
676
+ {
677
+ "epoch": 7.0,
678
+ "eval_loss": 2.8303465843200684,
679
+ "eval_runtime": 4.9148,
680
+ "eval_samples_per_second": 4.68,
681
+ "eval_steps_per_second": 1.221,
682
+ "step": 385
683
+ },
684
+ {
685
+ "epoch": 7.05,
686
+ "learning_rate": 4.088664665262091e-05,
687
+ "loss": 0.3461,
688
+ "step": 388
689
+ },
690
+ {
691
+ "epoch": 7.13,
692
+ "learning_rate": 4.064035173736804e-05,
693
+ "loss": 0.2204,
694
+ "step": 392
695
+ },
696
+ {
697
+ "epoch": 7.2,
698
+ "learning_rate": 4.039153688314145e-05,
699
+ "loss": 0.2423,
700
+ "step": 396
701
+ },
702
+ {
703
+ "epoch": 7.27,
704
+ "learning_rate": 4.014024217844167e-05,
705
+ "loss": 0.2308,
706
+ "step": 400
707
+ },
708
+ {
709
+ "epoch": 7.35,
710
+ "learning_rate": 3.98865081113172e-05,
711
+ "loss": 0.2553,
712
+ "step": 404
713
+ },
714
+ {
715
+ "epoch": 7.42,
716
+ "learning_rate": 3.9630375562841295e-05,
717
+ "loss": 0.2258,
718
+ "step": 408
719
+ },
720
+ {
721
+ "epoch": 7.49,
722
+ "learning_rate": 3.937188580052518e-05,
723
+ "loss": 0.264,
724
+ "step": 412
725
+ },
726
+ {
727
+ "epoch": 7.56,
728
+ "learning_rate": 3.911108047166924e-05,
729
+ "loss": 0.2548,
730
+ "step": 416
731
+ },
732
+ {
733
+ "epoch": 7.64,
734
+ "learning_rate": 3.884800159665276e-05,
735
+ "loss": 0.272,
736
+ "step": 420
737
+ },
738
+ {
739
+ "epoch": 7.71,
740
+ "learning_rate": 3.858269156216383e-05,
741
+ "loss": 0.2596,
742
+ "step": 424
743
+ },
744
+ {
745
+ "epoch": 7.78,
746
+ "learning_rate": 3.8315193114369996e-05,
747
+ "loss": 0.24,
748
+ "step": 428
749
+ },
750
+ {
751
+ "epoch": 7.85,
752
+ "learning_rate": 3.804554935203115e-05,
753
+ "loss": 0.2434,
754
+ "step": 432
755
+ },
756
+ {
757
+ "epoch": 7.93,
758
+ "learning_rate": 3.7773803719555514e-05,
759
+ "loss": 0.2694,
760
+ "step": 436
761
+ },
762
+ {
763
+ "epoch": 8.0,
764
+ "learning_rate": 3.7500000000000003e-05,
765
+ "loss": 0.2736,
766
+ "step": 440
767
+ },
768
+ {
769
+ "epoch": 8.0,
770
+ "gpt4_scores": 0.5833333333333334,
771
+ "step": 440
772
+ },
773
+ {
774
+ "epoch": 8.0,
775
+ "eval_loss": 3.1702194213867188,
776
+ "eval_runtime": 4.9672,
777
+ "eval_samples_per_second": 4.63,
778
+ "eval_steps_per_second": 1.208,
779
+ "step": 440
780
+ },
781
+ {
782
+ "epoch": 8.07,
783
+ "learning_rate": 3.7224182308015975e-05,
784
+ "loss": 0.1419,
785
+ "step": 444
786
+ },
787
+ {
788
+ "epoch": 8.15,
789
+ "learning_rate": 3.694639508274158e-05,
790
+ "loss": 0.148,
791
+ "step": 448
792
+ },
793
+ {
794
+ "epoch": 8.22,
795
+ "learning_rate": 3.6666683080641846e-05,
796
+ "loss": 0.1635,
797
+ "step": 452
798
+ },
799
+ {
800
+ "epoch": 8.29,
801
+ "learning_rate": 3.638509136829758e-05,
802
+ "loss": 0.1667,
803
+ "step": 456
804
+ },
805
+ {
806
+ "epoch": 8.36,
807
+ "learning_rate": 3.610166531514436e-05,
808
+ "loss": 0.164,
809
+ "step": 460
810
+ },
811
+ {
812
+ "epoch": 8.44,
813
+ "learning_rate": 3.581645058616271e-05,
814
+ "loss": 0.1428,
815
+ "step": 464
816
+ },
817
+ {
818
+ "epoch": 8.51,
819
+ "learning_rate": 3.552949313452067e-05,
820
+ "loss": 0.1392,
821
+ "step": 468
822
+ },
823
+ {
824
+ "epoch": 8.58,
825
+ "learning_rate": 3.5240839194169885e-05,
826
+ "loss": 0.1557,
827
+ "step": 472
828
+ },
829
+ {
830
+ "epoch": 8.65,
831
+ "learning_rate": 3.495053527239656e-05,
832
+ "loss": 0.1714,
833
+ "step": 476
834
+ },
835
+ {
836
+ "epoch": 8.73,
837
+ "learning_rate": 3.465862814232822e-05,
838
+ "loss": 0.1622,
839
+ "step": 480
840
+ },
841
+ {
842
+ "epoch": 8.8,
843
+ "learning_rate": 3.436516483539781e-05,
844
+ "loss": 0.18,
845
+ "step": 484
846
+ },
847
+ {
848
+ "epoch": 8.87,
849
+ "learning_rate": 3.4070192633766025e-05,
850
+ "loss": 0.1654,
851
+ "step": 488
852
+ },
853
+ {
854
+ "epoch": 8.95,
855
+ "learning_rate": 3.3773759062703396e-05,
856
+ "loss": 0.1598,
857
+ "step": 492
858
+ },
859
+ {
860
+ "epoch": 9.0,
861
+ "gpt4_scores": 0.5499999999999999,
862
+ "step": 495
863
+ },
864
+ {
865
+ "epoch": 9.0,
866
+ "eval_loss": 3.373944044113159,
867
+ "eval_runtime": 4.9104,
868
+ "eval_samples_per_second": 4.684,
869
+ "eval_steps_per_second": 1.222,
870
+ "step": 495
871
+ },
872
+ {
873
+ "epoch": 9.02,
874
+ "learning_rate": 3.3475911882933015e-05,
875
+ "loss": 0.1388,
876
+ "step": 496
877
+ },
878
+ {
879
+ "epoch": 9.09,
880
+ "learning_rate": 3.3176699082935545e-05,
881
+ "loss": 0.1064,
882
+ "step": 500
883
+ },
884
+ {
885
+ "epoch": 9.16,
886
+ "learning_rate": 3.2876168871217325e-05,
887
+ "loss": 0.1153,
888
+ "step": 504
889
+ },
890
+ {
891
+ "epoch": 9.24,
892
+ "learning_rate": 3.257436966854319e-05,
893
+ "loss": 0.1225,
894
+ "step": 508
895
+ },
896
+ {
897
+ "epoch": 9.31,
898
+ "learning_rate": 3.2271350100134975e-05,
899
+ "loss": 0.1146,
900
+ "step": 512
901
+ },
902
+ {
903
+ "epoch": 9.38,
904
+ "learning_rate": 3.19671589878372e-05,
905
+ "loss": 0.1202,
906
+ "step": 516
907
+ },
908
+ {
909
+ "epoch": 9.45,
910
+ "learning_rate": 3.166184534225087e-05,
911
+ "loss": 0.1132,
912
+ "step": 520
913
+ },
914
+ {
915
+ "epoch": 9.53,
916
+ "learning_rate": 3.135545835483718e-05,
917
+ "loss": 0.106,
918
+ "step": 524
919
+ },
920
+ {
921
+ "epoch": 9.6,
922
+ "learning_rate": 3.104804738999169e-05,
923
+ "loss": 0.1127,
924
+ "step": 528
925
+ },
926
+ {
927
+ "epoch": 9.67,
928
+ "learning_rate": 3.073966197709103e-05,
929
+ "loss": 0.1059,
930
+ "step": 532
931
+ },
932
+ {
933
+ "epoch": 9.75,
934
+ "learning_rate": 3.0430351802512698e-05,
935
+ "loss": 0.1089,
936
+ "step": 536
937
+ },
938
+ {
939
+ "epoch": 9.82,
940
+ "learning_rate": 3.012016670162977e-05,
941
+ "loss": 0.1139,
942
+ "step": 540
943
+ },
944
+ {
945
+ "epoch": 9.89,
946
+ "learning_rate": 2.9809156650781528e-05,
947
+ "loss": 0.1247,
948
+ "step": 544
949
+ },
950
+ {
951
+ "epoch": 9.96,
952
+ "learning_rate": 2.9497371759221347e-05,
953
+ "loss": 0.117,
954
+ "step": 548
955
+ },
956
+ {
957
+ "epoch": 10.0,
958
+ "gpt4_scores": 0.7000000000000001,
959
+ "step": 550
960
+ },
961
+ {
962
+ "epoch": 10.0,
963
+ "eval_loss": 3.4963889122009277,
964
+ "eval_runtime": 4.9532,
965
+ "eval_samples_per_second": 4.643,
966
+ "eval_steps_per_second": 1.211,
967
+ "step": 550
968
+ },
969
+ {
970
+ "epoch": 10.04,
971
+ "learning_rate": 2.918486226104327e-05,
972
+ "loss": 0.0894,
973
+ "step": 552
974
+ },
975
+ {
976
+ "epoch": 10.11,
977
+ "learning_rate": 2.8871678507088312e-05,
978
+ "loss": 0.0835,
979
+ "step": 556
980
+ },
981
+ {
982
+ "epoch": 10.18,
983
+ "learning_rate": 2.8557870956832132e-05,
984
+ "loss": 0.0913,
985
+ "step": 560
986
+ },
987
+ {
988
+ "epoch": 10.25,
989
+ "learning_rate": 2.8243490170255043e-05,
990
+ "loss": 0.0737,
991
+ "step": 564
992
+ },
993
+ {
994
+ "epoch": 10.33,
995
+ "learning_rate": 2.792858679969596e-05,
996
+ "loss": 0.0897,
997
+ "step": 568
998
+ },
999
+ {
1000
+ "epoch": 10.4,
1001
+ "learning_rate": 2.761321158169134e-05,
1002
+ "loss": 0.0825,
1003
+ "step": 572
1004
+ },
1005
+ {
1006
+ "epoch": 10.47,
1007
+ "learning_rate": 2.7297415328800692e-05,
1008
+ "loss": 0.0845,
1009
+ "step": 576
1010
+ },
1011
+ {
1012
+ "epoch": 10.55,
1013
+ "learning_rate": 2.698124892141971e-05,
1014
+ "loss": 0.0819,
1015
+ "step": 580
1016
+ },
1017
+ {
1018
+ "epoch": 10.62,
1019
+ "learning_rate": 2.6664763299582602e-05,
1020
+ "loss": 0.0868,
1021
+ "step": 584
1022
+ },
1023
+ {
1024
+ "epoch": 10.69,
1025
+ "learning_rate": 2.6348009454754653e-05,
1026
+ "loss": 0.083,
1027
+ "step": 588
1028
+ },
1029
+ {
1030
+ "epoch": 10.76,
1031
+ "learning_rate": 2.6031038421616683e-05,
1032
+ "loss": 0.0937,
1033
+ "step": 592
1034
+ },
1035
+ {
1036
+ "epoch": 10.84,
1037
+ "learning_rate": 2.5713901269842404e-05,
1038
+ "loss": 0.0872,
1039
+ "step": 596
1040
+ },
1041
+ {
1042
+ "epoch": 10.91,
1043
+ "learning_rate": 2.5396649095870202e-05,
1044
+ "loss": 0.0869,
1045
+ "step": 600
1046
+ },
1047
+ {
1048
+ "epoch": 10.98,
1049
+ "learning_rate": 2.507933301467056e-05,
1050
+ "loss": 0.0856,
1051
+ "step": 604
1052
+ },
1053
+ {
1054
+ "epoch": 11.0,
1055
+ "gpt4_scores": 0.48333333333333334,
1056
+ "step": 605
1057
+ },
1058
+ {
1059
+ "epoch": 11.0,
1060
+ "eval_loss": 3.6468489170074463,
1061
+ "eval_runtime": 4.9449,
1062
+ "eval_samples_per_second": 4.651,
1063
+ "eval_steps_per_second": 1.213,
1064
+ "step": 605
1065
+ },
1066
+ {
1067
+ "epoch": 11.05,
1068
+ "learning_rate": 2.4762004151510584e-05,
1069
+ "loss": 0.0808,
1070
+ "step": 608
1071
+ },
1072
+ {
1073
+ "epoch": 11.13,
1074
+ "learning_rate": 2.4444713633716765e-05,
1075
+ "loss": 0.0641,
1076
+ "step": 612
1077
+ },
1078
+ {
1079
+ "epoch": 11.2,
1080
+ "learning_rate": 2.4127512582437485e-05,
1081
+ "loss": 0.0699,
1082
+ "step": 616
1083
+ },
1084
+ {
1085
+ "epoch": 11.27,
1086
+ "learning_rate": 2.3810452104406444e-05,
1087
+ "loss": 0.0692,
1088
+ "step": 620
1089
+ },
1090
+ {
1091
+ "epoch": 11.35,
1092
+ "learning_rate": 2.349358328370854e-05,
1093
+ "loss": 0.0689,
1094
+ "step": 624
1095
+ },
1096
+ {
1097
+ "epoch": 11.42,
1098
+ "learning_rate": 2.3176957173549235e-05,
1099
+ "loss": 0.0763,
1100
+ "step": 628
1101
+ },
1102
+ {
1103
+ "epoch": 11.49,
1104
+ "learning_rate": 2.2860624788029013e-05,
1105
+ "loss": 0.0609,
1106
+ "step": 632
1107
+ },
1108
+ {
1109
+ "epoch": 11.56,
1110
+ "learning_rate": 2.2544637093924074e-05,
1111
+ "loss": 0.0679,
1112
+ "step": 636
1113
+ },
1114
+ {
1115
+ "epoch": 11.64,
1116
+ "learning_rate": 2.222904500247473e-05,
1117
+ "loss": 0.0675,
1118
+ "step": 640
1119
+ },
1120
+ {
1121
+ "epoch": 11.71,
1122
+ "learning_rate": 2.1913899361182632e-05,
1123
+ "loss": 0.0741,
1124
+ "step": 644
1125
+ },
1126
+ {
1127
+ "epoch": 11.78,
1128
+ "learning_rate": 2.1599250945618402e-05,
1129
+ "loss": 0.0644,
1130
+ "step": 648
1131
+ },
1132
+ {
1133
+ "epoch": 11.85,
1134
+ "learning_rate": 2.1285150451240713e-05,
1135
+ "loss": 0.0672,
1136
+ "step": 652
1137
+ },
1138
+ {
1139
+ "epoch": 11.93,
1140
+ "learning_rate": 2.09716484852284e-05,
1141
+ "loss": 0.0722,
1142
+ "step": 656
1143
+ },
1144
+ {
1145
+ "epoch": 12.0,
1146
+ "learning_rate": 2.0658795558326743e-05,
1147
+ "loss": 0.0681,
1148
+ "step": 660
1149
+ },
1150
+ {
1151
+ "epoch": 12.0,
1152
+ "gpt4_scores": 0.7166666666666667,
1153
+ "step": 660
1154
+ },
1155
+ {
1156
+ "epoch": 12.0,
1157
+ "eval_loss": 3.8001902103424072,
1158
+ "eval_runtime": 4.9149,
1159
+ "eval_samples_per_second": 4.68,
1160
+ "eval_steps_per_second": 1.221,
1161
+ "step": 660
1162
+ },
1163
+ {
1164
+ "epoch": 12.07,
1165
+ "learning_rate": 2.034664207670925e-05,
1166
+ "loss": 0.0662,
1167
+ "step": 664
1168
+ },
1169
+ {
1170
+ "epoch": 12.15,
1171
+ "learning_rate": 2.003523833385637e-05,
1172
+ "loss": 0.0572,
1173
+ "step": 668
1174
+ },
1175
+ {
1176
+ "epoch": 12.22,
1177
+ "learning_rate": 1.972463450245226e-05,
1178
+ "loss": 0.061,
1179
+ "step": 672
1180
+ },
1181
+ {
1182
+ "epoch": 12.29,
1183
+ "learning_rate": 1.9414880626301147e-05,
1184
+ "loss": 0.0611,
1185
+ "step": 676
1186
+ },
1187
+ {
1188
+ "epoch": 12.36,
1189
+ "learning_rate": 1.9106026612264316e-05,
1190
+ "loss": 0.0631,
1191
+ "step": 680
1192
+ },
1193
+ {
1194
+ "epoch": 12.44,
1195
+ "learning_rate": 1.879812222221929e-05,
1196
+ "loss": 0.0586,
1197
+ "step": 684
1198
+ },
1199
+ {
1200
+ "epoch": 12.51,
1201
+ "learning_rate": 1.84912170650422e-05,
1202
+ "loss": 0.0597,
1203
+ "step": 688
1204
+ },
1205
+ {
1206
+ "epoch": 12.58,
1207
+ "learning_rate": 1.8185360588615058e-05,
1208
+ "loss": 0.0566,
1209
+ "step": 692
1210
+ },
1211
+ {
1212
+ "epoch": 12.65,
1213
+ "learning_rate": 1.7880602071858692e-05,
1214
+ "loss": 0.0569,
1215
+ "step": 696
1216
+ },
1217
+ {
1218
+ "epoch": 12.73,
1219
+ "learning_rate": 1.7576990616793137e-05,
1220
+ "loss": 0.0548,
1221
+ "step": 700
1222
+ },
1223
+ {
1224
+ "epoch": 12.8,
1225
+ "learning_rate": 1.7274575140626318e-05,
1226
+ "loss": 0.058,
1227
+ "step": 704
1228
+ },
1229
+ {
1230
+ "epoch": 12.87,
1231
+ "learning_rate": 1.697340436787273e-05,
1232
+ "loss": 0.0512,
1233
+ "step": 708
1234
+ },
1235
+ {
1236
+ "epoch": 12.95,
1237
+ "learning_rate": 1.667352682250298e-05,
1238
+ "loss": 0.058,
1239
+ "step": 712
1240
+ },
1241
+ {
1242
+ "epoch": 13.0,
1243
+ "gpt4_scores": 0.5,
1244
+ "step": 715
1245
+ },
1246
+ {
1247
+ "epoch": 13.0,
1248
+ "eval_loss": 3.918207883834839,
1249
+ "eval_runtime": 4.9544,
1250
+ "eval_samples_per_second": 4.642,
1251
+ "eval_steps_per_second": 1.211,
1252
+ "step": 715
1253
+ },
1254
+ {
1255
+ "epoch": 13.02,
1256
+ "learning_rate": 1.637499082012574e-05,
1257
+ "loss": 0.0681,
1258
+ "step": 716
1259
+ },
1260
+ {
1261
+ "epoch": 13.09,
1262
+ "learning_rate": 1.6077844460203206e-05,
1263
+ "loss": 0.0547,
1264
+ "step": 720
1265
+ },
1266
+ {
1267
+ "epoch": 13.16,
1268
+ "learning_rate": 1.5782135618301486e-05,
1269
+ "loss": 0.0449,
1270
+ "step": 724
1271
+ },
1272
+ {
1273
+ "epoch": 13.24,
1274
+ "learning_rate": 1.5487911938376924e-05,
1275
+ "loss": 0.0555,
1276
+ "step": 728
1277
+ },
1278
+ {
1279
+ "epoch": 13.31,
1280
+ "learning_rate": 1.5195220825099862e-05,
1281
+ "loss": 0.0508,
1282
+ "step": 732
1283
+ },
1284
+ {
1285
+ "epoch": 13.38,
1286
+ "learning_rate": 1.4904109436216884e-05,
1287
+ "loss": 0.0527,
1288
+ "step": 736
1289
+ },
1290
+ {
1291
+ "epoch": 13.45,
1292
+ "learning_rate": 1.4614624674952842e-05,
1293
+ "loss": 0.0521,
1294
+ "step": 740
1295
+ },
1296
+ {
1297
+ "epoch": 13.53,
1298
+ "learning_rate": 1.4326813182453958e-05,
1299
+ "loss": 0.0554,
1300
+ "step": 744
1301
+ },
1302
+ {
1303
+ "epoch": 13.6,
1304
+ "learning_rate": 1.4040721330273062e-05,
1305
+ "loss": 0.05,
1306
+ "step": 748
1307
+ },
1308
+ {
1309
+ "epoch": 13.67,
1310
+ "learning_rate": 1.3756395212898359e-05,
1311
+ "loss": 0.0509,
1312
+ "step": 752
1313
+ },
1314
+ {
1315
+ "epoch": 13.75,
1316
+ "learning_rate": 1.3473880640326725e-05,
1317
+ "loss": 0.0517,
1318
+ "step": 756
1319
+ },
1320
+ {
1321
+ "epoch": 13.82,
1322
+ "learning_rate": 1.3193223130682936e-05,
1323
+ "loss": 0.0557,
1324
+ "step": 760
1325
+ },
1326
+ {
1327
+ "epoch": 13.89,
1328
+ "learning_rate": 1.2914467902885901e-05,
1329
+ "loss": 0.0524,
1330
+ "step": 764
1331
+ },
1332
+ {
1333
+ "epoch": 13.96,
1334
+ "learning_rate": 1.2637659869363083e-05,
1335
+ "loss": 0.0532,
1336
+ "step": 768
1337
+ },
1338
+ {
1339
+ "epoch": 14.0,
1340
+ "gpt4_scores": 0.5166666666666667,
1341
+ "step": 770
1342
+ },
1343
+ {
1344
+ "epoch": 14.0,
1345
+ "eval_loss": 4.011674880981445,
1346
+ "eval_runtime": 4.9088,
1347
+ "eval_samples_per_second": 4.685,
1348
+ "eval_steps_per_second": 1.222,
1349
+ "step": 770
1350
+ },
1351
+ {
1352
+ "epoch": 14.04,
1353
+ "learning_rate": 1.2362843628814267e-05,
1354
+ "loss": 0.0456,
1355
+ "step": 772
1356
+ },
1357
+ {
1358
+ "epoch": 14.11,
1359
+ "learning_rate": 1.2090063459025955e-05,
1360
+ "loss": 0.0476,
1361
+ "step": 776
1362
+ },
1363
+ {
1364
+ "epoch": 14.18,
1365
+ "learning_rate": 1.181936330973744e-05,
1366
+ "loss": 0.0501,
1367
+ "step": 780
1368
+ },
1369
+ {
1370
+ "epoch": 14.25,
1371
+ "learning_rate": 1.155078679555969e-05,
1372
+ "loss": 0.0497,
1373
+ "step": 784
1374
+ },
1375
+ {
1376
+ "epoch": 14.33,
1377
+ "learning_rate": 1.1284377188948258e-05,
1378
+ "loss": 0.0496,
1379
+ "step": 788
1380
+ },
1381
+ {
1382
+ "epoch": 14.4,
1383
+ "learning_rate": 1.1020177413231334e-05,
1384
+ "loss": 0.0467,
1385
+ "step": 792
1386
+ },
1387
+ {
1388
+ "epoch": 14.47,
1389
+ "learning_rate": 1.0758230035694031e-05,
1390
+ "loss": 0.0539,
1391
+ "step": 796
1392
+ },
1393
+ {
1394
+ "epoch": 14.55,
1395
+ "learning_rate": 1.049857726072005e-05,
1396
+ "loss": 0.0446,
1397
+ "step": 800
1398
+ },
1399
+ {
1400
+ "epoch": 14.62,
1401
+ "learning_rate": 1.024126092299176e-05,
1402
+ "loss": 0.0477,
1403
+ "step": 804
1404
+ },
1405
+ {
1406
+ "epoch": 14.69,
1407
+ "learning_rate": 9.986322480749927e-06,
1408
+ "loss": 0.0529,
1409
+ "step": 808
1410
+ },
1411
+ {
1412
+ "epoch": 14.76,
1413
+ "learning_rate": 9.733803009114045e-06,
1414
+ "loss": 0.0466,
1415
+ "step": 812
1416
+ },
1417
+ {
1418
+ "epoch": 14.84,
1419
+ "learning_rate": 9.483743193464408e-06,
1420
+ "loss": 0.0518,
1421
+ "step": 816
1422
+ },
1423
+ {
1424
+ "epoch": 14.91,
1425
+ "learning_rate": 9.236183322886945e-06,
1426
+ "loss": 0.0479,
1427
+ "step": 820
1428
+ },
1429
+ {
1430
+ "epoch": 14.98,
1431
+ "learning_rate": 8.991163283681944e-06,
1432
+ "loss": 0.0443,
1433
+ "step": 824
1434
+ },
1435
+ {
1436
+ "epoch": 15.0,
1437
+ "gpt4_scores": 0.6666666666666666,
1438
+ "step": 825
1439
+ },
1440
+ {
1441
+ "epoch": 15.0,
1442
+ "eval_loss": 4.091893672943115,
1443
+ "eval_runtime": 4.9566,
1444
+ "eval_samples_per_second": 4.64,
1445
+ "eval_steps_per_second": 1.211,
1446
+ "step": 825
1447
+ },
1448
+ {
1449
+ "epoch": 15.05,
1450
+ "learning_rate": 8.748722552937689e-06,
1451
+ "loss": 0.0471,
1452
+ "step": 828
1453
+ },
1454
+ {
1455
+ "epoch": 15.13,
1456
+ "learning_rate": 8.508900192169964e-06,
1457
+ "loss": 0.0467,
1458
+ "step": 832
1459
+ },
1460
+ {
1461
+ "epoch": 15.2,
1462
+ "learning_rate": 8.271734841028553e-06,
1463
+ "loss": 0.0498,
1464
+ "step": 836
1465
+ },
1466
+ {
1467
+ "epoch": 15.27,
1468
+ "learning_rate": 8.0372647110717e-06,
1469
+ "loss": 0.0476,
1470
+ "step": 840
1471
+ },
1472
+ {
1473
+ "epoch": 15.35,
1474
+ "learning_rate": 7.805527579609576e-06,
1475
+ "loss": 0.0481,
1476
+ "step": 844
1477
+ },
1478
+ {
1479
+ "epoch": 15.42,
1480
+ "learning_rate": 7.576560783617668e-06,
1481
+ "loss": 0.0466,
1482
+ "step": 848
1483
+ },
1484
+ {
1485
+ "epoch": 15.49,
1486
+ "learning_rate": 7.350401213721089e-06,
1487
+ "loss": 0.0455,
1488
+ "step": 852
1489
+ },
1490
+ {
1491
+ "epoch": 15.56,
1492
+ "learning_rate": 7.127085308250914e-06,
1493
+ "loss": 0.048,
1494
+ "step": 856
1495
+ },
1496
+ {
1497
+ "epoch": 15.64,
1498
+ "learning_rate": 6.906649047373246e-06,
1499
+ "loss": 0.0422,
1500
+ "step": 860
1501
+ },
1502
+ {
1503
+ "epoch": 15.71,
1504
+ "learning_rate": 6.689127947292231e-06,
1505
+ "loss": 0.0442,
1506
+ "step": 864
1507
+ },
1508
+ {
1509
+ "epoch": 15.78,
1510
+ "learning_rate": 6.4745570545277075e-06,
1511
+ "loss": 0.0414,
1512
+ "step": 868
1513
+ },
1514
+ {
1515
+ "epoch": 15.85,
1516
+ "learning_rate": 6.2629709402686535e-06,
1517
+ "loss": 0.0455,
1518
+ "step": 872
1519
+ },
1520
+ {
1521
+ "epoch": 15.93,
1522
+ "learning_rate": 6.054403694803079e-06,
1523
+ "loss": 0.0519,
1524
+ "step": 876
1525
+ },
1526
+ {
1527
+ "epoch": 16.0,
1528
+ "learning_rate": 5.848888922025553e-06,
1529
+ "loss": 0.0464,
1530
+ "step": 880
1531
+ },
1532
+ {
1533
+ "epoch": 16.07,
1534
+ "learning_rate": 5.646459734022938e-06,
1535
+ "loss": 0.0443,
1536
+ "step": 884
1537
+ },
1538
+ {
1539
+ "epoch": 16.15,
1540
+ "learning_rate": 5.4471487457395225e-06,
1541
+ "loss": 0.0405,
1542
+ "step": 888
1543
+ },
1544
+ {
1545
+ "epoch": 16.22,
1546
+ "learning_rate": 5.250988069722096e-06,
1547
+ "loss": 0.0441,
1548
+ "step": 892
1549
+ },
1550
+ {
1551
+ "epoch": 16.29,
1552
+ "learning_rate": 5.058009310946119e-06,
1553
+ "loss": 0.0412,
1554
+ "step": 896
1555
+ },
1556
+ {
1557
+ "epoch": 16.36,
1558
+ "learning_rate": 4.868243561723535e-06,
1559
+ "loss": 0.0468,
1560
+ "step": 900
1561
+ },
1562
+ {
1563
+ "epoch": 16.44,
1564
+ "learning_rate": 4.681721396693303e-06,
1565
+ "loss": 0.0476,
1566
+ "step": 904
1567
+ },
1568
+ {
1569
+ "epoch": 16.51,
1570
+ "learning_rate": 4.498472867895223e-06,
1571
+ "loss": 0.0481,
1572
+ "step": 908
1573
+ },
1574
+ {
1575
+ "epoch": 16.58,
1576
+ "learning_rate": 4.318527499928074e-06,
1577
+ "loss": 0.0485,
1578
+ "step": 912
1579
+ },
1580
+ {
1581
+ "epoch": 16.65,
1582
+ "learning_rate": 4.141914285192619e-06,
1583
+ "loss": 0.0441,
1584
+ "step": 916
1585
+ },
1586
+ {
1587
+ "epoch": 16.73,
1588
+ "learning_rate": 3.968661679220468e-06,
1589
+ "loss": 0.0469,
1590
+ "step": 920
1591
+ },
1592
+ {
1593
+ "epoch": 16.8,
1594
+ "learning_rate": 3.798797596089351e-06,
1595
+ "loss": 0.0436,
1596
+ "step": 924
1597
+ },
1598
+ {
1599
+ "epoch": 16.87,
1600
+ "learning_rate": 3.632349403925664e-06,
1601
+ "loss": 0.0528,
1602
+ "step": 928
1603
+ },
1604
+ {
1605
+ "epoch": 16.95,
1606
+ "learning_rate": 3.4693439204949858e-06,
1607
+ "loss": 0.0461,
1608
+ "step": 932
1609
+ },
1610
+ {
1611
+ "epoch": 17.0,
1612
+ "gpt4_scores": 0.4833333333333334,
1613
+ "step": 935
1614
+ },
1615
+ {
1616
+ "epoch": 17.0,
1617
+ "eval_loss": 4.148024082183838,
1618
+ "eval_runtime": 4.957,
1619
+ "eval_samples_per_second": 4.64,
1620
+ "eval_steps_per_second": 1.21,
1621
+ "step": 935
1622
+ },
1623
+ {
1624
+ "epoch": 17.02,
1625
+ "learning_rate": 3.3098074088812686e-06,
1626
+ "loss": 0.0472,
1627
+ "step": 936
1628
+ },
1629
+ {
1630
+ "epoch": 17.09,
1631
+ "learning_rate": 3.1537655732553768e-06,
1632
+ "loss": 0.0422,
1633
+ "step": 940
1634
+ },
1635
+ {
1636
+ "epoch": 17.16,
1637
+ "learning_rate": 3.0012435547336737e-06,
1638
+ "loss": 0.0415,
1639
+ "step": 944
1640
+ },
1641
+ {
1642
+ "epoch": 17.24,
1643
+ "learning_rate": 2.8522659273273604e-06,
1644
+ "loss": 0.0446,
1645
+ "step": 948
1646
+ },
1647
+ {
1648
+ "epoch": 17.31,
1649
+ "learning_rate": 2.7068566939831645e-06,
1650
+ "loss": 0.0477,
1651
+ "step": 952
1652
+ },
1653
+ {
1654
+ "epoch": 17.38,
1655
+ "learning_rate": 2.565039282716045e-06,
1656
+ "loss": 0.0448,
1657
+ "step": 956
1658
+ },
1659
+ {
1660
+ "epoch": 17.45,
1661
+ "learning_rate": 2.4268365428344736e-06,
1662
+ "loss": 0.0408,
1663
+ "step": 960
1664
+ },
1665
+ {
1666
+ "epoch": 17.53,
1667
+ "learning_rate": 2.29227074125907e-06,
1668
+ "loss": 0.0472,
1669
+ "step": 964
1670
+ },
1671
+ {
1672
+ "epoch": 17.6,
1673
+ "learning_rate": 2.1613635589349756e-06,
1674
+ "loss": 0.0485,
1675
+ "step": 968
1676
+ },
1677
+ {
1678
+ "epoch": 17.67,
1679
+ "learning_rate": 2.0341360873386674e-06,
1680
+ "loss": 0.0419,
1681
+ "step": 972
1682
+ },
1683
+ {
1684
+ "epoch": 17.75,
1685
+ "learning_rate": 1.9106088250797267e-06,
1686
+ "loss": 0.0515,
1687
+ "step": 976
1688
+ },
1689
+ {
1690
+ "epoch": 17.82,
1691
+ "learning_rate": 1.790801674598186e-06,
1692
+ "loss": 0.0502,
1693
+ "step": 980
1694
+ },
1695
+ {
1696
+ "epoch": 17.89,
1697
+ "learning_rate": 1.674733938957873e-06,
1698
+ "loss": 0.0422,
1699
+ "step": 984
1700
+ },
1701
+ {
1702
+ "epoch": 17.96,
1703
+ "learning_rate": 1.5624243187363441e-06,
1704
+ "loss": 0.0465,
1705
+ "step": 988
1706
+ },
1707
+ {
1708
+ "epoch": 18.0,
1709
+ "gpt4_scores": 0.4000000000000001,
1710
+ "step": 990
1711
+ },
1712
+ {
1713
+ "epoch": 18.0,
1714
+ "eval_loss": 4.16222620010376,
1715
+ "eval_runtime": 4.9609,
1716
+ "eval_samples_per_second": 4.636,
1717
+ "eval_steps_per_second": 1.209,
1718
+ "step": 990
1719
+ },
1720
+ {
1721
+ "epoch": 18.04,
1722
+ "learning_rate": 1.4538909090118846e-06,
1723
+ "loss": 0.0463,
1724
+ "step": 992
1725
+ },
1726
+ {
1727
+ "epoch": 18.11,
1728
+ "learning_rate": 1.3491511964480702e-06,
1729
+ "loss": 0.0433,
1730
+ "step": 996
1731
+ },
1732
+ {
1733
+ "epoch": 18.18,
1734
+ "learning_rate": 1.248222056476367e-06,
1735
+ "loss": 0.0388,
1736
+ "step": 1000
1737
+ },
1738
+ {
1739
+ "epoch": 18.25,
1740
+ "learning_rate": 1.1511197505771843e-06,
1741
+ "loss": 0.0491,
1742
+ "step": 1004
1743
+ },
1744
+ {
1745
+ "epoch": 18.33,
1746
+ "learning_rate": 1.0578599236598707e-06,
1747
+ "loss": 0.0486,
1748
+ "step": 1008
1749
+ },
1750
+ {
1751
+ "epoch": 18.4,
1752
+ "learning_rate": 9.684576015420278e-07,
1753
+ "loss": 0.0537,
1754
+ "step": 1012
1755
+ },
1756
+ {
1757
+ "epoch": 18.47,
1758
+ "learning_rate": 8.829271885286094e-07,
1759
+ "loss": 0.0464,
1760
+ "step": 1016
1761
+ },
1762
+ {
1763
+ "epoch": 18.55,
1764
+ "learning_rate": 8.012824650910938e-07,
1765
+ "loss": 0.0418,
1766
+ "step": 1020
1767
+ },
1768
+ {
1769
+ "epoch": 18.62,
1770
+ "learning_rate": 7.235365856472442e-07,
1771
+ "loss": 0.0506,
1772
+ "step": 1024
1773
+ },
1774
+ {
1775
+ "epoch": 18.69,
1776
+ "learning_rate": 6.497020764416633e-07,
1777
+ "loss": 0.0398,
1778
+ "step": 1028
1779
+ },
1780
+ {
1781
+ "epoch": 18.76,
1782
+ "learning_rate": 5.797908335276214e-07,
1783
+ "loss": 0.0418,
1784
+ "step": 1032
1785
+ },
1786
+ {
1787
+ "epoch": 18.84,
1788
+ "learning_rate": 5.1381412085037e-07,
1789
+ "loss": 0.041,
1790
+ "step": 1036
1791
+ },
1792
+ {
1793
+ "epoch": 18.91,
1794
+ "learning_rate": 4.517825684323324e-07,
1795
+ "loss": 0.0446,
1796
+ "step": 1040
1797
+ },
1798
+ {
1799
+ "epoch": 18.98,
1800
+ "learning_rate": 3.9370617066040726e-07,
1801
+ "loss": 0.049,
1802
+ "step": 1044
1803
+ },
1804
+ {
1805
+ "epoch": 19.0,
1806
+ "gpt4_scores": 0.5333333333333333,
1807
+ "step": 1045
1808
+ },
1809
+ {
1810
+ "epoch": 19.0,
1811
+ "eval_loss": 4.166710376739502,
1812
+ "eval_runtime": 4.9226,
1813
+ "eval_samples_per_second": 4.672,
1814
+ "eval_steps_per_second": 1.219,
1815
+ "step": 1045
1816
+ },
1817
+ {
1818
+ "epoch": 19.05,
1819
+ "learning_rate": 3.395942846757066e-07,
1820
+ "loss": 0.0449,
1821
+ "step": 1048
1822
+ },
1823
+ {
1824
+ "epoch": 19.13,
1825
+ "learning_rate": 2.894556288659395e-07,
1826
+ "loss": 0.0444,
1827
+ "step": 1052
1828
+ },
1829
+ {
1830
+ "epoch": 19.2,
1831
+ "learning_rate": 2.4329828146074095e-07,
1832
+ "loss": 0.0465,
1833
+ "step": 1056
1834
+ },
1835
+ {
1836
+ "epoch": 19.27,
1837
+ "learning_rate": 2.011296792301165e-07,
1838
+ "loss": 0.0464,
1839
+ "step": 1060
1840
+ },
1841
+ {
1842
+ "epoch": 19.35,
1843
+ "learning_rate": 1.6295661628624447e-07,
1844
+ "loss": 0.0446,
1845
+ "step": 1064
1846
+ },
1847
+ {
1848
+ "epoch": 19.42,
1849
+ "learning_rate": 1.2878524298882698e-07,
1850
+ "loss": 0.0425,
1851
+ "step": 1068
1852
+ },
1853
+ {
1854
+ "epoch": 19.49,
1855
+ "learning_rate": 9.862106495415469e-08,
1856
+ "loss": 0.054,
1857
+ "step": 1072
1858
+ },
1859
+ {
1860
+ "epoch": 19.56,
1861
+ "learning_rate": 7.246894216806355e-08,
1862
+ "loss": 0.0421,
1863
+ "step": 1076
1864
+ },
1865
+ {
1866
+ "epoch": 19.64,
1867
+ "learning_rate": 5.033308820289184e-08,
1868
+ "loss": 0.045,
1869
+ "step": 1080
1870
+ },
1871
+ {
1872
+ "epoch": 19.71,
1873
+ "learning_rate": 3.221706953860093e-08,
1874
+ "loss": 0.0455,
1875
+ "step": 1084
1876
+ },
1877
+ {
1878
+ "epoch": 19.78,
1879
+ "learning_rate": 1.812380498815991e-08,
1880
+ "loss": 0.0411,
1881
+ "step": 1088
1882
+ },
1883
+ {
1884
+ "epoch": 19.85,
1885
+ "learning_rate": 8.055565227271799e-09,
1886
+ "loss": 0.0504,
1887
+ "step": 1092
1888
+ },
1889
+ {
1890
+ "epoch": 19.93,
1891
+ "learning_rate": 2.0139724285161977e-09,
1892
+ "loss": 0.0432,
1893
+ "step": 1096
1894
+ },
1895
+ {
1896
+ "epoch": 20.0,
1897
+ "learning_rate": 0.0,
1898
+ "loss": 0.0416,
1899
+ "step": 1100
1900
+ },
1901
+ {
1902
+ "epoch": 20.0,
1903
+ "gpt4_scores": 0.43333333333333335,
1904
+ "step": 1100
1905
+ },
1906
+ {
1907
+ "epoch": 20.0,
1908
+ "eval_loss": 4.167160511016846,
1909
+ "eval_runtime": 4.9522,
1910
+ "eval_samples_per_second": 4.644,
1911
+ "eval_steps_per_second": 1.212,
1912
+ "step": 1100
1913
+ },
1914
+ {
1915
+ "epoch": 20.0,
1916
+ "step": 1100,
1917
+ "total_flos": 3.795619210798694e+16,
1918
+ "train_loss": 0.009085019393400713,
1919
+ "train_runtime": 1593.5865,
1920
+ "train_samples_per_second": 2.723,
1921
+ "train_steps_per_second": 0.69
1922
  }
1923
  ],
1924
  "logging_steps": 4,
1925
+ "max_steps": 1100,
1926
  "num_input_tokens_seen": 0,
1927
+ "num_train_epochs": 20,
1928
+ "save_steps": 55,
1929
+ "total_flos": 3.795619210798694e+16,
1930
  "train_batch_size": 4,
1931
  "trial_name": null,
1932
  "trial_params": null