MHGanainy commited on
Commit
3148899
·
verified ·
1 Parent(s): e69abb1

MHGanainy/gpt2-xl-lora-multi-0

Browse files
Files changed (5) hide show
  1. README.md +2 -0
  2. all_results.json +10 -10
  3. eval_results.json +5 -5
  4. train_results.json +5 -5
  5. trainer_state.json +371 -77
README.md CHANGED
@@ -15,6 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
15
  # gpt2-xl-lora-multi-0
16
 
17
  This model is a fine-tuned version of [openai-community/gpt2-xl](https://huggingface.co/openai-community/gpt2-xl) on an unknown dataset.
 
 
18
 
19
  ## Model description
20
 
 
15
  # gpt2-xl-lora-multi-0
16
 
17
  This model is a fine-tuned version of [openai-community/gpt2-xl](https://huggingface.co/openai-community/gpt2-xl) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 2.3155
20
 
21
  ## Model description
22
 
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 1.0,
3
- "eval_loss": 2.001722574234009,
4
- "eval_runtime": 501.6413,
5
- "eval_samples_per_second": 7.834,
6
- "eval_steps_per_second": 0.49,
7
- "perplexity": 7.401795265510334,
8
- "total_flos": 2.56463655569195e+17,
9
- "train_loss": 2.512042249332775,
10
- "train_runtime": 561.7208,
11
- "train_samples_per_second": 50.117,
12
- "train_steps_per_second": 3.133
13
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "eval_loss": 2.3154852390289307,
4
+ "eval_runtime": 1837.7064,
5
+ "eval_samples_per_second": 7.781,
6
+ "eval_steps_per_second": 0.486,
7
+ "perplexity": 10.129837118692777,
8
+ "total_flos": 8.687706333480223e+17,
9
+ "train_loss": 2.596681281969236,
10
+ "train_runtime": 1897.3372,
11
+ "train_samples_per_second": 50.272,
12
+ "train_steps_per_second": 3.142
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "eval_loss": 2.001722574234009,
4
- "eval_runtime": 501.6413,
5
- "eval_samples_per_second": 7.834,
6
- "eval_steps_per_second": 0.49,
7
- "perplexity": 7.401795265510334
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "eval_loss": 2.3154852390289307,
4
+ "eval_runtime": 1837.7064,
5
+ "eval_samples_per_second": 7.781,
6
+ "eval_steps_per_second": 0.486,
7
+ "perplexity": 10.129837118692777
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "total_flos": 2.56463655569195e+17,
4
- "train_loss": 2.512042249332775,
5
- "train_runtime": 561.7208,
6
- "train_samples_per_second": 50.117,
7
- "train_steps_per_second": 3.133
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "total_flos": 8.687706333480223e+17,
4
+ "train_loss": 2.596681281969236,
5
+ "train_runtime": 1897.3372,
6
+ "train_samples_per_second": 50.272,
7
+ "train_steps_per_second": 3.142
8
  }
trainer_state.json CHANGED
@@ -3,142 +3,436 @@
3
  "best_model_checkpoint": null,
4
  "epoch": 1.0,
5
  "eval_steps": 500,
6
- "global_step": 1760,
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.056818181818181816,
13
- "grad_norm": 0.2199665755033493,
14
- "learning_rate": 1.378820184790334e-06,
15
- "loss": 3.1675,
16
  "step": 100
17
  },
18
  {
19
- "epoch": 0.11363636363636363,
20
- "grad_norm": 0.25001513957977295,
21
- "learning_rate": 2.8002842928216066e-06,
22
- "loss": 3.1713,
23
  "step": 200
24
  },
25
  {
26
- "epoch": 0.17045454545454544,
27
- "grad_norm": 0.2622998356819153,
28
- "learning_rate": 4.221748400852878e-06,
29
- "loss": 3.0232,
30
  "step": 300
31
  },
32
  {
33
- "epoch": 0.22727272727272727,
34
- "grad_norm": 0.22813254594802856,
35
- "learning_rate": 5.628997867803838e-06,
36
- "loss": 2.9501,
37
  "step": 400
38
  },
