gbemilekeonilude commited on
Commit
7ed552e
·
verified ·
1 Parent(s): 8985752

End of training

Browse files
Files changed (5) hide show
  1. README.md +1 -1
  2. all_results.json +16 -0
  3. eval_results.json +10 -0
  4. train_results.json +10 -0
  5. trainer_state.json +477 -0
README.md CHANGED
@@ -20,7 +20,7 @@ This model is a fine-tuned version of [EleutherAI/pythia-410m](https://huggingfa
20
  It achieves the following results on the evaluation set:
21
  - Loss: 1.6227
22
  - Accuracy: 0.1571
23
- - Num Input Tokens Seen: 1638400
24
 
25
  ## Model description
26
 
 
20
  It achieves the following results on the evaluation set:
21
  - Loss: 1.6227
22
  - Accuracy: 0.1571
23
+ - Num Input Tokens Seen: 1941504
24
 
25
  ## Model description
26
 
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.15714285714285714,
4
+ "eval_loss": 1.6226980686187744,
5
+ "eval_runtime": 0.3685,
6
+ "eval_samples": 70,
7
+ "eval_samples_per_second": 189.95,
8
+ "eval_steps_per_second": 24.422,
9
+ "num_input_tokens_seen": 1941504,
10
+ "total_flos": 3521692676653056.0,
11
+ "train_loss": 1.8215426835329724,
12
+ "train_runtime": 107.3388,
13
+ "train_samples": 628,
14
+ "train_samples_per_second": 17.552,
15
+ "train_steps_per_second": 2.208
16
+ }
eval_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.15714285714285714,
4
+ "eval_loss": 1.6226980686187744,
5
+ "eval_runtime": 0.3685,
6
+ "eval_samples": 70,
7
+ "eval_samples_per_second": 189.95,
8
+ "eval_steps_per_second": 24.422,
9
+ "num_input_tokens_seen": 1941504
10
+ }
train_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "num_input_tokens_seen": 1941504,
4
+ "total_flos": 3521692676653056.0,
5
+ "train_loss": 1.8215426835329724,
6
+ "train_runtime": 107.3388,
7
+ "train_samples": 628,
8
+ "train_samples_per_second": 17.552,
9
+ "train_steps_per_second": 2.208
10
+ }
trainer_state.json ADDED
@@ -0,0 +1,477 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.6226980686187744,
3
+ "best_model_checkpoint": "/data/user_data/gonilude/cpp_and_text_pythia_410m/checkpoint-200",
4
+ "epoch": 3.0,
5
+ "eval_steps": 50,
6
+ "global_step": 237,
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,
13
+ "eval_accuracy": 0.3,
14
+ "eval_loss": 2.5906739234924316,
15
+ "eval_runtime": 0.536,
16
+ "eval_samples_per_second": 130.596,
17
+ "eval_steps_per_second": 16.791,
18
+ "num_input_tokens_seen": 0,
19
+ "step": 0
20
+ },
21
+ {
22
+ "epoch": 0.012658227848101266,
23
+ "grad_norm": NaN,
24
+ "learning_rate": 0.0,
25
+ "loss": 2.2691,
26
+ "num_input_tokens_seen": 8192,
27
+ "step": 1
28
+ },
29
+ {
30
+ "epoch": 0.06329113924050633,
31
+ "grad_norm": Infinity,
32
+ "learning_rate": 0.0,
33
+ "loss": 2.9877,
34
+ "num_input_tokens_seen": 40960,
35
+ "step": 5
36
+ },
37
+ {
38
+ "epoch": 0.12658227848101267,
39
+ "grad_norm": Infinity,
40
+ "learning_rate": 1e-05,
41
+ "loss": 2.2275,
42
+ "num_input_tokens_seen": 81920,
43
+ "step": 10
44
+ },
45
+ {
46
+ "epoch": 0.189873417721519,
47
+ "grad_norm": 274.08111572265625,
48
+ "learning_rate": 1.9999058994907564e-05,
49
+ "loss": 3.2099,
50
+ "num_input_tokens_seen": 122880,
51
+ "step": 15
52
+ },
53
+ {
54
+ "epoch": 0.25316455696202533,
55
+ "grad_norm": 166.55935668945312,
56
+ "learning_rate": 1.99661424082419e-05,
57
+ "loss": 2.5884,
58
+ "num_input_tokens_seen": 163840,
59
+ "step": 20
60
+ },
61
+ {
62
+ "epoch": 0.31645569620253167,
63
+ "grad_norm": 110.16832733154297,
64
+ "learning_rate": 1.9886352515311134e-05,
65
+ "loss": 2.4323,
66
+ "num_input_tokens_seen": 204800,
67
+ "step": 25
68
+ },
69
+ {
70
+ "epoch": 0.379746835443038,
71
+ "grad_norm": 82.22343444824219,
72
+ "learning_rate": 1.9760064588305347e-05,
73
+ "loss": 2.3275,
74
+ "num_input_tokens_seen": 245760,
75
+ "step": 30
76
+ },
77
+ {
78
+ "epoch": 0.4430379746835443,
79
+ "grad_norm": 56.37868118286133,
80
+ "learning_rate": 1.9587872591512583e-05,
81
+ "loss": 1.8227,
82
+ "num_input_tokens_seen": 286720,
83
+ "step": 35
84
+ },
85
+ {
86
+ "epoch": 0.5063291139240507,
87
+ "grad_norm": 54.27399444580078,
88
+ "learning_rate": 1.9370586387753532e-05,
89
+ "loss": 1.998,
90
+ "num_input_tokens_seen": 327680,
91
+ "step": 40
92
+ },
93
+ {
94
+ "epoch": 0.569620253164557,
95
+ "grad_norm": 44.79275894165039,
96
+ "learning_rate": 1.9109227929390378e-05,
97
+ "loss": 1.972,
98
+ "num_input_tokens_seen": 368640,
99
+ "step": 45
100
+ },
101
+ {
102
+ "epoch": 0.6329113924050633,
103
+ "grad_norm": 46.56148910522461,
104
+ "learning_rate": 1.8805026451824547e-05,
105
+ "loss": 1.9728,
106
+ "num_input_tokens_seen": 409600,
107
+ "step": 50
108
+ },
109
+ {
110
+ "epoch": 0.6329113924050633,
111
+ "eval_accuracy": 0.18571428571428572,
112
+ "eval_loss": 2.392421245574951,
113
+ "eval_runtime": 0.367,
114
+ "eval_samples_per_second": 190.718,
115
+ "eval_steps_per_second": 24.521,
116
+ "num_input_tokens_seen": 409600,
117
+ "step": 50
118
+ },
119
+ {
120
+ "epoch": 0.6962025316455697,
121
+ "grad_norm": 35.83555221557617,
122
+ "learning_rate": 1.8459412692089497e-05,
123
+ "loss": 2.219,
124
+ "num_input_tokens_seen": 450560,
125
+ "step": 55
126
+ },
127
+ {
128
+ "epoch": 0.759493670886076,
129
+ "grad_norm": 33.77033996582031,
130
+ "learning_rate": 1.8074012159730034e-05,
131
+ "loss": 1.8699,
132
+ "num_input_tokens_seen": 491520,
133
+ "step": 60
134
+ },
135
+ {
136
+ "epoch": 0.8227848101265823,
137
+ "grad_norm": 51.212223052978516,
138
+ "learning_rate": 1.765063749161688e-05,
139
+ "loss": 1.9769,
140
+ "num_input_tokens_seen": 532480,
141
+ "step": 65
142
+ },
143
+ {
144
+ "epoch": 0.8860759493670886,
145
+ "grad_norm": 25.946151733398438,
146
+ "learning_rate": 1.719127992665376e-05,
147
+ "loss": 1.8649,
148
+ "num_input_tokens_seen": 573440,
149
+ "step": 70
150
+ },
151
+ {
152
+ "epoch": 0.9493670886075949,
153
+ "grad_norm": 34.178245544433594,
154
+ "learning_rate": 1.6698099940473644e-05,
155
+ "loss": 1.8005,
156
+ "num_input_tokens_seen": 614400,
157
+ "step": 75
158
+ },
159
+ {
160
+ "epoch": 1.0126582278481013,
161
+ "grad_norm": 26.01968765258789,
162
+ "learning_rate": 1.6173417084171537e-05,
163
+ "loss": 1.6863,
164
+ "num_input_tokens_seen": 655360,
165
+ "step": 80
166
+ },
167
+ {
168
+ "epoch": 1.0759493670886076,
169
+ "grad_norm": 17.587852478027344,
170
+ "learning_rate": 1.5619699074864864e-05,
171
+ "loss": 1.6306,
172
+ "num_input_tokens_seen": 696320,
173
+ "step": 85
174
+ },
175
+ {
176
+ "epoch": 1.139240506329114,
177
+ "grad_norm": 25.147842407226562,
178
+ "learning_rate": 1.50395501893913e-05,
179
+ "loss": 1.5408,
180
+ "num_input_tokens_seen": 737280,
181
+ "step": 90
182
+ },
183
+ {
184
+ "epoch": 1.2025316455696202,
185
+ "grad_norm": 32.01517868041992,
186
+ "learning_rate": 1.4435699015731449e-05,
187
+ "loss": 1.9078,
188
+ "num_input_tokens_seen": 778240,
189
+ "step": 95
190
+ },
191
+ {
192
+ "epoch": 1.2658227848101267,
193
+ "grad_norm": 22.242891311645508,
194
+ "learning_rate": 1.3810985619764573e-05,
195
+ "loss": 1.6016,
196
+ "num_input_tokens_seen": 819200,
197
+ "step": 100
198
+ },
199
+ {
200
+ "epoch": 1.2658227848101267,
201
+ "eval_accuracy": 0.14285714285714285,
202
+ "eval_loss": 1.9393833875656128,
203
+ "eval_runtime": 0.4244,
204
+ "eval_samples_per_second": 164.94,
205
+ "eval_steps_per_second": 21.207,
206
+ "num_input_tokens_seen": 819200,
207
+ "step": 100
208
+ },
209
+ {
210
+ "epoch": 1.3291139240506329,
211
+ "grad_norm": 36.45238494873047,
212
+ "learning_rate": 1.3168348187715353e-05,
213
+ "loss": 1.7416,
214
+ "num_input_tokens_seen": 860160,
215
+ "step": 105
216
+ },
217
+ {
218
+ "epoch": 1.3924050632911391,
219
+ "grad_norm": 14.048996925354004,
220
+ "learning_rate": 1.2510809207115666e-05,
221
+ "loss": 1.6521,
222
+ "num_input_tokens_seen": 901120,
223
+ "step": 110
224
+ },
225
+ {
226
+ "epoch": 1.4556962025316456,
227
+ "grad_norm": 52.22154235839844,
228
+ "learning_rate": 1.1841461251275868e-05,
229
+ "loss": 1.7971,
230
+ "num_input_tokens_seen": 942080,
231
+ "step": 115
232
+ },
233
+ {
234
+ "epoch": 1.518987341772152,
235
+ "grad_norm": 48.804534912109375,
236
+ "learning_rate": 1.1163452434124773e-05,
237
+ "loss": 1.7401,
238
+ "num_input_tokens_seen": 983040,
239
+ "step": 120
240
+ },
241
+ {
242
+ "epoch": 1.5822784810126582,
243
+ "grad_norm": 31.043750762939453,
244
+ "learning_rate": 1.0479971603828001e-05,
245
+ "loss": 1.6436,
246
+ "num_input_tokens_seen": 1024000,
247
+ "step": 125
248
+ },
249
+ {
250
+ "epoch": 1.6455696202531644,
251
+ "grad_norm": 28.90459632873535,
252
+ "learning_rate": 9.79423334482279e-06,
253
+ "loss": 1.6048,
254
+ "num_input_tokens_seen": 1064960,
255
+ "step": 130
256
+ },
257
+ {
258
+ "epoch": 1.7088607594936709,
259
+ "grad_norm": 33.064090728759766,
260
+ "learning_rate": 9.109462858808586e-06,
261
+ "loss": 1.6833,
262
+ "num_input_tokens_seen": 1105920,
263
+ "step": 135
264
+ },
265
+ {
266
+ "epoch": 1.7721518987341773,
267
+ "grad_norm": 37.580997467041016,
268
+ "learning_rate": 8.428880795801965e-06,
269
+ "loss": 1.5788,
270
+ "num_input_tokens_seen": 1146880,
271
+ "step": 140
272
+ },
273
+ {
274
+ "epoch": 1.8354430379746836,
275
+ "grad_norm": 36.66419219970703,
276
+ "learning_rate": 7.75568810659924e-06,
277
+ "loss": 1.7446,
278
+ "num_input_tokens_seen": 1187840,
279
+ "step": 145
280
+ },
281
+ {
282
+ "epoch": 1.8987341772151898,
283
+ "grad_norm": 35.965087890625,
284
+ "learning_rate": 7.093050987889547e-06,
285
+ "loss": 1.7223,
286
+ "num_input_tokens_seen": 1228800,
287
+ "step": 150
288
+ },
289
+ {
290
+ "epoch": 1.8987341772151898,
291
+ "eval_accuracy": 0.14285714285714285,
292
+ "eval_loss": 1.6300362348556519,
293
+ "eval_runtime": 0.3719,
294
+ "eval_samples_per_second": 188.224,
295
+ "eval_steps_per_second": 24.2,
296
+ "num_input_tokens_seen": 1228800,
297
+ "step": 150
298
+ },
299
+ {
300
+ "epoch": 1.9620253164556962,
301
+ "grad_norm": 36.96841812133789,
302
+ "learning_rate": 6.444085990825338e-06,
303
+ "loss": 1.628,
304
+ "num_input_tokens_seen": 1269760,
305
+ "step": 155
306
+ },
307
+ {
308
+ "epoch": 2.0253164556962027,
309
+ "grad_norm": 20.582347869873047,
310
+ "learning_rate": 5.811845363088477e-06,
311
+ "loss": 1.6389,
312
+ "num_input_tokens_seen": 1310720,
313
+ "step": 160
314
+ },
315
+ {
316
+ "epoch": 2.088607594936709,
317
+ "grad_norm": 40.09123229980469,
318
+ "learning_rate": 5.199302693391958e-06,
319
+ "loss": 1.5496,
320
+ "num_input_tokens_seen": 1351680,
321
+ "step": 165
322
+ },
323
+ {
324
+ "epoch": 2.151898734177215,
325
+ "grad_norm": 21.198436737060547,
326
+ "learning_rate": 4.609338925934743e-06,
327
+ "loss": 1.5789,
328
+ "num_input_tokens_seen": 1392640,
329
+ "step": 170
330
+ },
331
+ {
332
+ "epoch": 2.2151898734177213,
333
+ "grad_norm": 42.79294967651367,
334
+ "learning_rate": 4.044728810587406e-06,
335
+ "loss": 1.7237,
336
+ "num_input_tokens_seen": 1433600,
337
+ "step": 175
338
+ },
339
+ {
340
+ "epoch": 2.278481012658228,
341
+ "grad_norm": 25.977487564086914,
342
+ "learning_rate": 3.508127852536698e-06,
343
+ "loss": 1.6412,
344
+ "num_input_tokens_seen": 1474560,
345
+ "step": 180
346
+ },
347
+ {
348
+ "epoch": 2.3417721518987342,
349
+ "grad_norm": 12.27081298828125,
350
+ "learning_rate": 3.0020598227682794e-06,
351
+ "loss": 1.5877,
352
+ "num_input_tokens_seen": 1515520,
353
+ "step": 185
354
+ },
355
+ {
356
+ "epoch": 2.4050632911392404,
357
+ "grad_norm": 29.93262481689453,
358
+ "learning_rate": 2.5289048881289256e-06,
359
+ "loss": 1.5655,
360
+ "num_input_tokens_seen": 1556480,
361
+ "step": 190
362
+ },
363
+ {
364
+ "epoch": 2.4683544303797467,
365
+ "grad_norm": 11.503835678100586,
366
+ "learning_rate": 2.090888416795582e-06,
367
+ "loss": 1.6816,
368
+ "num_input_tokens_seen": 1597440,
369
+ "step": 195
370
+ },
371
+ {
372
+ "epoch": 2.5316455696202533,
373
+ "grad_norm": 32.70045852661133,
374
+ "learning_rate": 1.69007051180199e-06,
375
+ "loss": 1.6031,
376
+ "num_input_tokens_seen": 1638400,
377
+ "step": 200
378
+ },
379
+ {
380
+ "epoch": 2.5316455696202533,
381
+ "eval_accuracy": 0.15714285714285714,
382
+ "eval_loss": 1.6226980686187744,
383
+ "eval_runtime": 0.3713,
384
+ "eval_samples_per_second": 188.508,
385
+ "eval_steps_per_second": 24.237,
386
+ "num_input_tokens_seen": 1638400,
387
+ "step": 200
388
+ },
389
+ {
390
+ "epoch": 2.5949367088607596,
391
+ "grad_norm": 25.854286193847656,
392
+ "learning_rate": 1.3283363218493962e-06,
393
+ "loss": 1.5904,
394
+ "num_input_tokens_seen": 1679360,
395
+ "step": 205
396
+ },
397
+ {
398
+ "epoch": 2.6582278481012658,
399
+ "grad_norm": 19.382221221923828,
400
+ "learning_rate": 1.0073871749720221e-06,
401
+ "loss": 1.5788,
402
+ "num_input_tokens_seen": 1720320,
403
+ "step": 210
404
+ },
405
+ {
406
+ "epoch": 2.721518987341772,
407
+ "grad_norm": 22.248796463012695,
408
+ "learning_rate": 7.287325767579756e-07,
409
+ "loss": 1.6045,
410
+ "num_input_tokens_seen": 1761280,
411
+ "step": 215
412
+ },
413
+ {
414
+ "epoch": 2.7848101265822782,
415
+ "grad_norm": 23.392297744750977,
416
+ "learning_rate": 4.936831107599749e-07,
417
+ "loss": 1.5288,
418
+ "num_input_tokens_seen": 1802240,
419
+ "step": 220
420
+ },
421
+ {
422
+ "epoch": 2.848101265822785,
423
+ "grad_norm": 22.441410064697266,
424
+ "learning_rate": 3.033442744870685e-07,
425
+ "loss": 1.5869,
426
+ "num_input_tokens_seen": 1843200,
427
+ "step": 225
428
+ },
429
+ {
430
+ "epoch": 2.911392405063291,
431
+ "grad_norm": 31.935455322265625,
432
+ "learning_rate": 1.5861127996827597e-07,
433
+ "loss": 1.5976,
434
+ "num_input_tokens_seen": 1884160,
435
+ "step": 230
436
+ },
437
+ {
438
+ "epoch": 2.9746835443037973,
439
+ "grad_norm": 13.112102508544922,
440
+ "learning_rate": 6.016484334238515e-08,
441
+ "loss": 1.613,
442
+ "num_input_tokens_seen": 1925120,
443
+ "step": 235
444
+ },
445
+ {
446
+ "epoch": 3.0,
447
+ "num_input_tokens_seen": 1941504,
448
+ "step": 237,
449
+ "total_flos": 3521692676653056.0,
450
+ "train_loss": 1.8215426835329724,
451
+ "train_runtime": 107.3388,
452
+ "train_samples_per_second": 17.552,
453
+ "train_steps_per_second": 2.208
454
+ }
455
+ ],
456
+ "logging_steps": 5,
457
+ "max_steps": 237,
458
+ "num_input_tokens_seen": 1941504,
459
+ "num_train_epochs": 3,
460
+ "save_steps": 50,
461
+ "stateful_callbacks": {
462
+ "TrainerControl": {
463
+ "args": {
464
+ "should_epoch_stop": false,
465
+ "should_evaluate": false,
466
+ "should_log": false,
467
+ "should_save": true,
468
+ "should_training_stop": true
469
+ },
470
+ "attributes": {}
471
+ }
472
+ },
473
+ "total_flos": 3521692676653056.0,
474
+ "train_batch_size": 4,
475
+ "trial_name": null,
476
+ "trial_params": null
477
+ }