kanishka commited on
Commit
259acde
1 Parent(s): 3531423

End of training

Browse files
all_results.json CHANGED
@@ -1,15 +1,15 @@
1
  {
2
  "epoch": 10.0,
3
- "eval_accuracy": 0.48671446129354073,
4
- "eval_loss": 2.5570006370544434,
5
- "eval_runtime": 4.0107,
6
  "eval_samples": 4491,
7
- "eval_samples_per_second": 1119.749,
8
- "eval_steps_per_second": 2.244,
9
- "perplexity": 12.897076232743924,
10
- "train_loss": 2.8811990956791114,
11
- "train_runtime": 625.422,
12
  "train_samples": 46845,
13
- "train_samples_per_second": 749.014,
14
- "train_steps_per_second": 11.704
15
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "eval_accuracy": 0.5000412022645466,
4
+ "eval_loss": 2.4759199619293213,
5
+ "eval_runtime": 3.3905,
6
  "eval_samples": 4491,
7
+ "eval_samples_per_second": 1324.582,
8
+ "eval_steps_per_second": 10.618,
9
+ "perplexity": 11.892642860763285,
10
+ "train_loss": 2.5673805945557975,
11
+ "train_runtime": 750.0006,
12
  "train_samples": 46845,
13
+ "train_samples_per_second": 624.6,
14
+ "train_steps_per_second": 39.04
15
  }
eval_results.json CHANGED
@@ -1,10 +1,10 @@
1
  {
2
  "epoch": 10.0,
3
- "eval_accuracy": 0.48671446129354073,
4
- "eval_loss": 2.5570006370544434,
5
- "eval_runtime": 4.0107,
6
  "eval_samples": 4491,
7
- "eval_samples_per_second": 1119.749,
8
- "eval_steps_per_second": 2.244,
9
- "perplexity": 12.897076232743924
10
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "eval_accuracy": 0.5000412022645466,
4
+ "eval_loss": 2.4759199619293213,
5
+ "eval_runtime": 3.3905,
6
  "eval_samples": 4491,
7
+ "eval_samples_per_second": 1324.582,
8
+ "eval_steps_per_second": 10.618,
9
+ "perplexity": 11.892642860763285
10
  }
runs/Mar25_19-52-30_phyl-ling-p01.la.utexas.edu/events.out.tfevents.1711415108.phyl-ling-p01.la.utexas.edu.1050359.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbbc4b0c26111b10b7e950e148b91ba594f18f34a44c3fb484455b37804618ad
3
+ size 417
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 10.0,
3
- "train_loss": 2.8811990956791114,
4
- "train_runtime": 625.422,
5
  "train_samples": 46845,
6
- "train_samples_per_second": 749.014,
7
- "train_steps_per_second": 11.704
8
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "train_loss": 2.5673805945557975,
4
+ "train_runtime": 750.0006,
5
  "train_samples": 46845,
6
+ "train_samples_per_second": 624.6,
7
+ "train_steps_per_second": 39.04
8
  }
trainer_state.json CHANGED
@@ -3,200 +3,524 @@
3
  "best_model_checkpoint": null,
4
  "epoch": 10.0,
5
  "eval_steps": 500,
6
- "global_step": 7320,
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.68,
 
13
  "learning_rate": 6.25e-05,
14
- "loss": 6.078,
15
  "step": 500
16
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  {
18
  "epoch": 1.0,
19
- "eval_accuracy": 0.3944669741933561,
20
- "eval_loss": 3.533393621444702,
21
- "eval_runtime": 3.6343,
22
- "eval_samples_per_second": 1235.73,
23
- "eval_steps_per_second": 2.476,
24
- "step": 732
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  },
26
  {
27
  "epoch": 1.37,
28
- "learning_rate": 0.000125,
29
- "loss": 3.4808,
30
- "step": 1000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  },
32
  {
33
  "epoch": 2.0,
34
- "eval_accuracy": 0.4324659818324313,
35
- "eval_loss": 3.121781349182129,
36
- "eval_runtime": 3.7953,
37
- "eval_samples_per_second": 1183.308,
38
- "eval_steps_per_second": 2.371,
39
- "step": 1464
40
  },
