yyx123 commited on
Commit
7cc8af0
1 Parent(s): 53cdbc3

Model save

Browse files
README.md ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - sft
7
+ - generated_from_trainer
8
+ datasets:
9
+ - generator
10
+ base_model: 01-ai/Yi-6B
11
+ model-index:
12
+ - name: Yi-6B-zhihu4
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # Yi-6B-zhihu4
20
+
21
+ This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 2.4907
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 0.0002
43
+ - train_batch_size: 1
44
+ - eval_batch_size: 1
45
+ - seed: 42
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: cosine
48
+ - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 1
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss |
54
+ |:-------------:|:-----:|:----:|:---------------:|
55
+ | 2.744 | 1.0 | 766 | 2.4907 |
56
+
57
+
58
+ ### Framework versions
59
+
60
+ - PEFT 0.7.1
61
+ - Transformers 4.36.2
62
+ - Pytorch 2.2.2+cu118
63
+ - Datasets 2.14.6
64
+ - Tokenizers 0.15.2
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bfd8e88e095006cffad4bdb81d8786fd177e70d0c72fb058b59598b44acefa63
3
  size 72673912
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fda17e3dc13c848b3b167f7289485a733fc13a391eb8ea68ebadc4e8d2fccd4
3
  size 72673912
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_loss": 2.4906539916992188,
4
+ "eval_runtime": 232.3523,
5
+ "eval_samples": 2561,
6
+ "eval_samples_per_second": 3.297,
7
+ "eval_steps_per_second": 3.297,
8
+ "train_loss": 2.5803672159309485,
9
+ "train_runtime": 1034.6215,
10
+ "train_samples": 2561,
11
+ "train_samples_per_second": 0.74,
12
+ "train_steps_per_second": 0.74
13
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_loss": 2.4906539916992188,
4
+ "eval_runtime": 232.3523,
5
+ "eval_samples": 2561,
6
+ "eval_samples_per_second": 3.297,
7
+ "eval_steps_per_second": 3.297
8
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "train_loss": 2.5803672159309485,
4
+ "train_runtime": 1034.6215,
5
+ "train_samples": 2561,
6
+ "train_samples_per_second": 0.74,
7
+ "train_steps_per_second": 0.74
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,962 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.0,
5
+ "eval_steps": 500,
6
+ "global_step": 766,
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.0,
13
+ "learning_rate": 2.5974025974025976e-06,
14
+ "loss": 2.4814,
15
+ "step": 1
16
+ },
17
+ {
18
+ "epoch": 0.01,
19
+ "learning_rate": 1.2987012987012986e-05,
20
+ "loss": 2.2941,
21
+ "step": 5
22
+ },
23
+ {
24
+ "epoch": 0.01,
25
+ "learning_rate": 2.5974025974025972e-05,
26
+ "loss": 2.4997,
27
+ "step": 10
28
+ },
29
+ {
30
+ "epoch": 0.02,
31
+ "learning_rate": 3.8961038961038966e-05,
32
+ "loss": 2.7226,
33
+ "step": 15
34
+ },
35
+ {
36
+ "epoch": 0.03,
37
+ "learning_rate": 5.1948051948051944e-05,
38
+ "loss": 2.8458,
39
+ "step": 20
40
+ },
41
+ {
42
+ "epoch": 0.03,
43
+ "learning_rate": 6.493506493506494e-05,
44
+ "loss": 2.6056,
45
+ "step": 25
46
+ },
47
+ {
48
+ "epoch": 0.04,
49
+ "learning_rate": 7.792207792207793e-05,
50
+ "loss": 2.6706,
51
+ "step": 30
52
+ },
53
+ {
54
+ "epoch": 0.05,
55
+ "learning_rate": 9.090909090909092e-05,
56
+ "loss": 2.5805,
57
+ "step": 35
58
+ },
59
+ {
60
+ "epoch": 0.05,
61
+ "learning_rate": 0.00010389610389610389,
62
+ "loss": 2.5799,
63
+ "step": 40
64
+ },
65
+ {
66
+ "epoch": 0.06,
67
+ "learning_rate": 0.00011688311688311689,
68
+ "loss": 2.7861,
69
+ "step": 45
70
+ },
71
+ {
72
+ "epoch": 0.07,
73
+ "learning_rate": 0.00012987012987012987,
74
+ "loss": 2.4062,
75
+ "step": 50
76
+ },
77
+ {
78
+ "epoch": 0.07,
79
+ "learning_rate": 0.00014285714285714287,
80
+ "loss": 2.7231,
81
+ "step": 55
82
+ },
83
+ {
84
+ "epoch": 0.08,
85
+ "learning_rate": 0.00015584415584415587,
86
+ "loss": 2.8154,
87
+ "step": 60
88
+ },
89
+ {
90
+ "epoch": 0.08,
91
+ "learning_rate": 0.00016883116883116884,
92
+ "loss": 2.785,
93
+ "step": 65
94
+ },
95
+ {
96
+ "epoch": 0.09,
97
+ "learning_rate": 0.00018181818181818183,
98
+ "loss": 2.393,
99
+ "step": 70
100
+ },
101
+ {
102
+ "epoch": 0.1,
103
+ "learning_rate": 0.0001948051948051948,
104
+ "loss": 2.6932,
105
+ "step": 75
106
+ },
107
+ {
108
+ "epoch": 0.1,
109
+ "learning_rate": 0.00019999064449912997,
110
+ "loss": 2.7158,
111
+ "step": 80
112
+ },
113
+ {
114
+ "epoch": 0.11,
115
+ "learning_rate": 0.00019993347833297808,
116
+ "loss": 2.4839,
117
+ "step": 85
118
+ },
119
+ {
120
+ "epoch": 0.12,
121
+ "learning_rate": 0.00019982437317643217,
122
+ "loss": 2.8402,
123
+ "step": 90
124
+ },
125
+ {
126
+ "epoch": 0.12,
127
+ "learning_rate": 0.00019966338573533063,
128
+ "loss": 2.5689,
129
+ "step": 95
130
+ },
131
+ {
132
+ "epoch": 0.13,
133
+ "learning_rate": 0.0001994505996805844,
134
+ "loss": 2.4474,
135
+ "step": 100
136
+ },
137
+ {
138
+ "epoch": 0.14,
139
+ "learning_rate": 0.0001991861256046899,
140
+ "loss": 2.7085,
141
+ "step": 105
142
+ },
143
+ {
144
+ "epoch": 0.14,
145
+ "learning_rate": 0.00019887010096425055,
146
+ "loss": 2.6015,
147
+ "step": 110
148
+ },
149
+ {
150
+ "epoch": 0.15,
151
+ "learning_rate": 0.0001985026900085352,
152
+ "loss": 2.6907,
153
+ "step": 115
154
+ },
155
+ {
156
+ "epoch": 0.16,
157
+ "learning_rate": 0.00019808408369411241,
158
+ "loss": 2.6831,
159
+ "step": 120
160
+ },
161
+ {
162
+ "epoch": 0.16,
163
+ "learning_rate": 0.00019761449958560326,
164
+ "loss": 2.1675,
165
+ "step": 125
166
+ },
167
+ {
168
+ "epoch": 0.17,
169
+ "learning_rate": 0.0001970941817426052,
170
+ "loss": 2.8732,
171
+ "step": 130
172
+ },
173
+ {
174
+ "epoch": 0.18,
175
+ "learning_rate": 0.00019652340059284583,
176
+ "loss": 2.8546,
177
+ "step": 135
178
+ },
179
+ {
180
+ "epoch": 0.18,
181
+ "learning_rate": 0.00019590245279163169,
182
+ "loss": 2.7912,
183
+ "step": 140
184
+ },
185
+ {
186
+ "epoch": 0.19,
187
+ "learning_rate": 0.00019523166106766616,
188
+ "loss": 2.7565,
189
+ "step": 145
190
+ },
191
+ {
192
+ "epoch": 0.2,
193
+ "learning_rate": 0.00019451137405531563,
194
+ "loss": 2.8615,
195
+ "step": 150
196
+ },
197
+ {
198
+ "epoch": 0.2,
199
+ "learning_rate": 0.0001937419661134121,
200
+ "loss": 2.6343,
201
+ "step": 155
202
+ },
203
+ {
204
+ "epoch": 0.21,
205
+ "learning_rate": 0.00019292383713068518,
206
+ "loss": 2.446,
207
+ "step": 160
208
+ },
209
+ {
210
+ "epoch": 0.22,
211
+ "learning_rate": 0.00019205741231792596,
212
+ "loss": 2.9245,
213
+ "step": 165
214
+ },
215
+ {
216
+ "epoch": 0.22,
217
+ "learning_rate": 0.00019114314198698967,
218
+ "loss": 2.5026,
219
+ "step": 170
220
+ },
221
+ {
222
+ "epoch": 0.23,
223
+ "learning_rate": 0.00019018150131675253,
224
+ "loss": 2.842,
225
+ "step": 175
226
+ },
227
+ {
228
+ "epoch": 0.23,
229
+ "learning_rate": 0.0001891729901061445,
230
+ "loss": 2.5944,
231
+ "step": 180
232
+ },
233
+ {
234
+ "epoch": 0.24,
235
+ "learning_rate": 0.0001881181325143858,
236
+ "loss": 2.767,
237
+ "step": 185
238
+ },
239
+ {
240
+ "epoch": 0.25,
241
+ "learning_rate": 0.00018701747678856286,
242
+ "loss": 2.4217,
243
+ "step": 190
244
+ },
245
+ {
246
+ "epoch": 0.25,
247
+ "learning_rate": 0.00018587159497868448,
248
+ "loss": 2.5336,
249
+ "step": 195
250
+ },
251
+ {
252
+ "epoch": 0.26,
253
+ "learning_rate": 0.00018468108264036747,
254
+ "loss": 2.6071,
255
+ "step": 200
256
+ },
257
+ {
258
+ "epoch": 0.27,
259
+ "learning_rate": 0.00018344655852530468,
260
+ "loss": 2.6432,
261
+ "step": 205
262
+ },
263
+ {
264
+ "epoch": 0.27,
265
+ "learning_rate": 0.00018216866425967818,
266
+ "loss": 2.526,
267
+ "step": 210
268
+ },
269
+ {
270
+ "epoch": 0.28,
271
+ "learning_rate": 0.00018084806401068285,
272
+ "loss": 2.6892,
273
+ "step": 215
274
+ },
275
+ {
276
+ "epoch": 0.29,
277
+ "learning_rate": 0.00017948544414133534,
278
+ "loss": 2.9056,
279
+ "step": 220
280
+ },
281
+ {
282
+ "epoch": 0.29,
283
+ "learning_rate": 0.00017808151285374652,
284
+ "loss": 2.784,
285
+ "step": 225
286
+ },
287
+ {
288
+ "epoch": 0.3,
289
+ "learning_rate": 0.00017663699982104373,
290
+ "loss": 2.2989,
291
+ "step": 230
292
+ },
293
+ {
294
+ "epoch": 0.31,
295
+ "learning_rate": 0.00017515265580813355,
296
+ "loss": 2.4406,
297
+ "step": 235
298
+ },
299
+ {
300
+ "epoch": 0.31,
301
+ "learning_rate": 0.0001736292522815027,
302
+ "loss": 2.3931,
303
+ "step": 240
304
+ },
305
+ {
306
+ "epoch": 0.32,
307
+ "learning_rate": 0.00017206758100825917,
308
+ "loss": 2.6808,
309
+ "step": 245
310
+ },
311
+ {
312
+ "epoch": 0.33,
313
+ "learning_rate": 0.00017046845364462285,
314
+ "loss": 2.3481,
315
+ "step": 250
316
+ },
317
+ {
318
+ "epoch": 0.33,
319
+ "learning_rate": 0.00016883270131407864,
320
+ "loss": 2.704,
321
+ "step": 255
322
+ },
323
+ {
324
+ "epoch": 0.34,
325
+ "learning_rate": 0.00016716117417541227,
326
+ "loss": 2.6067,
327
+ "step": 260
328
+ },
329
+ {
330
+ "epoch": 0.35,
331
+ "learning_rate": 0.0001654547409808521,
332
+ "loss": 2.4372,
333
+ "step": 265
334
+ },
335
+ {
336
+ "epoch": 0.35,
337
+ "learning_rate": 0.000163714288624548,
338
+ "loss": 2.3988,
339
+ "step": 270
340
+ },
341
+ {
342
+ "epoch": 0.36,
343
+ "learning_rate": 0.00016194072168162048,
344
+ "loss": 2.43,
345
+ "step": 275
346
+ },
347
+ {
348
+ "epoch": 0.37,
349
+ "learning_rate": 0.0001601349619380211,
350
+ "loss": 2.948,
351
+ "step": 280
352
+ },
353
+ {
354
+ "epoch": 0.37,
355
+ "learning_rate": 0.0001582979479114472,
356
+ "loss": 2.5301,
357
+ "step": 285
358
+ },
359
+ {
360
+ "epoch": 0.38,
361
+ "learning_rate": 0.00015643063436356102,
362
+ "loss": 2.6635,
363
+ "step": 290
364
+ },
365
+ {
366
+ "epoch": 0.39,
367
+ "learning_rate": 0.0001545339918037658,
368
+ "loss": 2.4107,
369
+ "step": 295
370
+ },
371
+ {
372
+ "epoch": 0.39,
373
+ "learning_rate": 0.00015260900598479765,
374
+ "loss": 2.4512,
375
+ "step": 300
376
+ },
377
+ {
378
+ "epoch": 0.4,
379
+ "learning_rate": 0.0001506566773903946,
380
+ "loss": 2.661,
381
+ "step": 305
382
+ },
383
+ {
384
+ "epoch": 0.4,
385
+ "learning_rate": 0.0001486780207153097,
386
+ "loss": 2.5907,
387
+ "step": 310
388
+ },
389
+ {
390
+ "epoch": 0.41,
391
+ "learning_rate": 0.00014667406433793774,
392
+ "loss": 2.589,
393
+ "step": 315
394
+ },
395
+ {
396
+ "epoch": 0.42,
397
+ "learning_rate": 0.00014464584978583057,
398
+ "loss": 2.7578,
399
+ "step": 320
400
+ },
401
+ {
402
+ "epoch": 0.42,
403
+ "learning_rate": 0.000142594431194378,
404
+ "loss": 2.4308,
405
+ "step": 325
406
+ },
407
+ {
408
+ "epoch": 0.43,
409
+ "learning_rate": 0.00014052087475893615,
410
+ "loss": 2.3724,
411
+ "step": 330
412
+ },
413
+ {
414
+ "epoch": 0.44,
415
+ "learning_rate": 0.00013842625818068758,
416
+ "loss": 2.6396,
417
+ "step": 335
418
+ },
419
+ {
420
+ "epoch": 0.44,
421
+ "learning_rate": 0.00013631167010652178,
422
+ "loss": 2.6514,
423
+ "step": 340
424
+ },
425
+ {
426
+ "epoch": 0.45,
427
+ "learning_rate": 0.00013417820956322633,
428
+ "loss": 2.5516,
429
+ "step": 345
430
+ },
431
+ {
432
+ "epoch": 0.46,
433
+ "learning_rate": 0.00013202698538628376,
434
+ "loss": 2.7136,
435
+ "step": 350
436
+ },
437
+ {
438
+ "epoch": 0.46,
439
+ "learning_rate": 0.0001298591156435701,
440
+ "loss": 2.7312,
441
+ "step": 355
442
+ },
443
+ {
444
+ "epoch": 0.47,
445
+ "learning_rate": 0.00012767572705425513,
446
+ "loss": 2.3132,
447
+ "step": 360
448
+ },
449
+ {
450
+ "epoch": 0.48,
451
+ "learning_rate": 0.00012547795440320626,
452
+ "loss": 2.4825,
453
+ "step": 365
454
+ },
455
+ {
456
+ "epoch": 0.48,
457
+ "learning_rate": 0.00012326693995120064,
458
+ "loss": 2.6723,
459
+ "step": 370
460
+ },
461
+ {
462
+ "epoch": 0.49,
463
+ "learning_rate": 0.0001210438328412511,
464
+ "loss": 2.5739,
465
+ "step": 375
466
+ },
467
+ {
468
+ "epoch": 0.5,
469
+ "learning_rate": 0.00011880978850135593,
470
+ "loss": 2.5385,
471
+ "step": 380
472
+ },
473
+ {
474
+ "epoch": 0.5,
475
+ "learning_rate": 0.00011656596804398158,
476
+ "loss": 2.5449,
477
+ "step": 385
478
+ },
479
+ {
480
+ "epoch": 0.51,
481
+ "learning_rate": 0.00011431353766259121,
482
+ "loss": 2.2918,
483
+ "step": 390
484
+ },
485
+ {
486
+ "epoch": 0.52,
487
+ "learning_rate": 0.0001120536680255323,
488
+ "loss": 2.4322,
489
+ "step": 395
490
+ },
491
+ {
492
+ "epoch": 0.52,
493
+ "learning_rate": 0.00010978753366759877,
494
+ "loss": 2.6421,
495
+ "step": 400
496
+ },
497
+ {
498
+ "epoch": 0.53,
499
+ "learning_rate": 0.00010751631237958309,
500
+ "loss": 2.4996,
501
+ "step": 405
502
+ },
503
+ {
504
+ "epoch": 0.54,
505
+ "learning_rate": 0.00010524118459613672,
506
+ "loss": 2.3949,
507
+ "step": 410
508
+ },
509
+ {
510
+ "epoch": 0.54,
511
+ "learning_rate": 0.00010296333278225599,
512
+ "loss": 2.384,
513
+ "step": 415
514
+ },
515
+ {
516
+ "epoch": 0.55,
517
+ "learning_rate": 0.00010068394081871289,
518
+ "loss": 2.4835,
519
+ "step": 420
520
+ },
521
+ {
522
+ "epoch": 0.55,
523
+ "learning_rate": 9.840419338675015e-05,
524
+ "loss": 2.6381,
525
+ "step": 425
526
+ },
527
+ {
528
+ "epoch": 0.56,
529
+ "learning_rate": 9.612527535236006e-05,
530
+ "loss": 2.5099,
531
+ "step": 430
532
+ },
533
+ {
534
+ "epoch": 0.57,
535
+ "learning_rate": 9.384837115046746e-05,
536
+ "loss": 2.5638,
537
+ "step": 435
538
+ },
539
+ {
540
+ "epoch": 0.57,
541
+ "learning_rate": 9.157466416933662e-05,
542
+ "loss": 2.6704,
543
+ "step": 440
544
+ },
545
+ {
546
+ "epoch": 0.58,
547
+ "learning_rate": 8.930533613552232e-05,
548
+ "loss": 2.3547,
549
+ "step": 445
550
+ },
551
+ {
552
+ "epoch": 0.59,
553
+ "learning_rate": 8.704156649968422e-05,
554
+ "loss": 2.2018,
555
+ "step": 450
556
+ },
557
+ {
558
+ "epoch": 0.59,
559
+ "learning_rate": 8.478453182358461e-05,
560
+ "loss": 2.3659,
561
+ "step": 455
562
+ },
563
+ {
564
+ "epoch": 0.6,
565
+ "learning_rate": 8.253540516858695e-05,
566
+ "loss": 2.8535,
567
+ "step": 460
568
+ },
569
+ {
570
+ "epoch": 0.61,
571
+ "learning_rate": 8.029535548597451e-05,
572
+ "loss": 2.6786,
573
+ "step": 465
574
+ },
575
+ {
576
+ "epoch": 0.61,
577
+ "learning_rate": 7.806554700940441e-05,
578
+ "loss": 2.7602,
579
+ "step": 470
580
+ },
581
+ {
582
+ "epoch": 0.62,
583
+ "learning_rate": 7.584713864981387e-05,
584
+ "loss": 2.6715,
585
+ "step": 475
586
+ },
587
+ {
588
+ "epoch": 0.63,
589
+ "learning_rate": 7.364128339309326e-05,
590
+ "loss": 2.5784,
591
+ "step": 480
592
+ },
593
+ {
594
+ "epoch": 0.63,
595
+ "learning_rate": 7.14491277008377e-05,
596
+ "loss": 2.4982,
597
+ "step": 485
598
+ },
599
+ {
600
+ "epoch": 0.64,
601
+ "learning_rate": 6.927181091449061e-05,
602
+ "loss": 2.2037,
603
+ "step": 490
604
+ },
605
+ {
606
+ "epoch": 0.65,
607
+ "learning_rate": 6.711046466318702e-05,
608
+ "loss": 2.6169,
609
+ "step": 495
610
+ },
611
+ {
612
+ "epoch": 0.65,
613
+ "learning_rate": 6.496621227560581e-05,
614
+ "loss": 2.7077,
615
+ "step": 500
616
+ },
617
+ {
618
+ "epoch": 0.66,
619
+ "learning_rate": 6.28401681961355e-05,
620
+ "loss": 2.7811,
621
+ "step": 505
622
+ },
623
+ {
624
+ "epoch": 0.67,
625
+ "learning_rate": 6.073343740565807e-05,
626
+ "loss": 2.3697,
627
+ "step": 510
628
+ },
629
+ {
630
+ "epoch": 0.67,
631
+ "learning_rate": 5.86471148472507e-05,
632
+ "loss": 2.6585,
633
+ "step": 515
634
+ },
635
+ {
636
+ "epoch": 0.68,
637
+ "learning_rate": 5.658228485710517e-05,
638
+ "loss": 2.5007,
639
+ "step": 520
640
+ },
641
+ {
642
+ "epoch": 0.69,
643
+ "learning_rate": 5.45400206009595e-05,
644
+ "loss": 2.6819,
645
+ "step": 525
646
+ },
647
+ {
648
+ "epoch": 0.69,
649
+ "learning_rate": 5.2521383516335376e-05,
650
+ "loss": 2.6446,
651
+ "step": 530
652
+ },
653
+ {
654
+ "epoch": 0.7,
655
+ "learning_rate": 5.0527422760871435e-05,
656
+ "loss": 2.5785,
657
+ "step": 535
658
+ },
659
+ {
660
+ "epoch": 0.7,
661
+ "learning_rate": 4.855917466703853e-05,
662
+ "loss": 2.5346,
663
+ "step": 540
664
+ },
665
+ {
666
+ "epoch": 0.71,
667
+ "learning_rate": 4.661766220352097e-05,
668
+ "loss": 2.6784,
669
+ "step": 545
670
+ },
671
+ {
672
+ "epoch": 0.72,
673
+ "learning_rate": 4.470389444354286e-05,
674
+ "loss": 2.6187,
675
+ "step": 550
676
+ },
677
+ {
678
+ "epoch": 0.72,
679
+ "learning_rate": 4.281886604041729e-05,
680
+ "loss": 2.3566,
681
+ "step": 555
682
+ },
683
+ {
684
+ "epoch": 0.73,
685
+ "learning_rate": 4.096355671058918e-05,
686
+ "loss": 2.5479,
687
+ "step": 560
688
+ },
689
+ {
690
+ "epoch": 0.74,
691
+ "learning_rate": 3.9138930724441816e-05,
692
+ "loss": 2.5891,
693
+ "step": 565
694
+ },
695
+ {
696
+ "epoch": 0.74,
697
+ "learning_rate": 3.73459364051313e-05,
698
+ "loss": 2.6546,
699
+ "step": 570
700
+ },
701
+ {
702
+ "epoch": 0.75,
703
+ "learning_rate": 3.558550563570903e-05,
704
+ "loss": 2.5385,
705
+ "step": 575
706
+ },
707
+ {
708
+ "epoch": 0.76,
709
+ "learning_rate": 3.385855337478894e-05,
710
+ "loss": 2.4146,
711
+ "step": 580
712
+ },
713
+ {
714
+ "epoch": 0.76,
715
+ "learning_rate": 3.2165977181010874e-05,
716
+ "loss": 2.5971,
717
+ "step": 585
718
+ },
719
+ {
720
+ "epoch": 0.77,
721
+ "learning_rate": 3.0508656746547505e-05,
722
+ "loss": 2.5462,
723
+ "step": 590
724
+ },
725
+ {
726
+ "epoch": 0.78,
727
+ "learning_rate": 2.8887453439896728e-05,
728
+ "loss": 2.8763,
729
+ "step": 595
730
+ },
731
+ {
732
+ "epoch": 0.78,
733
+ "learning_rate": 2.730320985819791e-05,
734
+ "loss": 2.7621,
735
+ "step": 600
736
+ },
737
+ {
738
+ "epoch": 0.79,
739
+ "learning_rate": 2.575674938930408e-05,
740
+ "loss": 2.1617,
741
+ "step": 605
742
+ },
743
+ {
744
+ "epoch": 0.8,
745
+ "learning_rate": 2.4248875783837987e-05,
746
+ "loss": 2.3758,
747
+ "step": 610
748
+ },
749
+ {
750
+ "epoch": 0.8,
751
+ "learning_rate": 2.2780372737454124e-05,
752
+ "loss": 2.4131,
753
+ "step": 615
754
+ },
755
+ {
756
+ "epoch": 0.81,
757
+ "learning_rate": 2.135200348352423e-05,
758
+ "loss": 2.5124,
759
+ "step": 620
760
+ },
761
+ {
762
+ "epoch": 0.82,
763
+ "learning_rate": 1.9964510396457748e-05,
764
+ "loss": 2.4138,
765
+ "step": 625
766
+ },
767
+ {
768
+ "epoch": 0.82,
769
+ "learning_rate": 1.8618614605863284e-05,
770
+ "loss": 2.8193,
771
+ "step": 630
772
+ },
773
+ {
774
+ "epoch": 0.83,
775
+ "learning_rate": 1.7315015621752006e-05,
776
+ "loss": 1.9123,
777
+ "step": 635
778
+ },
779
+ {
780
+ "epoch": 0.84,
781
+ "learning_rate": 1.6054390970977174e-05,
782
+ "loss": 2.5428,
783
+ "step": 640
784
+ },
785
+ {
786
+ "epoch": 0.84,
787
+ "learning_rate": 1.4837395845099423e-05,
788
+ "loss": 2.9045,
789
+ "step": 645
790
+ },
791
+ {
792
+ "epoch": 0.85,
793
+ "learning_rate": 1.3664662759860258e-05,
794
+ "loss": 2.62,
795
+ "step": 650
796
+ },
797
+ {
798
+ "epoch": 0.86,
799
+ "learning_rate": 1.2536801226441208e-05,
800
+ "loss": 2.2114,
801
+ "step": 655
802
+ },
803
+ {
804
+ "epoch": 0.86,
805
+ "learning_rate": 1.1454397434679021e-05,
806
+ "loss": 2.5337,
807
+ "step": 660
808
+ },
809
+ {
810
+ "epoch": 0.87,
811
+ "learning_rate": 1.0418013948401917e-05,
812
+ "loss": 2.5415,
813
+ "step": 665
814
+ },
815
+ {
816
+ "epoch": 0.87,
817
+ "learning_rate": 9.428189413045285e-06,
818
+ "loss": 2.683,
819
+ "step": 670
820
+ },
821
+ {
822
+ "epoch": 0.88,
823
+ "learning_rate": 8.485438275698154e-06,
824
+ "loss": 2.691,
825
+ "step": 675
826
+ },
827
+ {
828
+ "epoch": 0.89,
829
+ "learning_rate": 7.590250517727016e-06,
830
+ "loss": 2.4456,
831
+ "step": 680
832
+ },
833
+ {
834
+ "epoch": 0.89,
835
+ "learning_rate": 6.7430914001148424e-06,
836
+ "loss": 2.5518,
837
+ "step": 685
838
+ },
839
+ {
840
+ "epoch": 0.9,
841
+ "learning_rate": 5.944401221648432e-06,
842
+ "loss": 2.7904,
843
+ "step": 690
844
+ },
845
+ {
846
+ "epoch": 0.91,
847
+ "learning_rate": 5.194595090079457e-06,
848
+ "loss": 2.4716,
849
+ "step": 695
850
+ },
851
+ {
852
+ "epoch": 0.91,
853
+ "learning_rate": 4.494062706378166e-06,
854
+ "loss": 2.4181,
855
+ "step": 700
856
+ },
857
+ {
858
+ "epoch": 0.92,
859
+ "learning_rate": 3.843168162191857e-06,
860
+ "loss": 2.5981,
861
+ "step": 705
862
+ },
863
+ {
864
+ "epoch": 0.93,
865
+ "learning_rate": 3.242249750613502e-06,
866
+ "loss": 2.4747,
867
+ "step": 710
868
+ },
869
+ {
870
+ "epoch": 0.93,
871
+ "learning_rate": 2.6916197903588127e-06,
872
+ "loss": 2.4181,
873
+ "step": 715
874
+ },
875
+ {
876
+ "epoch": 0.94,
877
+ "learning_rate": 2.1915644634429854e-06,
878
+ "loss": 2.7945,
879
+ "step": 720
880
+ },
881
+ {
882
+ "epoch": 0.95,
883
+ "learning_rate": 1.7423436664417593e-06,
884
+ "loss": 2.5029,
885
+ "step": 725
886
+ },
887
+ {
888
+ "epoch": 0.95,
889
+ "learning_rate": 1.3441908754139198e-06,
890
+ "loss": 2.335,
891
+ "step": 730
892
+ },
893
+ {
894
+ "epoch": 0.96,
895
+ "learning_rate": 9.973130245554018e-07,
896
+ "loss": 2.6024,
897
+ "step": 735
898
+ },
899
+ {
900
+ "epoch": 0.97,
901
+ "learning_rate": 7.018903986483083e-07,
902
+ "loss": 2.6012,
903
+ "step": 740
904
+ },
905
+ {
906
+ "epoch": 0.97,
907
+ "learning_rate": 4.58076539360397e-07,
908
+ "loss": 2.5945,
909
+ "step": 745
910
+ },
911
+ {
912
+ "epoch": 0.98,
913
+ "learning_rate": 2.659981654440591e-07,
914
+ "loss": 2.7643,
915
+ "step": 750
916
+ },
917
+ {
918
+ "epoch": 0.99,
919
+ "learning_rate": 1.2575510687601322e-07,
920
+ "loss": 2.7358,
921
+ "step": 755
922
+ },
923
+ {
924
+ "epoch": 0.99,
925
+ "learning_rate": 3.742025297219787e-08,
926
+ "loss": 2.778,
927
+ "step": 760
928
+ },
929
+ {
930
+ "epoch": 1.0,
931
+ "learning_rate": 1.039514504497241e-09,
932
+ "loss": 2.744,
933
+ "step": 765
934
+ },
935
+ {
936
+ "epoch": 1.0,
937
+ "eval_loss": 2.4906539916992188,
938
+ "eval_runtime": 232.4011,
939
+ "eval_samples_per_second": 3.296,
940
+ "eval_steps_per_second": 3.296,
941
+ "step": 766
942
+ },
943
+ {
944
+ "epoch": 1.0,
945
+ "step": 766,
946
+ "total_flos": 5.492443574383411e+16,
947
+ "train_loss": 2.5803672159309485,
948
+ "train_runtime": 1034.6215,
949
+ "train_samples_per_second": 0.74,
950
+ "train_steps_per_second": 0.74
951
+ }
952
+ ],
953
+ "logging_steps": 5,
954
+ "max_steps": 766,
955
+ "num_input_tokens_seen": 0,
956
+ "num_train_epochs": 1,
957
+ "save_steps": 100,
958
+ "total_flos": 5.492443574383411e+16,
959
+ "train_batch_size": 1,
960
+ "trial_name": null,
961
+ "trial_params": null
962
+ }