39
  {
40
- "epoch": 0.2840909090909091,
41
- "grad_norm": 0.22347629070281982,
42
- "learning_rate": 7.050461975835111e-06,
43
- "loss": 2.7414,
44
  "step": 500
45
  },
46
  {
47
- "epoch": 0.3409090909090909,
48
- "grad_norm": 0.28061142563819885,
49
- "learning_rate": 8.471926083866383e-06,
50
- "loss": 2.5525,
51
  "step": 600
52
  },
53
  {
54
- "epoch": 0.3977272727272727,
55
- "grad_norm": 0.36509305238723755,
56
- "learning_rate": 9.893390191897656e-06,
57
- "loss": 2.4894,
58
  "step": 700
59
  },
60
  {
61
- "epoch": 0.45454545454545453,
62
- "grad_norm": 0.323941707611084,
63
- "learning_rate": 1.1314854299928929e-05,
64
- "loss": 2.4402,
65
  "step": 800
66
  },
67
  {
68
- "epoch": 0.5113636363636364,
69
- "grad_norm": 0.24332045018672943,
70
- "learning_rate": 1.27363184079602e-05,
71
- "loss": 2.4124,
72
  "step": 900
73
  },
74
  {
75
- "epoch": 0.5681818181818182,
76
- "grad_norm": 0.23532022535800934,
77
- "learning_rate": 1.4157782515991471e-05,
78
- "loss": 2.3748,
79
  "step": 1000
80
  },
81
  {
82
- "epoch": 0.625,
83
- "grad_norm": 0.3698301911354065,
84
- "learning_rate": 1.5579246624022746e-05,
85
- "loss": 2.3434,
86
  "step": 1100
87
  },
88
  {
89
- "epoch": 0.6818181818181818,
90
- "grad_norm": 0.3040252923965454,
91
- "learning_rate": 1.7000710732054017e-05,
92
- "loss": 2.2856,
93
  "step": 1200
94
  },
95
  {
96
- "epoch": 0.7386363636363636,
97
- "grad_norm": 0.4269082546234131,
98
- "learning_rate": 1.842217484008529e-05,
99
- "loss": 2.2652,
100
  "step": 1300
101
  },
102
  {
103
- "epoch": 0.7954545454545454,
104
- "grad_norm": 0.30837833881378174,
105
- "learning_rate": 1.9843638948116563e-05,
106
- "loss": 2.2236,
107
  "step": 1400
108
  },
109
  {
110
- "epoch": 0.8522727272727273,
111
- "grad_norm": 0.4666430652141571,
112
- "learning_rate": 1.702371288083127e-05,
113
- "loss": 2.1853,
114
  "step": 1500
115
  },
116
  {
117
- "epoch": 0.9090909090909091,
118
- "grad_norm": 0.3418198525905609,
119
- "learning_rate": 8.88983117848755e-06,
120
- "loss": 2.1622,
121
  "step": 1600
122
  },
123
  {
124
- "epoch": 0.9659090909090909,
125
- "grad_norm": 0.32931217551231384,
126
- "learning_rate": 1.5787274364278004e-06,
127
- "loss": 2.1652,
128
  "step": 1700
129
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
130
  {
131
  "epoch": 1.0,
132
- "step": 1760,
133
- "total_flos": 2.56463655569195e+17,
134
- "train_loss": 2.512042249332775,
135
- "train_runtime": 561.7208,
136
- "train_samples_per_second": 50.117,
137
- "train_steps_per_second": 3.133
138
  }
139
  ],
140
  "logging_steps": 100,
141
- "max_steps": 1760,
142
  "num_input_tokens_seen": 0,
143
  "num_train_epochs": 1,
144
  "save_steps": 500,
@@ -154,7 +448,7 @@
154
  "attributes": {}
155
  }
156
  },
157
- "total_flos": 2.56463655569195e+17,
158
  "train_batch_size": 2,
159
  "trial_name": null,
160
  "trial_params": null
 
3
  "best_model_checkpoint": null,