41
  {
42
  "epoch": 2.05,
43
- "learning_rate": 0.0001875,
44
- "loss": 3.1425,
45
- "step": 1500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  },
47
  {
48
  "epoch": 2.73,
49
- "learning_rate": 0.00025,
50
- "loss": 2.9363,
51
- "step": 2000
 
 
 
 
 
 
 
 
52
  },
53
  {
54
  "epoch": 3.0,
55
- "eval_accuracy": 0.45024958052588115,
56
- "eval_loss": 2.9133858680725098,
57
- "eval_runtime": 3.764,
58
- "eval_samples_per_second": 1193.139,
59
- "eval_steps_per_second": 2.391,
60
- "step": 2196
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61
  },
62
  {
63
  "epoch": 3.42,
64
- "learning_rate": 0.0003125,
65
- "loss": 2.803,
66
- "step": 2500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
67
  },
68
  {
69
  "epoch": 4.0,
70
- "eval_accuracy": 0.46225434245569,
71
- "eval_loss": 2.783372402191162,
72
- "eval_runtime": 3.9861,
73
- "eval_samples_per_second": 1126.66,
74
- "eval_steps_per_second": 2.258,
75
- "step": 2928
76
  },
77
  {
78
  "epoch": 4.1,
79
- "learning_rate": 0.000375,
80
- "loss": 2.6954,
81
- "step": 3000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
82
  },
83
  {
84
  "epoch": 4.78,
85
- "learning_rate": 0.00043750000000000006,
86
- "loss": 2.6112,
87
- "step": 3500
 
 
 
 
 
 
 
 
88
  },
89
  {
90
  "epoch": 5.0,
91
- "eval_accuracy": 0.4700792661438362,
92
- "eval_loss": 2.6976747512817383,
93
- "eval_runtime": 4.0708,
94
- "eval_samples_per_second": 1103.233,
95
- "eval_steps_per_second": 2.211,
96
- "step": 3660
 
 
 
 
 
 
 
 
 
 
 
 
 
 
97
  },
98
  {
99
  "epoch": 5.46,
100
- "learning_rate": 0.0005,
101
- "loss": 2.5369,
102
- "step": 4000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
103
  },
104
  {
105
  "epoch": 6.0,
106
- "eval_accuracy": 0.4759703133300722,
107
- "eval_loss": 2.644042491912842,
108
- "eval_runtime": 4.218,
109
- "eval_samples_per_second": 1064.732,
110
- "eval_steps_per_second": 2.134,
111
- "step": 4392
112
  },
113
  {
114
  "epoch": 6.15,
115
- "learning_rate": 0.0005625000000000001,
116
- "loss": 2.4935,
117
- "step": 4500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
118
  },
119
  {
120
  "epoch": 6.83,
121
- "learning_rate": 0.000625,
122
- "loss": 2.4479,
123
- "step": 5000
 
 
 
 
 
 
 
 
 
 
124
  },
125
  {
126
  "epoch": 7.0,
127
- "eval_accuracy": 0.4810758875581434,
128
- "eval_loss": 2.601984739303589,
129
- "eval_runtime": 4.273,
130
- "eval_samples_per_second": 1051.027,
131
- "eval_steps_per_second": 2.106,
132
- "step": 5124
 
 
 
 
 
 
 
 
 
 
 
 
133
  },
134
  {
135
  "epoch": 7.51,
136
- "learning_rate": 0.0006875,
137
- "loss": 2.3999,
138
- "step": 5500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
139
  },
140
  {
141
  "epoch": 8.0,
142
- "eval_accuracy": 0.483818029760308,
143
- "eval_loss": 2.581047773361206,
144
- "eval_runtime": 4.1042,
145
- "eval_samples_per_second": 1094.257,
146
- "eval_steps_per_second": 2.193,
147
- "step": 5856
 
 
 
 
 
 
 
148
  },
