gbemilekeonilude commited on
Commit
2057512
·
verified ·
1 Parent(s): cb16fe9

End of training

Browse files
Files changed (5) hide show
  1. README.md +3 -3
  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
@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [EleutherAI/pythia-70m](https://huggingface.co/EleutherAI/pythia-70m) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 1.6633
22
- - Accuracy: 0.1972
23
- - Num Input Tokens Seen: 1638400
24
 
25
  ## Model description
26
 
 
18
 
19
  This model is a fine-tuned version of [EleutherAI/pythia-70m](https://huggingface.co/EleutherAI/pythia-70m) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 1.6595
22
+ - Accuracy: 0.2113
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.2112676056338028,
4
+ "eval_loss": 1.6594650745391846,
5
+ "eval_runtime": 0.0959,
6
+ "eval_samples": 71,
7
+ "eval_samples_per_second": 740.562,
8
+ "eval_steps_per_second": 93.874,
9
+ "num_input_tokens_seen": 1941504,
10
+ "total_flos": 220374931341312.0,
11
+ "train_loss": 2.1441552528349157,
12
+ "train_runtime": 22.4706,
13
+ "train_samples": 631,
14
+ "train_samples_per_second": 84.243,
15
+ "train_steps_per_second": 10.547
16
+ }
eval_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.2112676056338028,
4
+ "eval_loss": 1.6594650745391846,
5
+ "eval_runtime": 0.0959,
6
+ "eval_samples": 71,
7
+ "eval_samples_per_second": 740.562,
8
+ "eval_steps_per_second": 93.874,
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": 220374931341312.0,
5
+ "train_loss": 2.1441552528349157,
6
+ "train_runtime": 22.4706,
7
+ "train_samples": 631,
8
+ "train_samples_per_second": 84.243,
9
+ "train_steps_per_second": 10.547
10
+ }
trainer_state.json ADDED
@@ -0,0 +1,477 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.6594650745391846,
3
+ "best_model_checkpoint": "/data/user_data/gonilude/python_and_text_pythia_70m/checkpoint-150",
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.22535211267605634,
14
+ "eval_loss": 7.193855285644531,
15
+ "eval_runtime": 0.2991,
16
+ "eval_samples_per_second": 237.392,
17
+ "eval_steps_per_second": 30.092,
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": 8.2293,
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": 8.6977,
34
+ "num_input_tokens_seen": 40960,
35
+ "step": 5
36
+ },
37
+ {
38
+ "epoch": 0.12658227848101267,
39
+ "grad_norm": 434.0960388183594,
40
+ "learning_rate": 1e-05,
41
+ "loss": 8.1873,
42
+ "num_input_tokens_seen": 81920,
43
+ "step": 10
44
+ },
45
+ {
46
+ "epoch": 0.189873417721519,
47
+ "grad_norm": 457.978271484375,
48
+ "learning_rate": 2e-05,
49
+ "loss": 5.5601,
50
+ "num_input_tokens_seen": 122880,
51
+ "step": 15
52
+ },
53
+ {
54
+ "epoch": 0.25316455696202533,
55
+ "grad_norm": 342.55340576171875,
56
+ "learning_rate": 1.9976483726428423e-05,
57
+ "loss": 3.0969,
58
+ "num_input_tokens_seen": 163840,
59
+ "step": 20
60
+ },
61
+ {
62
+ "epoch": 0.31645569620253167,
63
+ "grad_norm": 1078.478271484375,
64
+ "learning_rate": 1.9939835156657616e-05,
65
+ "loss": 2.4498,
66
+ "num_input_tokens_seen": 204800,
67
+ "step": 25
68
+ },
69
+ {
70
+ "epoch": 0.379746835443038,
71
+ "grad_norm": 316.6920166015625,
72
+ "learning_rate": 1.9841388720031727e-05,
73
+ "loss": 1.9535,
74
+ "num_input_tokens_seen": 245760,
75
+ "step": 30
76
+ },
77
+ {
78
+ "epoch": 0.4430379746835443,
79
+ "grad_norm": 692.9888916015625,
80
+ "learning_rate": 1.9729275686705832e-05,
81
+ "loss": 2.7533,
82
+ "num_input_tokens_seen": 286720,
83
+ "step": 35
84
+ },
85
+ {
86
+ "epoch": 0.5063291139240507,
87
+ "grad_norm": 223.78614807128906,
88
+ "learning_rate": 1.9547993211399753e-05,
89
+ "loss": 1.9428,
90
+ "num_input_tokens_seen": 327680,
91
+ "step": 40
92
+ },
93
+ {
94
+ "epoch": 0.569620253164557,
95
+ "grad_norm": 218.17137145996094,
96
+ "learning_rate": 1.932180409200991e-05,
97
+ "loss": 2.6774,
98
+ "num_input_tokens_seen": 368640,
99
+ "step": 45
100
+ },
101
+ {
102
+ "epoch": 0.6329113924050633,
103
+ "grad_norm": 219.60426330566406,
104
+ "learning_rate": 1.905177215357839e-05,
105
+ "loss": 1.7511,
106
+ "num_input_tokens_seen": 409600,
107
+ "step": 50
108
+ },
109
+ {
110
+ "epoch": 0.6329113924050633,
111
+ "eval_accuracy": 0.14084507042253522,
112
+ "eval_loss": 1.9312864542007446,
113
+ "eval_runtime": 0.1429,
114
+ "eval_samples_per_second": 496.992,
115
+ "eval_steps_per_second": 62.999,
116
+ "num_input_tokens_seen": 409600,
117
+ "step": 50
118
+ },
119
+ {
120
+ "epoch": 0.6962025316455697,
121
+ "grad_norm": 167.01634216308594,
122
+ "learning_rate": 1.8739167425092644e-05,
123
+ "loss": 1.8037,
124
+ "num_input_tokens_seen": 450560,
125
+ "step": 55
126
+ },
127
+ {
128
+ "epoch": 0.759493670886076,
129
+ "grad_norm": 194.86444091796875,
130
+ "learning_rate": 1.838546016621564e-05,
131
+ "loss": 1.7087,
132
+ "num_input_tokens_seen": 491520,
133
+ "step": 60
134
+ },
135
+ {
136
+ "epoch": 0.8227848101265823,
137
+ "grad_norm": 182.9615478515625,
138
+ "learning_rate": 1.7992313952280175e-05,
139
+ "loss": 1.8595,
140
+ "num_input_tokens_seen": 532480,
141
+ "step": 65
142
+ },
143
+ {
144
+ "epoch": 0.8860759493670886,
145
+ "grad_norm": 387.1624450683594,
146
+ "learning_rate": 1.7561577850070355e-05,
147
+ "loss": 1.7605,
148
+ "num_input_tokens_seen": 573440,
149
+ "step": 70
150
+ },
151
+ {
152
+ "epoch": 0.9493670886075949,
153
+ "grad_norm": 115.52404022216797,
154
+ "learning_rate": 1.709527772118953e-05,
155
+ "loss": 1.8325,
156
+ "num_input_tokens_seen": 614400,
157
+ "step": 75
158
+ },
159
+ {
160
+ "epoch": 1.0126582278481013,
161
+ "grad_norm": 225.36166381835938,
162
+ "learning_rate": 1.659560669391714e-05,
163
+ "loss": 1.805,
164
+ "num_input_tokens_seen": 655360,
165
+ "step": 80
166
+ },
167
+ {
168
+ "epoch": 1.0759493670886076,
169
+ "grad_norm": 363.2311096191406,
170
+ "learning_rate": 1.6064914848367818e-05,
171
+ "loss": 2.0046,
172
+ "num_input_tokens_seen": 696320,
173
+ "step": 85
174
+ },
175
+ {
176
+ "epoch": 1.139240506329114,
177
+ "grad_norm": 244.0623016357422,
178
+ "learning_rate": 1.5505698163465986e-05,
179
+ "loss": 1.7745,
180
+ "num_input_tokens_seen": 737280,
181
+ "step": 90
182
+ },
183
+ {
184
+ "epoch": 1.2025316455696202,
185
+ "grad_norm": 201.84446716308594,
186
+ "learning_rate": 1.4920586777721231e-05,
187
+ "loss": 1.7372,
188
+ "num_input_tokens_seen": 778240,
189
+ "step": 95
190
+ },
191
+ {
192
+ "epoch": 1.2658227848101267,
193
+ "grad_norm": 164.17694091796875,
194
+ "learning_rate": 1.4312332619016964e-05,
195
+ "loss": 1.8156,
196
+ "num_input_tokens_seen": 819200,
197
+ "step": 100
198
+ },
199
+ {
200
+ "epoch": 1.2658227848101267,
201
+ "eval_accuracy": 0.22535211267605634,
202
+ "eval_loss": 1.7096410989761353,
203
+ "eval_runtime": 0.1476,
204
+ "eval_samples_per_second": 481.143,
205
+ "eval_steps_per_second": 60.99,
206
+ "num_input_tokens_seen": 819200,
207
+ "step": 100
208
+ },
209
+ {
210
+ "epoch": 1.3291139240506329,
211
+ "grad_norm": 234.53009033203125,
212
+ "learning_rate": 1.3683796461592604e-05,
213
+ "loss": 1.7591,
214
+ "num_input_tokens_seen": 860160,
215
+ "step": 105
216
+ },
217
+ {
218
+ "epoch": 1.3924050632911391,
219
+ "grad_norm": 189.70046997070312,
220
+ "learning_rate": 1.3037934471093683e-05,
221
+ "loss": 1.6518,
222
+ "num_input_tokens_seen": 901120,
223
+ "step": 110
224
+ },
225
+ {
226
+ "epoch": 1.4556962025316456,
227
+ "grad_norm": 167.0009002685547,
228
+ "learning_rate": 1.2377784300971807e-05,
229
+ "loss": 1.7383,
230
+ "num_input_tokens_seen": 942080,
231
+ "step": 115
232
+ },
233
+ {
234
+ "epoch": 1.518987341772152,
235
+ "grad_norm": 210.74073791503906,
236
+ "learning_rate": 1.1706450805626762e-05,
237
+ "loss": 1.6693,
238
+ "num_input_tokens_seen": 983040,
239
+ "step": 120
240
+ },
241
+ {
242
+ "epoch": 1.5822784810126582,
243
+ "grad_norm": 113.87993621826172,
244
+ "learning_rate": 1.1027091437485404e-05,
245
+ "loss": 1.5721,
246
+ "num_input_tokens_seen": 1024000,
247
+ "step": 125
248
+ },
249
+ {
250
+ "epoch": 1.6455696202531644,
251
+ "grad_norm": 175.82958984375,
252
+ "learning_rate": 1.0342901396698658e-05,
253
+ "loss": 1.8908,
254
+ "num_input_tokens_seen": 1064960,
255
+ "step": 130
256
+ },
257
+ {
258
+ "epoch": 1.7088607594936709,
259
+ "grad_norm": 94.99879455566406,
260
+ "learning_rate": 9.657098603301347e-06,
261
+ "loss": 1.7056,
262
+ "num_input_tokens_seen": 1105920,
263
+ "step": 135
264
+ },
265
+ {
266
+ "epoch": 1.7721518987341773,
267
+ "grad_norm": 142.98709106445312,
268
+ "learning_rate": 8.9729085625146e-06,
269
+ "loss": 1.62,
270
+ "num_input_tokens_seen": 1146880,
271
+ "step": 140
272
+ },
273
+ {
274
+ "epoch": 1.8354430379746836,
275
+ "grad_norm": 149.57101440429688,
276
+ "learning_rate": 8.293549194373243e-06,
277
+ "loss": 1.627,
278
+ "num_input_tokens_seen": 1187840,
279
+ "step": 145
280
+ },
281
+ {
282
+ "epoch": 1.8987341772151898,
283
+ "grad_norm": 245.75396728515625,
284
+ "learning_rate": 7.622215699028196e-06,
285
+ "loss": 1.6432,
286
+ "num_input_tokens_seen": 1228800,
287
+ "step": 150
288
+ },
289
+ {
290
+ "epoch": 1.8987341772151898,
291
+ "eval_accuracy": 0.2112676056338028,
292
+ "eval_loss": 1.6594650745391846,
293
+ "eval_runtime": 0.1424,
294
+ "eval_samples_per_second": 498.688,
295
+ "eval_steps_per_second": 63.214,
296
+ "num_input_tokens_seen": 1228800,
297
+ "step": 150
298
+ },
299
+ {
300
+ "epoch": 1.9620253164556962,
301
+ "grad_norm": 89.35352325439453,
302
+ "learning_rate": 6.962065528906321e-06,
303
+ "loss": 1.6411,
304
+ "num_input_tokens_seen": 1269760,
305
+ "step": 155
306
+ },
307
+ {
308
+ "epoch": 2.0253164556962027,
309
+ "grad_norm": 126.32188415527344,
310
+ "learning_rate": 6.316203538407397e-06,
311
+ "loss": 1.5683,
312
+ "num_input_tokens_seen": 1310720,
313
+ "step": 160
314
+ },
315
+ {
316
+ "epoch": 2.088607594936709,
317
+ "grad_norm": 127.23784637451172,
318
+ "learning_rate": 5.687667380983037e-06,
319
+ "loss": 1.7256,
320
+ "num_input_tokens_seen": 1351680,
321
+ "step": 165
322
+ },
323
+ {
324
+ "epoch": 2.151898734177215,
325
+ "grad_norm": 161.1302947998047,
326
+ "learning_rate": 5.07941322227877e-06,
327
+ "loss": 1.6769,
328
+ "num_input_tokens_seen": 1392640,
329
+ "step": 170
330
+ },
331
+ {
332
+ "epoch": 2.2151898734177213,
333
+ "grad_norm": 203.9555206298828,
334
+ "learning_rate": 4.494301836534016e-06,
335
+ "loss": 1.561,
336
+ "num_input_tokens_seen": 1433600,
337
+ "step": 175
338
+ },
339
+ {
340
+ "epoch": 2.278481012658228,
341
+ "grad_norm": 234.33477783203125,
342
+ "learning_rate": 3.935085151632185e-06,
343
+ "loss": 1.5984,
344
+ "num_input_tokens_seen": 1474560,
345
+ "step": 180
346
+ },
347
+ {
348
+ "epoch": 2.3417721518987342,
349
+ "grad_norm": 168.90626525878906,
350
+ "learning_rate": 3.4043933060828606e-06,
351
+ "loss": 1.5909,
352
+ "num_input_tokens_seen": 1515520,
353
+ "step": 185
354
+ },
355
+ {
356
+ "epoch": 2.4050632911392404,
357
+ "grad_norm": 99.04336547851562,
358
+ "learning_rate": 2.9047222788104712e-06,
359
+ "loss": 1.5661,
360
+ "num_input_tokens_seen": 1556480,
361
+ "step": 190
362
+ },
363
+ {
364
+ "epoch": 2.4683544303797467,
365
+ "grad_norm": 272.1407470703125,
366
+ "learning_rate": 2.4384221499296466e-06,
367
+ "loss": 1.5917,
368
+ "num_input_tokens_seen": 1597440,
369
+ "step": 195
370
+ },
371
+ {
372
+ "epoch": 2.5316455696202533,
373
+ "grad_norm": 114.18236541748047,
374
+ "learning_rate": 2.007686047719831e-06,
375
+ "loss": 1.5408,
376
+ "num_input_tokens_seen": 1638400,
377
+ "step": 200
378
+ },
379
+ {
380
+ "epoch": 2.5316455696202533,
381
+ "eval_accuracy": 0.19718309859154928,
382
+ "eval_loss": 1.6633472442626953,
383
+ "eval_runtime": 0.1412,
384
+ "eval_samples_per_second": 502.839,
385
+ "eval_steps_per_second": 63.74,
386
+ "num_input_tokens_seen": 1638400,
387
+ "step": 200
388
+ },
389
+ {
390
+ "epoch": 2.5949367088607596,
391
+ "grad_norm": 157.7032928466797,
392
+ "learning_rate": 1.6145398337843654e-06,
393
+ "loss": 1.5757,
394
+ "num_input_tokens_seen": 1679360,
395
+ "step": 205
396
+ },
397
+ {
398
+ "epoch": 2.6582278481012658,
399
+ "grad_norm": 88.51991271972656,
400
+ "learning_rate": 1.2608325749073591e-06,
401
+ "loss": 1.5955,
402
+ "num_input_tokens_seen": 1720320,
403
+ "step": 210
404
+ },
405
+ {
406
+ "epoch": 2.721518987341772,
407
+ "grad_norm": 139.67247009277344,
408
+ "learning_rate": 9.482278464216121e-07,
409
+ "loss": 1.558,
410
+ "num_input_tokens_seen": 1761280,
411
+ "step": 215
412
+ },
413
+ {
414
+ "epoch": 2.7848101265822782,
415
+ "grad_norm": 130.27508544921875,
416
+ "learning_rate": 6.781959079900958e-07,
417
+ "loss": 1.5464,
418
+ "num_input_tokens_seen": 1802240,
419
+ "step": 220
420
+ },
421
+ {
422
+ "epoch": 2.848101265822785,
423
+ "grad_norm": 163.9243621826172,
424
+ "learning_rate": 4.520067886002488e-07,
425
+ "loss": 1.6119,
426
+ "num_input_tokens_seen": 1843200,
427
+ "step": 225
428
+ },
429
+ {
430
+ "epoch": 2.911392405063291,
431
+ "grad_norm": 116.64250946044922,
432
+ "learning_rate": 2.707243132941717e-07,
433
+ "loss": 1.5334,
434
+ "num_input_tokens_seen": 1884160,
435
+ "step": 230
436
+ },
437
+ {
438
+ "epoch": 2.9746835443037973,
439
+ "grad_norm": 202.17108154296875,
440
+ "learning_rate": 1.3520109972846918e-07,
441
+ "loss": 1.5555,
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": 220374931341312.0,
450
+ "train_loss": 2.1441552528349157,
451
+ "train_runtime": 22.4706,
452
+ "train_samples_per_second": 84.243,
453
+ "train_steps_per_second": 10.547
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": 220374931341312.0,
474
+ "train_batch_size": 4,
475
+ "trial_name": null,
476
+ "trial_params": null
477
+ }