4
  "epoch": 1.0,
5
  "eval_steps": 500,
6
+ "global_step": 5962,
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.01677289500167729,
13
+ "grad_norm": 0.048490237444639206,
14
+ "learning_rate": 4.193751310547285e-07,
15
+ "loss": 2.7454,
16
  "step": 100
17
  },
18
  {
19
+ "epoch": 0.03354579000335458,
20
+ "grad_norm": 0.05207354202866554,
21
+ "learning_rate": 8.38750262109457e-07,
22
+ "loss": 2.7456,
23
  "step": 200
24
  },
25
  {
26
+ "epoch": 0.050318685005031866,
27
+ "grad_norm": 0.06083838641643524,
28
+ "learning_rate": 1.2581253931641854e-06,
29
+ "loss": 2.7341,
30
  "step": 300
31
  },
32
  {
33
+ "epoch": 0.06709158000670916,
34
+ "grad_norm": 0.07314612716436386,
35
+ "learning_rate": 1.677500524218914e-06,
36
+ "loss": 2.7514,
37
  "step": 400
38
  },
39
  {
40
+ "epoch": 0.08386447500838645,
41
+ "grad_norm": 0.07996375858783722,
42
+ "learning_rate": 2.0968756552736426e-06,
43
+ "loss": 2.7325,
44
  "step": 500
45
  },
46
  {
47
+ "epoch": 0.10063737001006373,
48
+ "grad_norm": 0.09303133934736252,
49
+ "learning_rate": 2.516250786328371e-06,
50
+ "loss": 2.7265,
51
  "step": 600
52
  },
53
  {
54
+ "epoch": 0.11741026501174102,
55
+ "grad_norm": 0.11050993204116821,
56
+ "learning_rate": 2.9356259173830992e-06,
57
+ "loss": 2.7105,
58
  "step": 700
59
  },
60
  {
61
+ "epoch": 0.13418316001341832,
62
+ "grad_norm": 0.11374402791261673,
63
+ "learning_rate": 3.355001048437828e-06,
64
+ "loss": 2.6912,
65
  "step": 800
66
  },
67
  {
68
+ "epoch": 0.1509560550150956,
69
+ "grad_norm": 0.12549300491809845,
70
+ "learning_rate": 3.7743761794925563e-06,
71
+ "loss": 2.6951,
72
  "step": 900
73
  },
74
  {
75
+ "epoch": 0.1677289500167729,
76
+ "grad_norm": 0.1376326084136963,
77
+ "learning_rate": 4.193751310547285e-06,
78
+ "loss": 2.6876,
79
  "step": 1000
80
  },
81
  {
82
+ "epoch": 0.18450184501845018,
83
+ "grad_norm": 0.14521992206573486,
84
+ "learning_rate": 4.613126441602013e-06,
85
+ "loss": 2.6837,
86
  "step": 1100
87
  },
88
  {
89
+ "epoch": 0.20127474002012746,
90
+ "grad_norm": 0.1486833244562149,
91
+ "learning_rate": 5.032501572656742e-06,
92
+ "loss": 2.6621,
93
  "step": 1200
94
  },
95
  {
96
+ "epoch": 0.21804763502180477,
97
+ "grad_norm": 0.15669290721416473,
98
+ "learning_rate": 5.4518767037114705e-06,
99
+ "loss": 2.6594,
100
  "step": 1300
101
  },
102
  {
103
+ "epoch": 0.23482053002348205,
104
+ "grad_norm": 0.16500656306743622,
105
+ "learning_rate": 5.8712518347661985e-06,
106
+ "loss": 2.6575,
107
  "step": 1400
108
  },
109
  {
110
+ "epoch": 0.25159342502515936,
111
+ "grad_norm": 0.17107869684696198,
112
+ "learning_rate": 6.290626965820927e-06,
113
+ "loss": 2.6549,
114
  "step": 1500
115
  },
116
  {
117
+ "epoch": 0.26836632002683664,
118
+ "grad_norm": 0.19115546345710754,
119
+ "learning_rate": 6.710002096875656e-06,
120
+ "loss": 2.646,
121
  "step": 1600
122
  },