149
  {
150
  "epoch": 8.2,
151
- "learning_rate": 0.00075,
152
- "loss": 2.3804,
153
- "step": 6000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
154
  },
155
  {
156
  "epoch": 8.88,
157
- "learning_rate": 0.0008125,
158
- "loss": 2.3566,
159
- "step": 6500
 
160
  },
161
  {
162
  "epoch": 9.0,
163
- "eval_accuracy": 0.48580275160995656,
164
- "eval_loss": 2.5657148361206055,
165
- "eval_runtime": 4.0599,
166
- "eval_samples_per_second": 1106.186,
167
- "eval_steps_per_second": 2.217,
168
- "step": 6588
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
169
  },
170
  {
171
  "epoch": 9.56,
172
- "learning_rate": 0.0008750000000000001,
173
- "loss": 2.3251,
174
- "step": 7000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
175
  },
176
  {
177
  "epoch": 10.0,
178
- "eval_accuracy": 0.48671446129354073,
179
- "eval_loss": 2.5570006370544434,
180
- "eval_runtime": 4.3421,
181
- "eval_samples_per_second": 1034.281,
182
- "eval_steps_per_second": 2.073,
183
- "step": 7320
184
  },
185
  {
186
  "epoch": 10.0,
187
- "step": 7320,
188
  "total_flos": 2273237316403200.0,
189
- "train_loss": 2.8811990956791114,
190
- "train_runtime": 625.422,
191
- "train_samples_per_second": 749.014,
192
- "train_steps_per_second": 11.704
193
  }
194
  ],
195
  "logging_steps": 500,
196
- "max_steps": 7320,
 
197
  "num_train_epochs": 10,
198
  "save_steps": 2000,
199
  "total_flos": 2273237316403200.0,
 
200
  "trial_name": null,
201
  "trial_params": null
202
  }
 
3
  "best_model_checkpoint": null,
4
  "epoch": 10.0,
5
  "eval_steps": 500,
6
+ "global_step": 29280,
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.17,
13
+ "grad_norm": 1.1428395509719849,
14
  "learning_rate": 6.25e-05,
15
+ "loss": 6.1643,
16
  "step": 500
17
  },
18
+ {
19
+ "epoch": 0.34,
20
+ "grad_norm": 0.8605946898460388,
21
+ "learning_rate": 0.000125,
22
+ "loss": 3.6177,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 0.51,
27
+ "grad_norm": 1.0353199243545532,
28
+ "learning_rate": 0.0001875,
29
+ "loss": 3.3423,
30
+ "step": 1500
31
+ },
32
+ {
33
+ "epoch": 0.68,
34
+ "grad_norm": 1.0619558095932007,
35
+ "learning_rate": 0.00025,
36
+ "loss": 3.174,
37
+ "step": 2000
38
+ },
39
+ {
40
+ "epoch": 0.85,
41
+ "grad_norm": 0.8885842561721802,
42
+ "learning_rate": 0.0003125,
43
+ "loss": 3.0508,
44
+ "step": 2500
45
+ },
46
  {
47
  "epoch": 1.0,
48
+ "eval_accuracy": 0.4366703661040366,
49
+ "eval_loss": 3.018024444580078,
50
+ "eval_runtime": 3.3274,
51
+ "eval_samples_per_second": 1349.7,
52
+ "eval_steps_per_second": 10.819,
53
+ "step": 2928
54
+ },
55
+ {
56
+ "epoch": 1.02,
57
+ "grad_norm": 0.9355678558349609,
58
+ "learning_rate": 0.000375,
59
+ "loss": 2.9588,
60
+ "step": 3000
61
+ },
62
+ {
63
+ "epoch": 1.2,
64
+ "grad_norm": 0.9893432259559631,
65
+ "learning_rate": 0.00043750000000000006,
66
+ "loss": 2.8647,
67
+ "step": 3500
68
  },
