tyzhu commited on
Commit
c3e4e4b
1 Parent(s): d70f3dc

End of training

Browse files
Files changed (5) hide show
  1. README.md +14 -2
  2. all_results.json +16 -0
  3. eval_results.json +10 -0
  4. train_results.json +9 -0
  5. trainer_state.json +945 -0
README.md CHANGED
@@ -3,11 +3,23 @@ license: other
3
  base_model: Qwen/Qwen1.5-4B
4
  tags:
5
  - generated_from_trainer
 
 
6
  metrics:
7
  - accuracy
8
  model-index:
9
  - name: lmind_nq_train6000_eval6489_v1_recite_qa_v3_Qwen_Qwen1.5-4B_3e-5_lora2
10
- results: []
 
 
 
 
 
 
 
 
 
 
11
  library_name: peft
12
  ---
13
 
@@ -16,7 +28,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # lmind_nq_train6000_eval6489_v1_recite_qa_v3_Qwen_Qwen1.5-4B_3e-5_lora2
18
 
19
- This model is a fine-tuned version of [Qwen/Qwen1.5-4B](https://huggingface.co/Qwen/Qwen1.5-4B) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
  - Loss: 0.9359
22
  - Accuracy: 0.7135
 
3
  base_model: Qwen/Qwen1.5-4B
4
  tags:
5
  - generated_from_trainer
6
+ datasets:
7
+ - tyzhu/lmind_nq_train6000_eval6489_v1_recite_qa_v3
8
  metrics:
9
  - accuracy
10
  model-index:
11
  - name: lmind_nq_train6000_eval6489_v1_recite_qa_v3_Qwen_Qwen1.5-4B_3e-5_lora2
12
+ results:
13
+ - task:
14
+ name: Causal Language Modeling
15
+ type: text-generation
16
+ dataset:
17
+ name: tyzhu/lmind_nq_train6000_eval6489_v1_recite_qa_v3
18
+ type: tyzhu/lmind_nq_train6000_eval6489_v1_recite_qa_v3
19
+ metrics:
20
+ - name: Accuracy
21
+ type: accuracy
22
+ value: 0.7134618834080717
23
  library_name: peft
24
  ---
25
 
 
28
 
29
  # lmind_nq_train6000_eval6489_v1_recite_qa_v3_Qwen_Qwen1.5-4B_3e-5_lora2
30
 
31
+ This model is a fine-tuned version of [Qwen/Qwen1.5-4B](https://huggingface.co/Qwen/Qwen1.5-4B) on the tyzhu/lmind_nq_train6000_eval6489_v1_recite_qa_v3 dataset.
32
  It achieves the following results on the evaluation set:
33
  - Loss: 0.9359
34
  - Accuracy: 0.7135
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "eval_accuracy": 0.7134618834080717,
4
+ "eval_loss": 0.9358564019203186,
5
+ "eval_runtime": 6.0586,
6
+ "eval_samples": 500,
7
+ "eval_samples_per_second": 82.527,
8
+ "eval_steps_per_second": 10.398,
9
+ "perplexity": 2.549395830596446,
10
+ "total_flos": 9.64242245391745e+17,
11
+ "train_loss": 1.2100651977192927,
12
+ "train_runtime": 22841.5406,
13
+ "train_samples": 16925,
14
+ "train_samples_per_second": 14.819,
15
+ "train_steps_per_second": 0.463
16
+ }
eval_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "eval_accuracy": 0.7134618834080717,
4
+ "eval_loss": 0.9358564019203186,
5
+ "eval_runtime": 6.0586,
6
+ "eval_samples": 500,
7
+ "eval_samples_per_second": 82.527,
8
+ "eval_steps_per_second": 10.398,
9
+ "perplexity": 2.549395830596446
10
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "total_flos": 9.64242245391745e+17,
4
+ "train_loss": 1.2100651977192927,
5
+ "train_runtime": 22841.5406,
6
+ "train_samples": 16925,
7
+ "train_samples_per_second": 14.819,
8
+ "train_steps_per_second": 0.463
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,945 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 20.0,
5
+ "eval_steps": 500,
6
+ "global_step": 10580,
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.1890359168241966,
13
+ "grad_norm": 0.24623066186904907,
14
+ "learning_rate": 3e-05,
15
+ "loss": 2.0175,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.3780718336483932,
20
+ "grad_norm": 0.3154822587966919,
21
+ "learning_rate": 3e-05,
22
+ "loss": 1.9025,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.5671077504725898,
27
+ "grad_norm": 0.2886754274368286,
28
+ "learning_rate": 3e-05,
29
+ "loss": 1.8754,
30
+ "step": 300
31
+ },
32
+ {
33
+ "epoch": 0.7561436672967864,
34
+ "grad_norm": 0.33380141854286194,
35
+ "learning_rate": 3e-05,
36
+ "loss": 1.8795,
37
+ "step": 400
38
+ },
39
+ {
40
+ "epoch": 0.945179584120983,
41
+ "grad_norm": 0.3052491247653961,
42
+ "learning_rate": 3e-05,
43
+ "loss": 1.86,
44
+ "step": 500
45
+ },
46
+ {
47
+ "epoch": 1.0,
48
+ "eval_accuracy": 0.605677130044843,
49
+ "eval_loss": 1.6920065879821777,
50
+ "eval_runtime": 6.0828,
51
+ "eval_samples_per_second": 82.199,
52
+ "eval_steps_per_second": 10.357,
53
+ "step": 529
54
+ },
55
+ {
56
+ "epoch": 1.1342155009451795,
57
+ "grad_norm": 0.38703393936157227,
58
+ "learning_rate": 3e-05,
59
+ "loss": 1.8651,
60
+ "step": 600
61
+ },
62
+ {
63
+ "epoch": 1.3232514177693762,
64
+ "grad_norm": 0.5257138609886169,
65
+ "learning_rate": 3e-05,
66
+ "loss": 1.8238,
67
+ "step": 700
68
+ },
69
+ {
70
+ "epoch": 1.5122873345935728,
71
+ "grad_norm": 0.5151342749595642,
72
+ "learning_rate": 3e-05,
73
+ "loss": 1.8252,
74
+ "step": 800
75
+ },
76
+ {
77
+ "epoch": 1.7013232514177694,
78
+ "grad_norm": 0.4788012206554413,
79
+ "learning_rate": 3e-05,
80
+ "loss": 1.8178,
81
+ "step": 900
82
+ },
83
+ {
84
+ "epoch": 1.8903591682419658,
85
+ "grad_norm": 0.506370484828949,
86
+ "learning_rate": 3e-05,
87
+ "loss": 1.8271,
88
+ "step": 1000
89
+ },
90
+ {
91
+ "epoch": 2.0,
92
+ "eval_accuracy": 0.6118744394618834,
93
+ "eval_loss": 1.6425946950912476,
94
+ "eval_runtime": 6.042,
95
+ "eval_samples_per_second": 82.754,
96
+ "eval_steps_per_second": 10.427,
97
+ "step": 1058
98
+ },
99
+ {
100
+ "epoch": 2.0793950850661624,
101
+ "grad_norm": 0.537693440914154,
102
+ "learning_rate": 3e-05,
103
+ "loss": 1.795,
104
+ "step": 1100
105
+ },
106
+ {
107
+ "epoch": 2.268431001890359,
108
+ "grad_norm": 0.6806929111480713,
109
+ "learning_rate": 3e-05,
110
+ "loss": 1.773,
111
+ "step": 1200
112
+ },
113
+ {
114
+ "epoch": 2.4574669187145557,
115
+ "grad_norm": 0.7064961194992065,
116
+ "learning_rate": 3e-05,
117
+ "loss": 1.7609,
118
+ "step": 1300
119
+ },
120
+ {
121
+ "epoch": 2.6465028355387523,
122
+ "grad_norm": 0.7571425437927246,
123
+ "learning_rate": 3e-05,
124
+ "loss": 1.7536,
125
+ "step": 1400
126
+ },
127
+ {
128
+ "epoch": 2.835538752362949,
129
+ "grad_norm": 0.8069006204605103,
130
+ "learning_rate": 3e-05,
131
+ "loss": 1.7324,
132
+ "step": 1500
133
+ },
134
+ {
135
+ "epoch": 3.0,
136
+ "eval_accuracy": 0.6163408071748879,
137
+ "eval_loss": 1.59978187084198,
138
+ "eval_runtime": 6.4372,
139
+ "eval_samples_per_second": 77.673,
140
+ "eval_steps_per_second": 9.787,
141
+ "step": 1587
142
+ },
143
+ {
144
+ "epoch": 3.0245746691871456,
145
+ "grad_norm": 0.7909080982208252,
146
+ "learning_rate": 3e-05,
147
+ "loss": 1.7206,
148
+ "step": 1600
149
+ },
150
+ {
151
+ "epoch": 3.213610586011342,
152
+ "grad_norm": 0.8461490869522095,
153
+ "learning_rate": 3e-05,
154
+ "loss": 1.6696,
155
+ "step": 1700
156
+ },
157
+ {
158
+ "epoch": 3.402646502835539,
159
+ "grad_norm": 0.9591927528381348,
160
+ "learning_rate": 3e-05,
161
+ "loss": 1.672,
162
+ "step": 1800
163
+ },
164
+ {
165
+ "epoch": 3.5916824196597354,
166
+ "grad_norm": 1.0703200101852417,
167
+ "learning_rate": 3e-05,
168
+ "loss": 1.6712,
169
+ "step": 1900
170
+ },
171
+ {
172
+ "epoch": 3.780718336483932,
173
+ "grad_norm": 0.9409723281860352,
174
+ "learning_rate": 3e-05,
175
+ "loss": 1.6877,
176
+ "step": 2000
177
+ },
178
+ {
179
+ "epoch": 3.9697542533081287,
180
+ "grad_norm": 0.9087811708450317,
181
+ "learning_rate": 3e-05,
182
+ "loss": 1.6818,
183
+ "step": 2100
184
+ },
185
+ {
186
+ "epoch": 4.0,
187
+ "eval_accuracy": 0.621982062780269,
188
+ "eval_loss": 1.5579897165298462,
189
+ "eval_runtime": 6.0535,
190
+ "eval_samples_per_second": 82.597,
191
+ "eval_steps_per_second": 10.407,
192
+ "step": 2116
193
+ },
194
+ {
195
+ "epoch": 4.158790170132325,
196
+ "grad_norm": 1.0507543087005615,
197
+ "learning_rate": 3e-05,
198
+ "loss": 1.6195,
199
+ "step": 2200
200
+ },
201
+ {
202
+ "epoch": 4.3478260869565215,
203
+ "grad_norm": 1.1318222284317017,
204
+ "learning_rate": 3e-05,
205
+ "loss": 1.6039,
206
+ "step": 2300
207
+ },
208
+ {
209
+ "epoch": 4.536862003780718,
210
+ "grad_norm": 1.1028773784637451,
211
+ "learning_rate": 3e-05,
212
+ "loss": 1.5924,
213
+ "step": 2400
214
+ },
215
+ {
216
+ "epoch": 4.725897920604915,
217
+ "grad_norm": 1.1365679502487183,
218
+ "learning_rate": 3e-05,
219
+ "loss": 1.5903,
220
+ "step": 2500
221
+ },
222
+ {
223
+ "epoch": 4.914933837429111,
224
+ "grad_norm": 1.0776621103286743,
225
+ "learning_rate": 3e-05,
226
+ "loss": 1.5864,
227
+ "step": 2600
228
+ },
229
+ {
230
+ "epoch": 5.0,
231
+ "eval_accuracy": 0.6278206278026905,
232
+ "eval_loss": 1.5211377143859863,
233
+ "eval_runtime": 6.0103,
234
+ "eval_samples_per_second": 83.19,
235
+ "eval_steps_per_second": 10.482,
236
+ "step": 2645
237
+ },
238
+ {
239
+ "epoch": 5.103969754253308,
240
+ "grad_norm": 1.1829136610031128,
241
+ "learning_rate": 3e-05,
242
+ "loss": 1.5478,
243
+ "step": 2700
244
+ },
245
+ {
246
+ "epoch": 5.293005671077505,
247
+ "grad_norm": 1.1871124505996704,
248
+ "learning_rate": 3e-05,
249
+ "loss": 1.5151,
250
+ "step": 2800
251
+ },
252
+ {
253
+ "epoch": 5.482041587901701,
254
+ "grad_norm": 1.2089450359344482,
255
+ "learning_rate": 3e-05,
256
+ "loss": 1.5159,
257
+ "step": 2900
258
+ },
259
+ {
260
+ "epoch": 5.671077504725898,
261
+ "grad_norm": 1.3170595169067383,
262
+ "learning_rate": 3e-05,
263
+ "loss": 1.5241,
264
+ "step": 3000
265
+ },
266
+ {
267
+ "epoch": 5.8601134215500945,
268
+ "grad_norm": 1.439405083656311,
269
+ "learning_rate": 3e-05,
270
+ "loss": 1.5204,
271
+ "step": 3100
272
+ },
273
+ {
274
+ "epoch": 6.0,
275
+ "eval_accuracy": 0.6326726457399103,
276
+ "eval_loss": 1.4863040447235107,
277
+ "eval_runtime": 6.0604,
278
+ "eval_samples_per_second": 82.502,
279
+ "eval_steps_per_second": 10.395,
280
+ "step": 3174
281
+ },
282
+ {
283
+ "epoch": 6.049149338374291,
284
+ "grad_norm": 1.3573708534240723,
285
+ "learning_rate": 3e-05,
286
+ "loss": 1.506,
287
+ "step": 3200
288
+ },
289
+ {
290
+ "epoch": 6.238185255198488,
291
+ "grad_norm": 1.3852615356445312,
292
+ "learning_rate": 3e-05,
293
+ "loss": 1.4401,
294
+ "step": 3300
295
+ },
296
+ {
297
+ "epoch": 6.427221172022684,
298
+ "grad_norm": 1.391204595565796,
299
+ "learning_rate": 3e-05,
300
+ "loss": 1.4469,
301
+ "step": 3400
302
+ },
303
+ {
304
+ "epoch": 6.616257088846881,
305
+ "grad_norm": 1.5318756103515625,
306
+ "learning_rate": 3e-05,
307
+ "loss": 1.4608,
308
+ "step": 3500
309
+ },
310
+ {
311
+ "epoch": 6.805293005671078,
312
+ "grad_norm": 1.4233124256134033,
313
+ "learning_rate": 3e-05,
314
+ "loss": 1.4419,
315
+ "step": 3600
316
+ },
317
+ {
318
+ "epoch": 6.994328922495274,
319
+ "grad_norm": 1.4444003105163574,
320
+ "learning_rate": 3e-05,
321
+ "loss": 1.4481,
322
+ "step": 3700
323
+ },
324
+ {
325
+ "epoch": 7.0,
326
+ "eval_accuracy": 0.6371928251121076,
327
+ "eval_loss": 1.4516844749450684,
328
+ "eval_runtime": 6.0218,
329
+ "eval_samples_per_second": 83.032,
330
+ "eval_steps_per_second": 10.462,
331
+ "step": 3703
332
+ },
333
+ {
334
+ "epoch": 7.183364839319471,
335
+ "grad_norm": 1.607857346534729,
336
+ "learning_rate": 3e-05,
337
+ "loss": 1.3742,
338
+ "step": 3800
339
+ },
340
+ {
341
+ "epoch": 7.3724007561436675,
342
+ "grad_norm": 1.5741080045700073,
343
+ "learning_rate": 3e-05,
344
+ "loss": 1.383,
345
+ "step": 3900
346
+ },
347
+ {
348
+ "epoch": 7.561436672967864,
349
+ "grad_norm": 1.6515594720840454,
350
+ "learning_rate": 3e-05,
351
+ "loss": 1.3734,
352
+ "step": 4000
353
+ },
354
+ {
355
+ "epoch": 7.750472589792061,
356
+ "grad_norm": 1.704078197479248,
357
+ "learning_rate": 3e-05,
358
+ "loss": 1.3527,
359
+ "step": 4100
360
+ },
361
+ {
362
+ "epoch": 7.939508506616257,
363
+ "grad_norm": 1.7636491060256958,
364
+ "learning_rate": 3e-05,
365
+ "loss": 1.3768,
366
+ "step": 4200
367
+ },
368
+ {
369
+ "epoch": 8.0,
370
+ "eval_accuracy": 0.6429058295964125,
371
+ "eval_loss": 1.4120872020721436,
372
+ "eval_runtime": 6.7429,
373
+ "eval_samples_per_second": 74.152,
374
+ "eval_steps_per_second": 9.343,
375
+ "step": 4232
376
+ },
377
+ {
378
+ "epoch": 8.128544423440454,
379
+ "grad_norm": 1.691927433013916,
380
+ "learning_rate": 3e-05,
381
+ "loss": 1.3254,
382
+ "step": 4300
383
+ },
384
+ {
385
+ "epoch": 8.31758034026465,
386
+ "grad_norm": 1.695786476135254,
387
+ "learning_rate": 3e-05,
388
+ "loss": 1.2819,
389
+ "step": 4400
390
+ },
391
+ {
392
+ "epoch": 8.506616257088847,
393
+ "grad_norm": 1.8343569040298462,
394
+ "learning_rate": 3e-05,
395
+ "loss": 1.3225,
396
+ "step": 4500
397
+ },
398
+ {
399
+ "epoch": 8.695652173913043,
400
+ "grad_norm": 1.678654670715332,
401
+ "learning_rate": 3e-05,
402
+ "loss": 1.2961,
403
+ "step": 4600
404
+ },
405
+ {
406
+ "epoch": 8.88468809073724,
407
+ "grad_norm": 2.0212690830230713,
408
+ "learning_rate": 3e-05,
409
+ "loss": 1.2946,
410
+ "step": 4700
411
+ },
412
+ {
413
+ "epoch": 9.0,
414
+ "eval_accuracy": 0.6481973094170403,
415
+ "eval_loss": 1.3739463090896606,
416
+ "eval_runtime": 6.0696,
417
+ "eval_samples_per_second": 82.377,
418
+ "eval_steps_per_second": 10.38,
419
+ "step": 4761
420
+ },
421
+ {
422
+ "epoch": 9.073724007561436,
423
+ "grad_norm": 1.8944847583770752,
424
+ "learning_rate": 3e-05,
425
+ "loss": 1.2831,
426
+ "step": 4800
427
+ },
428
+ {
429
+ "epoch": 9.262759924385634,
430
+ "grad_norm": 1.9275157451629639,
431
+ "learning_rate": 3e-05,
432
+ "loss": 1.2022,
433
+ "step": 4900
434
+ },
435
+ {
436
+ "epoch": 9.45179584120983,
437
+ "grad_norm": 1.927822470664978,
438
+ "learning_rate": 3e-05,
439
+ "loss": 1.2259,
440
+ "step": 5000
441
+ },
442
+ {
443
+ "epoch": 9.640831758034027,
444
+ "grad_norm": 1.8706145286560059,
445
+ "learning_rate": 3e-05,
446
+ "loss": 1.2231,
447
+ "step": 5100
448
+ },
449
+ {
450
+ "epoch": 9.829867674858223,
451
+ "grad_norm": 1.858546257019043,
452
+ "learning_rate": 3e-05,
453
+ "loss": 1.243,
454
+ "step": 5200
455
+ },
456
+ {
457
+ "epoch": 10.0,
458
+ "eval_accuracy": 0.653237668161435,
459
+ "eval_loss": 1.3363806009292603,
460
+ "eval_runtime": 6.6274,
461
+ "eval_samples_per_second": 75.444,
462
+ "eval_steps_per_second": 9.506,
463
+ "step": 5290
464
+ },
465
+ {
466
+ "epoch": 10.01890359168242,
467
+ "grad_norm": 2.2799806594848633,
468
+ "learning_rate": 3e-05,
469
+ "loss": 1.2223,
470
+ "step": 5300
471
+ },
472
+ {
473
+ "epoch": 10.207939508506616,
474
+ "grad_norm": 2.0888822078704834,
475
+ "learning_rate": 3e-05,
476
+ "loss": 1.1567,
477
+ "step": 5400
478
+ },
479
+ {
480
+ "epoch": 10.396975425330814,
481
+ "grad_norm": 2.1389222145080566,
482
+ "learning_rate": 3e-05,
483
+ "loss": 1.1421,
484
+ "step": 5500
485
+ },
486
+ {
487
+ "epoch": 10.58601134215501,
488
+ "grad_norm": 2.253117322921753,
489
+ "learning_rate": 3e-05,
490
+ "loss": 1.1525,
491
+ "step": 5600
492
+ },
493
+ {
494
+ "epoch": 10.775047258979207,
495
+ "grad_norm": 2.3171472549438477,
496
+ "learning_rate": 3e-05,
497
+ "loss": 1.168,
498
+ "step": 5700
499
+ },
500
+ {
501
+ "epoch": 10.964083175803403,
502
+ "grad_norm": 2.1451621055603027,
503
+ "learning_rate": 3e-05,
504
+ "loss": 1.1425,
505
+ "step": 5800
506
+ },
507
+ {
508
+ "epoch": 11.0,
509
+ "eval_accuracy": 0.6593991031390134,
510
+ "eval_loss": 1.2968207597732544,
511
+ "eval_runtime": 5.9969,
512
+ "eval_samples_per_second": 83.377,
513
+ "eval_steps_per_second": 10.506,
514
+ "step": 5819
515
+ },
516
+ {
517
+ "epoch": 11.1531190926276,
518
+ "grad_norm": 2.0139310359954834,
519
+ "learning_rate": 3e-05,
520
+ "loss": 1.0814,
521
+ "step": 5900
522
+ },
523
+ {
524
+ "epoch": 11.342155009451796,
525
+ "grad_norm": 2.2693676948547363,
526
+ "learning_rate": 3e-05,
527
+ "loss": 1.0731,
528
+ "step": 6000
529
+ },
530
+ {
531
+ "epoch": 11.531190926275993,
532
+ "grad_norm": 2.4319376945495605,
533
+ "learning_rate": 3e-05,
534
+ "loss": 1.0842,
535
+ "step": 6100
536
+ },
537
+ {
538
+ "epoch": 11.720226843100189,
539
+ "grad_norm": 2.103640079498291,
540
+ "learning_rate": 3e-05,
541
+ "loss": 1.0839,
542
+ "step": 6200
543
+ },
544
+ {
545
+ "epoch": 11.909262759924385,
546
+ "grad_norm": 2.309508800506592,
547
+ "learning_rate": 3e-05,
548
+ "loss": 1.0847,
549
+ "step": 6300
550
+ },
551
+ {
552
+ "epoch": 12.0,
553
+ "eval_accuracy": 0.6652017937219731,
554
+ "eval_loss": 1.2538820505142212,
555
+ "eval_runtime": 6.0994,
556
+ "eval_samples_per_second": 81.975,
557
+ "eval_steps_per_second": 10.329,
558
+ "step": 6348
559
+ },
560
+ {
561
+ "epoch": 12.098298676748582,
562
+ "grad_norm": 2.184951066970825,
563
+ "learning_rate": 3e-05,
564
+ "loss": 1.0154,
565
+ "step": 6400
566
+ },
567
+ {
568
+ "epoch": 12.287334593572778,
569
+ "grad_norm": 2.6142184734344482,
570
+ "learning_rate": 3e-05,
571
+ "loss": 1.0129,
572
+ "step": 6500
573
+ },
574
+ {
575
+ "epoch": 12.476370510396976,
576
+ "grad_norm": 2.377812147140503,
577
+ "learning_rate": 3e-05,
578
+ "loss": 1.01,
579
+ "step": 6600
580
+ },
581
+ {
582
+ "epoch": 12.665406427221171,
583
+ "grad_norm": 2.252695083618164,
584
+ "learning_rate": 3e-05,
585
+ "loss": 1.0132,
586
+ "step": 6700
587
+ },
588
+ {
589
+ "epoch": 12.854442344045369,
590
+ "grad_norm": 2.4628734588623047,
591
+ "learning_rate": 3e-05,
592
+ "loss": 1.0152,
593
+ "step": 6800
594
+ },
595
+ {
596
+ "epoch": 13.0,
597
+ "eval_accuracy": 0.670609865470852,
598
+ "eval_loss": 1.216369390487671,
599
+ "eval_runtime": 6.115,
600
+ "eval_samples_per_second": 81.766,
601
+ "eval_steps_per_second": 10.302,
602
+ "step": 6877
603
+ },
604
+ {
605
+ "epoch": 13.043478260869565,
606
+ "grad_norm": 2.2027320861816406,
607
+ "learning_rate": 3e-05,
608
+ "loss": 1.0047,
609
+ "step": 6900
610
+ },
611
+ {
612
+ "epoch": 13.232514177693762,
613
+ "grad_norm": 2.1473910808563232,
614
+ "learning_rate": 3e-05,
615
+ "loss": 0.9308,
616
+ "step": 7000
617
+ },
618
+ {
619
+ "epoch": 13.421550094517958,
620
+ "grad_norm": 2.633467435836792,
621
+ "learning_rate": 3e-05,
622
+ "loss": 0.9265,
623
+ "step": 7100
624
+ },
625
+ {
626
+ "epoch": 13.610586011342155,
627
+ "grad_norm": 2.5691654682159424,
628
+ "learning_rate": 3e-05,
629
+ "loss": 0.9485,
630
+ "step": 7200
631
+ },
632
+ {
633
+ "epoch": 13.799621928166351,
634
+ "grad_norm": 2.6028811931610107,
635
+ "learning_rate": 3e-05,
636
+ "loss": 0.9502,
637
+ "step": 7300
638
+ },
639
+ {
640
+ "epoch": 13.988657844990549,
641
+ "grad_norm": 2.6901166439056396,
642
+ "learning_rate": 3e-05,
643
+ "loss": 0.9498,
644
+ "step": 7400
645
+ },
646
+ {
647
+ "epoch": 14.0,
648
+ "eval_accuracy": 0.6757937219730942,
649
+ "eval_loss": 1.176974892616272,
650
+ "eval_runtime": 6.0622,
651
+ "eval_samples_per_second": 82.479,
652
+ "eval_steps_per_second": 10.392,
653
+ "step": 7406
654
+ },
655
+ {
656
+ "epoch": 14.177693761814744,
657
+ "grad_norm": 2.624717950820923,
658
+ "learning_rate": 3e-05,
659
+ "loss": 0.8708,
660
+ "step": 7500
661
+ },
662
+ {
663
+ "epoch": 14.366729678638942,
664
+ "grad_norm": 2.5228078365325928,
665
+ "learning_rate": 3e-05,
666
+ "loss": 0.8685,
667
+ "step": 7600
668
+ },
669
+ {
670
+ "epoch": 14.555765595463138,
671
+ "grad_norm": 2.8440725803375244,
672
+ "learning_rate": 3e-05,
673
+ "loss": 0.88,
674
+ "step": 7700
675
+ },
676
+ {
677
+ "epoch": 14.744801512287335,
678
+ "grad_norm": 2.5001230239868164,
679
+ "learning_rate": 3e-05,
680
+ "loss": 0.8968,
681
+ "step": 7800
682
+ },
683
+ {
684
+ "epoch": 14.93383742911153,
685
+ "grad_norm": 2.868446111679077,
686
+ "learning_rate": 3e-05,
687
+ "loss": 0.8652,
688
+ "step": 7900
689
+ },
690
+ {
691
+ "epoch": 15.0,
692
+ "eval_accuracy": 0.6820717488789237,
693
+ "eval_loss": 1.1323232650756836,
694
+ "eval_runtime": 6.0411,
695
+ "eval_samples_per_second": 82.767,
696
+ "eval_steps_per_second": 10.429,
697
+ "step": 7935
698
+ },
699
+ {
700
+ "epoch": 15.122873345935728,
701
+ "grad_norm": 3.236562728881836,
702
+ "learning_rate": 3e-05,
703
+ "loss": 0.8354,
704
+ "step": 8000
705
+ },
706
+ {
707
+ "epoch": 15.311909262759924,
708
+ "grad_norm": 3.0811691284179688,
709
+ "learning_rate": 3e-05,
710
+ "loss": 0.8042,
711
+ "step": 8100
712
+ },
713
+ {
714
+ "epoch": 15.500945179584122,
715
+ "grad_norm": 2.6125237941741943,
716
+ "learning_rate": 3e-05,
717
+ "loss": 0.8097,
718
+ "step": 8200
719
+ },
720
+ {
721
+ "epoch": 15.689981096408317,
722
+ "grad_norm": 2.8925766944885254,
723
+ "learning_rate": 3e-05,
724
+ "loss": 0.8133,
725
+ "step": 8300
726
+ },
727
+ {
728
+ "epoch": 15.879017013232515,
729
+ "grad_norm": 2.927067518234253,
730
+ "learning_rate": 3e-05,
731
+ "loss": 0.8265,
732
+ "step": 8400
733
+ },
734
+ {
735
+ "epoch": 16.0,
736
+ "eval_accuracy": 0.690134529147982,
737
+ "eval_loss": 1.0825910568237305,
738
+ "eval_runtime": 6.0821,
739
+ "eval_samples_per_second": 82.209,
740
+ "eval_steps_per_second": 10.358,
741
+ "step": 8464
742
+ },
743
+ {
744
+ "epoch": 16.068052930056712,
745
+ "grad_norm": 2.8401355743408203,
746
+ "learning_rate": 3e-05,
747
+ "loss": 0.7933,
748
+ "step": 8500
749
+ },
750
+ {
751
+ "epoch": 16.257088846880908,
752
+ "grad_norm": 3.2201480865478516,
753
+ "learning_rate": 3e-05,
754
+ "loss": 0.7449,
755
+ "step": 8600
756
+ },
757
+ {
758
+ "epoch": 16.446124763705104,
759
+ "grad_norm": 3.2903711795806885,
760
+ "learning_rate": 3e-05,
761
+ "loss": 0.7565,
762
+ "step": 8700
763
+ },
764
+ {
765
+ "epoch": 16.6351606805293,
766
+ "grad_norm": 3.153130292892456,
767
+ "learning_rate": 3e-05,
768
+ "loss": 0.773,
769
+ "step": 8800
770
+ },
771
+ {
772
+ "epoch": 16.8241965973535,
773
+ "grad_norm": 2.8367865085601807,
774
+ "learning_rate": 3e-05,
775
+ "loss": 0.7432,
776
+ "step": 8900
777
+ },
778
+ {
779
+ "epoch": 17.0,
780
+ "eval_accuracy": 0.6959910313901345,
781
+ "eval_loss": 1.04633367061615,
782
+ "eval_runtime": 6.1125,
783
+ "eval_samples_per_second": 81.8,
784
+ "eval_steps_per_second": 10.307,
785
+ "step": 8993
786
+ },
787
+ {
788
+ "epoch": 17.013232514177695,
789
+ "grad_norm": 3.268256664276123,
790
+ "learning_rate": 3e-05,
791
+ "loss": 0.7576,
792
+ "step": 9000
793
+ },
794
+ {
795
+ "epoch": 17.20226843100189,
796
+ "grad_norm": 3.0888004302978516,
797
+ "learning_rate": 3e-05,
798
+ "loss": 0.6879,
799
+ "step": 9100
800
+ },
801
+ {
802
+ "epoch": 17.391304347826086,
803
+ "grad_norm": 2.908324718475342,
804
+ "learning_rate": 3e-05,
805
+ "loss": 0.6869,
806
+ "step": 9200
807
+ },
808
+ {
809
+ "epoch": 17.58034026465028,
810
+ "grad_norm": 4.358062267303467,
811
+ "learning_rate": 3e-05,
812
+ "loss": 0.7027,
813
+ "step": 9300
814
+ },
815
+ {
816
+ "epoch": 17.76937618147448,
817
+ "grad_norm": 3.1521716117858887,
818
+ "learning_rate": 3e-05,
819
+ "loss": 0.7188,
820
+ "step": 9400
821
+ },
822
+ {
823
+ "epoch": 17.958412098298677,
824
+ "grad_norm": 3.1939432621002197,
825
+ "learning_rate": 3e-05,
826
+ "loss": 0.7106,
827
+ "step": 9500
828
+ },
829
+ {
830
+ "epoch": 18.0,
831
+ "eval_accuracy": 0.7021973094170404,
832
+ "eval_loss": 1.0097612142562866,
833
+ "eval_runtime": 5.6404,
834
+ "eval_samples_per_second": 88.646,
835
+ "eval_steps_per_second": 11.169,
836
+ "step": 9522
837
+ },
838
+ {
839
+ "epoch": 18.147448015122873,
840
+ "grad_norm": 3.3058857917785645,
841
+ "learning_rate": 3e-05,
842
+ "loss": 0.6526,
843
+ "step": 9600
844
+ },
845
+ {
846
+ "epoch": 18.33648393194707,
847
+ "grad_norm": 2.845630645751953,
848
+ "learning_rate": 3e-05,
849
+ "loss": 0.6375,
850
+ "step": 9700
851
+ },
852
+ {
853
+ "epoch": 18.525519848771268,
854
+ "grad_norm": 3.524501323699951,
855
+ "learning_rate": 3e-05,
856
+ "loss": 0.65,
857
+ "step": 9800
858
+ },
859
+ {
860
+ "epoch": 18.714555765595463,
861
+ "grad_norm": 3.5869174003601074,
862
+ "learning_rate": 3e-05,
863
+ "loss": 0.6529,
864
+ "step": 9900
865
+ },
866
+ {
867
+ "epoch": 18.90359168241966,
868
+ "grad_norm": 2.8486292362213135,
869
+ "learning_rate": 3e-05,
870
+ "loss": 0.669,
871
+ "step": 10000
872
+ },
873
+ {
874
+ "epoch": 19.0,
875
+ "eval_accuracy": 0.7077937219730942,
876
+ "eval_loss": 0.9695693254470825,
877
+ "eval_runtime": 6.1993,
878
+ "eval_samples_per_second": 80.654,
879
+ "eval_steps_per_second": 10.162,
880
+ "step": 10051
881
+ },
882
+ {
883
+ "epoch": 19.092627599243855,
884
+ "grad_norm": 2.8552756309509277,
885
+ "learning_rate": 3e-05,
886
+ "loss": 0.6183,
887
+ "step": 10100
888
+ },
889
+ {
890
+ "epoch": 19.281663516068054,
891
+ "grad_norm": 2.949084758758545,
892
+ "learning_rate": 3e-05,
893
+ "loss": 0.5824,
894
+ "step": 10200
895
+ },
896
+ {
897
+ "epoch": 19.47069943289225,
898
+ "grad_norm": 3.137747049331665,
899
+ "learning_rate": 3e-05,
900
+ "loss": 0.6093,
901
+ "step": 10300
902
+ },
903
+ {
904
+ "epoch": 19.659735349716446,
905
+ "grad_norm": 2.7827863693237305,
906
+ "learning_rate": 3e-05,
907
+ "loss": 0.6155,
908
+ "step": 10400
909
+ },
910
+ {
911
+ "epoch": 19.84877126654064,
912
+ "grad_norm": 3.2223191261291504,
913
+ "learning_rate": 3e-05,
914
+ "loss": 0.6043,
915
+ "step": 10500
916
+ },
917
+ {
918
+ "epoch": 20.0,
919
+ "eval_accuracy": 0.7134618834080717,
920
+ "eval_loss": 0.9358564019203186,
921
+ "eval_runtime": 6.1149,
922
+ "eval_samples_per_second": 81.767,
923
+ "eval_steps_per_second": 10.303,
924
+ "step": 10580
925
+ },
926
+ {
927
+ "epoch": 20.0,
928
+ "step": 10580,
929
+ "total_flos": 9.64242245391745e+17,
930
+ "train_loss": 1.2100651977192927,
931
+ "train_runtime": 22841.5406,
932
+ "train_samples_per_second": 14.819,
933
+ "train_steps_per_second": 0.463
934
+ }
935
+ ],
936
+ "logging_steps": 100,
937
+ "max_steps": 10580,
938
+ "num_input_tokens_seen": 0,
939
+ "num_train_epochs": 20,
940
+ "save_steps": 500,
941
+ "total_flos": 9.64242245391745e+17,
942
+ "train_batch_size": 1,
943
+ "trial_name": null,
944
+ "trial_params": null
945
+ }