123
  {
124
+ "epoch": 0.2851392150285139,
125
+ "grad_norm": 0.20467370748519897,
126
+ "learning_rate": 7.129377227930384e-06,
127
+ "loss": 2.6481,
128
  "step": 1700
129
  },
130
+ {
131
+ "epoch": 0.3019121100301912,
132
+ "grad_norm": 0.19398543238639832,
133
+ "learning_rate": 7.548752358985113e-06,
134
+ "loss": 2.658,
135
+ "step": 1800
136
+ },
137
+ {
138
+ "epoch": 0.3186850050318685,
139
+ "grad_norm": 0.20218446850776672,
140
+ "learning_rate": 7.96812749003984e-06,
141
+ "loss": 2.6312,
142
+ "step": 1900
143
+ },
144
+ {
145
+ "epoch": 0.3354579000335458,
146
+ "grad_norm": 0.21208696067333221,
147
+ "learning_rate": 8.38750262109457e-06,
148
+ "loss": 2.6279,
149
+ "step": 2000
150
+ },
151
+ {
152
+ "epoch": 0.3522307950352231,
153
+ "grad_norm": 0.25312340259552,
154
+ "learning_rate": 8.806877752149298e-06,
155
+ "loss": 2.6271,
156
+ "step": 2100
157
+ },
158
+ {
159
+ "epoch": 0.36900369003690037,
160
+ "grad_norm": 0.21412670612335205,
161
+ "learning_rate": 9.226252883204026e-06,
162
+ "loss": 2.618,
163
+ "step": 2200
164
+ },
165
+ {
166
+ "epoch": 0.38577658503857765,
167
+ "grad_norm": 0.2197524607181549,
168
+ "learning_rate": 9.645628014258756e-06,
169
+ "loss": 2.6211,
170
+ "step": 2300
171
+ },
172
+ {
173
+ "epoch": 0.4025494800402549,
174
+ "grad_norm": 0.23409396409988403,
175
+ "learning_rate": 1.0065003145313484e-05,
176
+ "loss": 2.6159,
177
+ "step": 2400
178
+ },
179
+ {
180
+ "epoch": 0.41932237504193226,
181
+ "grad_norm": 0.24235695600509644,
182
+ "learning_rate": 1.0484378276368211e-05,
183
+ "loss": 2.6063,
184
+ "step": 2500
185
+ },
186
+ {
187
+ "epoch": 0.43609527004360954,
188
+ "grad_norm": 0.23461420834064484,
189
+ "learning_rate": 1.0903753407422941e-05,
190
+ "loss": 2.61,
191
+ "step": 2600
192
+ },
193
+ {
194
+ "epoch": 0.4528681650452868,
195
+ "grad_norm": 0.24338868260383606,
196
+ "learning_rate": 1.132312853847767e-05,
197
+ "loss": 2.577,
198
+ "step": 2700
199
+ },
200
+ {
201
+ "epoch": 0.4696410600469641,
202
+ "grad_norm": 0.26369062066078186,
203
+ "learning_rate": 1.1742503669532397e-05,
204
+ "loss": 2.5954,
205
+ "step": 2800
206
+ },
207
+ {
208
+ "epoch": 0.4864139550486414,
209
+ "grad_norm": 0.26913905143737793,
210
+ "learning_rate": 1.2161878800587127e-05,
211
+ "loss": 2.5754,
212
+ "step": 2900
213
+ },
214
+ {
215
+ "epoch": 0.5031868500503187,
216
+ "grad_norm": 0.2918139398097992,
217
+ "learning_rate": 1.2581253931641854e-05,
218
+ "loss": 2.577,
219
+ "step": 3000
220
+ },
221
+ {
222
+ "epoch": 0.5199597450519959,
223
+ "grad_norm": 0.2750920057296753,
224
+ "learning_rate": 1.3000629062696584e-05,
225
+ "loss": 2.578,
226
+ "step": 3100
227
+ },
228
+ {
229
+ "epoch": 0.5367326400536733,
230
+ "grad_norm": 0.2634315490722656,
231
+ "learning_rate": 1.3420004193751312e-05,
232
+ "loss": 2.5653,
233
+ "step": 3200
234
+ },
235
+ {
236
+ "epoch": 0.5535055350553506,
237
+ "grad_norm": 0.27983254194259644,
238
+ "learning_rate": 1.383937932480604e-05,
239
+ "loss": 2.5693,
240
+ "step": 3300
241
+ },
242
+ {
243
+ "epoch": 0.5702784300570278,
244
+ "grad_norm": 0.29475557804107666,
245
+ "learning_rate": 1.4258754455860768e-05,
246
+ "loss": 2.5698,
247
+ "step": 3400
248
+ },
249
+ {
250
+ "epoch": 0.5870513250587052,
251
+ "grad_norm": 0.29615071415901184,
252
+ "learning_rate": 1.4678129586915497e-05,
253
+ "loss": 2.5605,
254
+ "step": 3500
255
+ },
256
+ {
257
+ "epoch": 0.6038242200603824,
258
+ "grad_norm": 0.2656971216201782,
259
+ "learning_rate": 1.5097504717970225e-05,
260
+ "loss": 2.5599,
261
+ "step": 3600
262
+ },
263
+ {
264
+ "epoch": 0.6205971150620597,
265
+ "grad_norm": 0.2948819696903229,
266
+ "learning_rate": 1.5516879849024955e-05,
267
+ "loss": 2.5535,
268
+ "step": 3700
269
+ },
270
+ {
271
+ "epoch": 0.637370010063737,
272
+ "grad_norm": 0.3076747953891754,
273
+ "learning_rate": 1.593625498007968e-05,
274
+ "loss": 2.5525,
275
+ "step": 3800
276
+ },
277
+ {
278
+ "epoch": 0.6541429050654143,
279
+ "grad_norm": 0.31171539425849915,
280
+ "learning_rate": 1.635563011113441e-05,
281
+ "loss": 2.5563,
282
+ "step": 3900
283
+ },
284
+ {
285
+ "epoch": 0.6709158000670916,
286
+ "grad_norm": 0.3256972134113312,
287
+ "learning_rate": 1.677500524218914e-05,
288
+ "loss": 2.5491,
289
+ "step": 4000
290
+ },
291
+ {
292
+ "epoch": 0.6876886950687688,
293
+ "grad_norm": 0.2948451340198517,
294
+ "learning_rate": 1.719438037324387e-05,
295
+ "loss": 2.5493,
296
+ "step": 4100
297
+ },
298
+ {
299
+ "epoch": 0.7044615900704462,
300
+ "grad_norm": 0.29186537861824036,
301
+ "learning_rate": 1.7613755504298596e-05,
302
+ "loss": 2.5316,
303
+ "step": 4200
304
+ },
305
+ {
306
+ "epoch": 0.7212344850721234,
307
+ "grad_norm": 0.3038988411426544,
308
+ "learning_rate": 1.8033130635353326e-05,
309
+ "loss": 2.5322,
310
+ "step": 4300
311
+ },
312
+ {
313
+ "epoch": 0.7380073800738007,
314
+ "grad_norm": 0.32824885845184326,
315
+ "learning_rate": 1.8452505766408052e-05,
316
+ "loss": 2.5196,
317
+ "step": 4400
318
+ },
319
+ {
320
+ "epoch": 0.7547802750754781,
321
+ "grad_norm": 0.38570263981819153,
322
+ "learning_rate": 1.887188089746278e-05,
323
+ "loss": 2.5375,
324
+ "step": 4500
325
+ },
326
+ {
327
+ "epoch": 0.7715531700771553,
328
+ "grad_norm": 0.32352355122566223,
329
+ "learning_rate": 1.929125602851751e-05,
330
+ "loss": 2.5231,
331
+ "step": 4600
332
+ },
333
+ {
334
+ "epoch": 0.7883260650788326,
335
+ "grad_norm": 0.32961755990982056,
336
+ "learning_rate": 1.971063115957224e-05,
337
+ "loss": 2.5256,
338
+ "step": 4700
339
+ },
340
+ {
341
+ "epoch": 0.8050989600805099,
342
+ "grad_norm": 0.3353755474090576,
343
+ "learning_rate": 1.9966697943413548e-05,
344
+ "loss": 2.5265,
345
+ "step": 4800
346
+ },
347
+ {
348
+ "epoch": 0.8218718550821872,
349
+ "grad_norm": 0.3189639747142792,
350
+ "learning_rate": 1.941085760330316e-05,
351
+ "loss": 2.5308,
352
+ "step": 4900
353
+ },
354
+ {
355
+ "epoch": 0.8386447500838645,
356
+ "grad_norm": 0.3028644323348999,
357
+ "learning_rate": 1.8206178320980295e-05,
358
+ "loss": 2.5113,
359
+ "step": 5000
360
+ },
361
+ {
362
+ "epoch": 0.8554176450855417,
363
+ "grad_norm": 0.3632523715496063,
364
+ "learning_rate": 1.6435717652323097e-05,
365
+ "loss": 2.5153,
366
+ "step": 5100
367
+ },
368
+ {
369
+ "epoch": 0.8721905400872191,
370
+ "grad_norm": 0.3340721130371094,
371
+ "learning_rate": 1.4221541393659966e-05,
372
+ "loss": 2.5175,
373
+ "step": 5200
374
+ },
375
+ {
376
+ "epoch": 0.8889634350888963,
377
+ "grad_norm": 0.3549550175666809,
378
+ "learning_rate": 1.171630765991538e-05,
379
+ "loss": 2.5159,
380
+ "step": 5300
381
+ },
382
+ {
383
+ "epoch": 0.9057363300905736,
384
+ "grad_norm": 0.3556847870349884,
385
+ "learning_rate": 9.092741751409186e-06,
386
+ "loss": 2.5052,
387
+ "step": 5400
388
+ },
389
+ {
390
+ "epoch": 0.922509225092251,
391
+ "grad_norm": 0.33662235736846924,
392
+ "learning_rate": 6.531727472854617e-06,
393
+ "loss": 2.4994,
394
+ "step": 5500
395
+ },
396
+ {
397
+ "epoch": 0.9392821200939282,
398
+ "grad_norm": 0.3705320954322815,
399
+ "learning_rate": 4.209835957886196e-06,
400
+ "loss": 2.5104,
401
+ "step": 5600
402
+ },
403
+ {
404
+ "epoch": 0.9560550150956055,
405
+ "grad_norm": 0.3348437249660492,
406
+ "learning_rate": 2.287151834070226e-06,
407
+ "loss": 2.4979,
408
+ "step": 5700
409
+ },
410
+ {
411
+ "epoch": 0.9728279100972828,
412
+ "grad_norm": 0.3829787075519562,
413
+ "learning_rate": 8.962360629781164e-07,
414
+ "loss": 2.519,
415
+ "step": 5800
416
+ },
417
+ {
418
+ "epoch": 0.9896008050989601,
419
+ "grad_norm": 0.35299184918403625,
420
+ "learning_rate": 1.329864209512377e-07,
421
+ "loss": 2.5051,
422
+ "step": 5900
423
+ },
424
  {
425
  "epoch": 1.0,
426
+ "step": 5962,
427
+ "total_flos": 8.687706333480223e+17,
428
+ "train_loss": 2.596681281969236,
429
+ "train_runtime": 1897.3372,
430
+ "train_samples_per_second": 50.272,
431
+ "train_steps_per_second": 3.142
432
  }
433
  ],
434
  "logging_steps": 100,
435
+ "max_steps": 5962,
436
  "num_input_tokens_seen": 0,
437
  "num_train_epochs": 1,
438
  "save_steps": 500,
 
448
  "attributes": {}
449
  }
450
  },
451
+ "total_flos": 8.687706333480223e+17,
452
  "train_batch_size": 2,
453
  "trial_name": null,
454
  "trial_params": null