69
  {
70
  "epoch": 1.37,
71
+ "grad_norm": 0.8901029229164124,
72
+ "learning_rate": 0.0005,
73
+ "loss": 2.8204,
74
+ "step": 4000
75
+ },
76
+ {
77
+ "epoch": 1.54,
78
+ "grad_norm": 0.8454930186271667,
79
+ "learning_rate": 0.0005625000000000001,
80
+ "loss": 2.7743,
81
+ "step": 4500
82
+ },
83
+ {
84
+ "epoch": 1.71,
85
+ "grad_norm": 0.7920435667037964,
86
+ "learning_rate": 0.000625,
87
+ "loss": 2.7475,
88
+ "step": 5000
89
+ },
90
+ {
91
+ "epoch": 1.88,
92
+ "grad_norm": 0.862263023853302,
93
+ "learning_rate": 0.0006875,
94
+ "loss": 2.7062,
95
+ "step": 5500
96
  },
97
  {
98
  "epoch": 2.0,
99
+ "eval_accuracy": 0.45998208139814184,
100
+ "eval_loss": 2.785717725753784,
101
+ "eval_runtime": 3.419,
102
+ "eval_samples_per_second": 1313.535,
103
+ "eval_steps_per_second": 10.529,
104
+ "step": 5856
105
  },
106
  {
107
  "epoch": 2.05,
108
+ "grad_norm": 0.8128435611724854,
109
+ "learning_rate": 0.00075,
110
+ "loss": 2.667,
111
+ "step": 6000
112
+ },
113
+ {
114
+ "epoch": 2.22,
115
+ "grad_norm": 0.7912793755531311,
116
+ "learning_rate": 0.0008125,
117
+ "loss": 2.6257,
118
+ "step": 6500
119
+ },
120
+ {
121
+ "epoch": 2.39,
122
+ "grad_norm": 0.7729538679122925,
123
+ "learning_rate": 0.0008750000000000001,
124
+ "loss": 2.622,
125
+ "step": 7000
126
+ },
127
+ {
128
+ "epoch": 2.56,
129
+ "grad_norm": 0.7662726640701294,
130
+ "learning_rate": 0.0009375,
131
+ "loss": 2.5976,
132
+ "step": 7500
133
  },
134
  {
135
  "epoch": 2.73,
136
+ "grad_norm": 0.7259579300880432,
137
+ "learning_rate": 0.001,
138
+ "loss": 2.6013,
139
+ "step": 8000
140
+ },
141
+ {
142
+ "epoch": 2.9,
143
+ "grad_norm": 0.624014675617218,
144
+ "learning_rate": 0.0010625,
145
+ "loss": 2.5923,
146
+ "step": 8500
147
  },
148
  {
149
  "epoch": 3.0,
150
+ "eval_accuracy": 0.470033680659657,
151
+ "eval_loss": 2.690037488937378,
152
+ "eval_runtime": 3.4246,
153
+ "eval_samples_per_second": 1311.398,
154
+ "eval_steps_per_second": 10.512,
155
+ "step": 8784
156
+ },
157
+ {
158
+ "epoch": 3.07,
159
+ "grad_norm": 0.6166909337043762,
160
+ "learning_rate": 0.0011250000000000001,
161
+ "loss": 2.545,
162
+ "step": 9000
163
+ },
164
+ {
165
+ "epoch": 3.24,
166
+ "grad_norm": 0.6117091774940491,
167
+ "learning_rate": 0.0011875,
168
+ "loss": 2.5121,
169
+ "step": 9500
170
  },
171
  {
172
  "epoch": 3.42,
173
+ "grad_norm": 0.5614470839500427,
174
+ "learning_rate": 0.00125,
175
+ "loss": 2.5241,
176
+ "step": 10000
177
+ },
178
+ {
179
+ "epoch": 3.59,
180
+ "grad_norm": 0.5524864792823792,
181
+ "learning_rate": 0.0013125,
182
+ "loss": 2.5018,
183
+ "step": 10500
184
+ },
185
+ {
186
+ "epoch": 3.76,
187
+ "grad_norm": 0.488916277885437,
188
+ "learning_rate": 0.001375,
189
+ "loss": 2.5242,
190
+ "step": 11000
191
+ },
192
+ {
193
+ "epoch": 3.93,
194
+ "grad_norm": 0.49240800738334656,
195
+ "learning_rate": 0.0014375000000000002,
196
+ "loss": 2.5183,
197
+ "step": 11500
198
  },
199
  {
200
  "epoch": 4.0,
201
+ "eval_accuracy": 0.4760229119656636,
202
+ "eval_loss": 2.6405022144317627,
203
+ "eval_runtime": 3.3846,
204
+ "eval_samples_per_second": 1326.91,
205
+ "eval_steps_per_second": 10.637,
206
+ "step": 11712
207
  },
208
  {
209
  "epoch": 4.1,
210
+ "grad_norm": 0.4792906641960144,
211
+ "learning_rate": 0.0015,
212
+ "loss": 2.4636,
213
+ "step": 12000
214
+ },
215
+ {
216
+ "epoch": 4.27,
217
+ "grad_norm": 0.44715985655784607,
218
+ "learning_rate": 0.0015625,
219
+ "loss": 2.4505,
220
+ "step": 12500
221
+ },
222
+ {
223
+ "epoch": 4.44,
224
+ "grad_norm": 0.4706518054008484,
225
+ "learning_rate": 0.001625,
226
+ "loss": 2.4493,
227
+ "step": 13000
228
+ },
229
+ {
230
+ "epoch": 4.61,
231
+ "grad_norm": 0.4098159372806549,
232
+ "learning_rate": 0.0016875,
233
+ "loss": 2.4672,
234
+ "step": 13500
235
  },
236
  {
237
  "epoch": 4.78,
238
+ "grad_norm": 0.4300088584423065,
239
+ "learning_rate": 0.0017500000000000003,
240
+ "loss": 2.4597,
241
+ "step": 14000
242
+ },
243
+ {
244
+ "epoch": 4.95,
245
+ "grad_norm": 0.4115521013736725,
246
+ "learning_rate": 0.0018124999999999999,
247
+ "loss": 2.4632,
248
+ "step": 14500
249
  },
250
  {
251
  "epoch": 5.0,
252
+ "eval_accuracy": 0.4799397570293693,
253
+ "eval_loss": 2.610959768295288,
254
+ "eval_runtime": 3.4241,
255
+ "eval_samples_per_second": 1311.574,
256
+ "eval_steps_per_second": 10.514,
257
+ "step": 14640
258
+ },
259
+ {
260
+ "epoch": 5.12,
261
+ "grad_norm": 0.36413952708244324,
262
+ "learning_rate": 0.001875,
263
+ "loss": 2.4085,
264
+ "step": 15000
265
+ },
266
+ {
267
+ "epoch": 5.29,
268
+ "grad_norm": 0.4081680178642273,
269
+ "learning_rate": 0.0019375000000000002,
270
+ "loss": 2.408,
271
+ "step": 15500
272
  },
273
  {
274
  "epoch": 5.46,
275
+ "grad_norm": 0.3759631812572479,
276
+ "learning_rate": 0.002,
277
+ "loss": 2.4168,
278
+ "step": 16000
279
+ },
280
+ {
281
+ "epoch": 5.64,
282
+ "grad_norm": 0.3406212627887726,
283
+ "learning_rate": 0.0020625,
284
+ "loss": 2.419,
285
+ "step": 16500
286
+ },
287
+ {
288
+ "epoch": 5.81,
289
+ "grad_norm": 0.3810693621635437,
290
+ "learning_rate": 0.002125,
291
+ "loss": 2.4227,
292
+ "step": 17000
293
+ },
294
+ {
295
+ "epoch": 5.98,
296
+ "grad_norm": 0.40893515944480896,
297
+ "learning_rate": 0.0021874999999999998,
298
+ "loss": 2.4241,
299
+ "step": 17500
300
  },
301
  {
302
  "epoch": 6.0,
303
+ "eval_accuracy": 0.48350945109817184,
304
+ "eval_loss": 2.5840396881103516,
305
+ "eval_runtime": 3.3805,
306
+ "eval_samples_per_second": 1328.515,
307
+ "eval_steps_per_second": 10.649,
308
+ "step": 17568
309
  },
310
  {
311
  "epoch": 6.15,
312
+ "grad_norm": 0.332253098487854,
313
+ "learning_rate": 0.0022500000000000003,
314
+ "loss": 2.3589,
315
+ "step": 18000
316
+ },
317
+ {
318
+ "epoch": 6.32,
319
+ "grad_norm": 0.3078455924987793,
320
+ "learning_rate": 0.0023125000000000003,
321
+ "loss": 2.3668,
322
+ "step": 18500
323
+ },
324
+ {
325
+ "epoch": 6.49,
326
+ "grad_norm": 0.3293627202510834,
327
+ "learning_rate": 0.002375,
328
+ "loss": 2.3857,
329
+ "step": 19000
330
+ },
331
+ {
332
+ "epoch": 6.66,
333
+ "grad_norm": 0.3553139567375183,
334
+ "learning_rate": 0.0024375,
335
+ "loss": 2.3861,
336
+ "step": 19500
337
  },
338
  {
339
  "epoch": 6.83,
340
+ "grad_norm": 0.35184377431869507,
341
+ "learning_rate": 0.0025,
342
+ "loss": 2.3815,
343
+ "step": 20000
344
+ },
345
+ {
346
+ "epoch": 7.0,
347
+ "eval_accuracy": 0.48506812399953014,
348
+ "eval_loss": 2.572838306427002,
349
+ "eval_runtime": 3.4336,
350
+ "eval_samples_per_second": 1307.964,
351
+ "eval_steps_per_second": 10.485,
352
+ "step": 20496
353
  },
354
  {
355
  "epoch": 7.0,
356
+ "grad_norm": 0.3210340440273285,
357
+ "learning_rate": 0.0025625,
358
+ "loss": 2.3912,
359
+ "step": 20500
360
+ },
361
+ {
362
+ "epoch": 7.17,
363
+ "grad_norm": 0.37861162424087524,
364
+ "learning_rate": 0.002625,
365
+ "loss": 2.315,
366
+ "step": 21000
367
+ },
368
+ {
369
+ "epoch": 7.34,
370
+ "grad_norm": 0.368950217962265,
371
+ "learning_rate": 0.0026875000000000002,
372
+ "loss": 2.3418,
373
+ "step": 21500
374
  },
375
  {
376
  "epoch": 7.51,
377
+ "grad_norm": 0.3538954257965088,
378
+ "learning_rate": 0.00275,
379
+ "loss": 2.3559,
380
+ "step": 22000
381
+ },
382
+ {
383
+ "epoch": 7.68,
384
+ "grad_norm": 0.33648544549942017,
385
+ "learning_rate": 0.0028125,
386
+ "loss": 2.3544,
387
+ "step": 22500
388
+ },
389
+ {
390
+ "epoch": 7.86,
391
+ "grad_norm": 0.2958958148956299,
392
+ "learning_rate": 0.0028750000000000004,
393
+ "loss": 2.3595,
394
+ "step": 23000
395
  },
396
  {
397
  "epoch": 8.0,
398
+ "eval_accuracy": 0.48671446129354073,
399
+ "eval_loss": 2.558120012283325,
400
+ "eval_runtime": 3.4225,
401
+ "eval_samples_per_second": 1312.192,
402
+ "eval_steps_per_second": 10.519,
403
+ "step": 23424
404
+ },
405
+ {
406
+ "epoch": 8.03,
407
+ "grad_norm": 0.3472735285758972,
408
+ "learning_rate": 0.0029375,
409
+ "loss": 2.3436,
410
+ "step": 23500
411
  },
412
  {
413
  "epoch": 8.2,
414
+ "grad_norm": 0.3491041660308838,
415
+ "learning_rate": 0.003,
416
+ "loss": 2.2902,
417
+ "step": 24000
418
+ },
419
+ {
420
+ "epoch": 8.37,
421
+ "grad_norm": 0.36862629652023315,
422
+ "learning_rate": 0.002715909090909091,
423
+ "loss": 2.301,
424
+ "step": 24500
425
+ },
426
+ {
427
+ "epoch": 8.54,
428
+ "grad_norm": 0.3976895809173584,
429
+ "learning_rate": 0.0024318181818181817,
430
+ "loss": 2.3072,
431
+ "step": 25000
432
+ },
433
+ {
434
+ "epoch": 8.71,
435
+ "grad_norm": 0.33942127227783203,
436
+ "learning_rate": 0.002147727272727273,
437
+ "loss": 2.2934,
438
+ "step": 25500
439
  },
440
  {
441
  "epoch": 8.88,
442
+ "grad_norm": 0.3659175932407379,
443
+ "learning_rate": 0.0018636363636363638,
444
+ "loss": 2.2838,
445
+ "step": 26000
446
  },
447
  {
448
  "epoch": 9.0,
449
+ "eval_accuracy": 0.49492686159721017,
450
+ "eval_loss": 2.5014333724975586,
451
+ "eval_runtime": 3.4108,
452
+ "eval_samples_per_second": 1316.695,
453
+ "eval_steps_per_second": 10.555,
454
+ "step": 26352
455
+ },
456
+ {
457
+ "epoch": 9.05,
458
+ "grad_norm": 0.3276906907558441,
459
+ "learning_rate": 0.0015795454545454546,
460
+ "loss": 2.2536,
461
+ "step": 26500
462
+ },
463
+ {
464
+ "epoch": 9.22,
465
+ "grad_norm": 0.29987043142318726,
466
+ "learning_rate": 0.0012954545454545456,
467
+ "loss": 2.1713,
468
+ "step": 27000
469
+ },
470
+ {
471
+ "epoch": 9.39,
472
+ "grad_norm": 0.29404929280281067,
473
+ "learning_rate": 0.0010113636363636364,
474
+ "loss": 2.1643,
475
+ "step": 27500
476
  },
477
  {
478
  "epoch": 9.56,
479
+ "grad_norm": 0.26416483521461487,
480
+ "learning_rate": 0.0007272727272727273,
481
+ "loss": 2.1585,
482
+ "step": 28000
483
+ },
484
+ {
485
+ "epoch": 9.73,
486
+ "grad_norm": 0.3155469596385956,
487
+ "learning_rate": 0.0004431818181818182,
488
+ "loss": 2.1498,
489
+ "step": 28500
490
+ },
491
+ {
492
+ "epoch": 9.9,
493
+ "grad_norm": 0.35845381021499634,
494
+ "learning_rate": 0.0001590909090909091,
495
+ "loss": 2.1364,
496
+ "step": 29000
497
  },
498
  {
499
  "epoch": 10.0,
500
+ "eval_accuracy": 0.5000412022645466,
501
+ "eval_loss": 2.4759199619293213,
502
+ "eval_runtime": 3.4484,
503
+ "eval_samples_per_second": 1302.334,
504
+ "eval_steps_per_second": 10.44,
505
+ "step": 29280
506
  },
507
  {
508
  "epoch": 10.0,
509
+ "step": 29280,
510
  "total_flos": 2273237316403200.0,
511
+ "train_loss": 2.5673805945557975,
512
+ "train_runtime": 750.0006,
513
+ "train_samples_per_second": 624.6,
514
+ "train_steps_per_second": 39.04
515
  }
516
  ],
517
  "logging_steps": 500,
518
+ "max_steps": 29280,
519
+ "num_input_tokens_seen": 0,
520
  "num_train_epochs": 10,
521
  "save_steps": 2000,
522
  "total_flos": 2273237316403200.0,
523
+ "train_batch_size": 16,
524
  "trial_name": null,
525
  "trial_params": null
526
  }