kanishka commited on
Commit
e405215
1 Parent(s): 7952f4d

End of training

Browse files
all_results.json CHANGED
@@ -1,15 +1,15 @@
1
  {
2
  "epoch": 10.0,
3
- "eval_accuracy": 0.4867319941720712,
4
- "eval_loss": 2.555483818054199,
5
- "eval_runtime": 4.1095,
6
  "eval_samples": 4491,
7
- "eval_samples_per_second": 1092.835,
8
- "eval_steps_per_second": 2.19,
9
- "perplexity": 12.877528531376772,
10
- "train_loss": 2.8862294931880763,
11
- "train_runtime": 610.8087,
12
  "train_samples": 46845,
13
- "train_samples_per_second": 766.934,
14
- "train_steps_per_second": 11.984
15
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "eval_accuracy": 0.49974139004167567,
4
+ "eval_loss": 2.4754340648651123,
5
+ "eval_runtime": 3.3524,
6
  "eval_samples": 4491,
7
+ "eval_samples_per_second": 1339.619,
8
+ "eval_steps_per_second": 10.738,
9
+ "perplexity": 11.886865664186647,
10
+ "train_loss": 2.566941033035028,
11
+ "train_runtime": 742.3412,
12
  "train_samples": 46845,
13
+ "train_samples_per_second": 631.044,
14
+ "train_steps_per_second": 39.443
15
  }
eval_results.json CHANGED
@@ -1,10 +1,10 @@
1
  {
2
  "epoch": 10.0,
3
- "eval_accuracy": 0.4867319941720712,
4
- "eval_loss": 2.555483818054199,
5
- "eval_runtime": 4.1095,
6
  "eval_samples": 4491,
7
- "eval_samples_per_second": 1092.835,
8
- "eval_steps_per_second": 2.19,
9
- "perplexity": 12.877528531376772
10
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "eval_accuracy": 0.49974139004167567,
4
+ "eval_loss": 2.4754340648651123,
5
+ "eval_runtime": 3.3524,
6
  "eval_samples": 4491,
7
+ "eval_samples_per_second": 1339.619,
8
+ "eval_steps_per_second": 10.738,
9
+ "perplexity": 11.886865664186647
10
  }
runs/Mar25_21-06-31_phyl-ling-p01.la.utexas.edu/events.out.tfevents.1711419559.phyl-ling-p01.la.utexas.edu.1054509.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a439ad4571163633e764508d72ba0681449c7c725d11bdb47b9679e53a082947
3
+ size 417
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 10.0,
3
- "train_loss": 2.8862294931880763,
4
- "train_runtime": 610.8087,
5
  "train_samples": 46845,
6
- "train_samples_per_second": 766.934,
7
- "train_steps_per_second": 11.984
8
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "train_loss": 2.566941033035028,
4
+ "train_runtime": 742.3412,
5
  "train_samples": 46845,
6
+ "train_samples_per_second": 631.044,
7
+ "train_steps_per_second": 39.443
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.1031,
15
  "step": 500
16
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  {
18
  "epoch": 1.0,
19
- "eval_accuracy": 0.3938620898840551,
20
- "eval_loss": 3.5406155586242676,
21
- "eval_runtime": 3.8659,
22
- "eval_samples_per_second": 1161.71,
23
- "eval_steps_per_second": 2.328,
24
- "step": 732
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  },
26
  {
27
  "epoch": 1.37,
28
- "learning_rate": 0.000125,
29
- "loss": 3.4871,
30
- "step": 1000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  },
32
  {
33
  "epoch": 2.0,
34
- "eval_accuracy": 0.4330568398389079,
35
- "eval_loss": 3.117299795150757,
36
- "eval_runtime": 3.7874,
37
- "eval_samples_per_second": 1185.775,
38
- "eval_steps_per_second": 2.376,
39
- "step": 1464
40
  },
41
  {
42
  "epoch": 2.05,
43
- "learning_rate": 0.0001875,
44
- "loss": 3.1416,
45
- "step": 1500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  },
47
  {
48
  "epoch": 2.73,
49
- "learning_rate": 0.00025,
50
- "loss": 2.9412,
51
- "step": 2000
 
 
 
 
 
 
 
 
52
  },
53
  {
54
  "epoch": 3.0,
55
- "eval_accuracy": 0.44953949894539735,
56
- "eval_loss": 2.9172728061676025,
57
- "eval_runtime": 3.7836,
58
- "eval_samples_per_second": 1186.975,
59
- "eval_steps_per_second": 2.379,
60
- "step": 2196
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61
  },
62
  {
63
  "epoch": 3.42,
64
- "learning_rate": 0.0003125,
65
- "loss": 2.8011,
66
- "step": 2500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
67
  },
68
  {
69
  "epoch": 4.0,
70
- "eval_accuracy": 0.4617967343260449,
71
- "eval_loss": 2.7835493087768555,
72
- "eval_runtime": 3.9326,
73
- "eval_samples_per_second": 1141.995,
74
- "eval_steps_per_second": 2.289,
75
- "step": 2928
76
  },
77
  {
78
  "epoch": 4.1,
79
- "learning_rate": 0.000375,
80
- "loss": 2.7056,
81
- "step": 3000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
82
  },
83
  {
84
  "epoch": 4.78,
85
- "learning_rate": 0.00043750000000000006,
86
- "loss": 2.6172,
87
- "step": 3500
 
 
 
 
 
 
 
 
88
  },
89
  {
90
  "epoch": 5.0,
91
- "eval_accuracy": 0.47097344294888993,
92
- "eval_loss": 2.697683095932007,
93
- "eval_runtime": 3.8698,
94
- "eval_samples_per_second": 1160.528,
95
- "eval_steps_per_second": 2.326,
96
- "step": 3660
 
 
 
 
 
 
 
 
 
 
 
 
 
 
97
  },
98
  {
99
  "epoch": 5.46,
100
- "learning_rate": 0.0005,
101
- "loss": 2.5378,
102
- "step": 4000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
103
  },
104
  {
105
  "epoch": 6.0,
106
- "eval_accuracy": 0.4763963622783625,
107
- "eval_loss": 2.6400082111358643,
108
- "eval_runtime": 3.8272,
109
- "eval_samples_per_second": 1173.433,
110
- "eval_steps_per_second": 2.352,
111
- "step": 4392
112
  },
113
  {
114
  "epoch": 6.15,
115
- "learning_rate": 0.0005625000000000001,
116
- "loss": 2.4963,
117
- "step": 4500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
118
  },
119
  {
120
  "epoch": 6.83,
121
- "learning_rate": 0.000625,
122
- "loss": 2.4512,
123
- "step": 5000
 
124
  },
125
  {
126
  "epoch": 7.0,
127
- "eval_accuracy": 0.4802588554186238,
128
- "eval_loss": 2.6059224605560303,
129
- "eval_runtime": 3.9524,
130
- "eval_samples_per_second": 1136.269,
131
- "eval_steps_per_second": 2.277,
132
- "step": 5124
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
133
  },
134
  {
135
  "epoch": 7.51,
136
- "learning_rate": 0.0006875,
137
- "loss": 2.4037,
138
- "step": 5500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
139
  },
140
  {
141
  "epoch": 8.0,
142
- "eval_accuracy": 0.4835830891879998,
143
- "eval_loss": 2.5811188220977783,
144
- "eval_runtime": 3.7863,
145
- "eval_samples_per_second": 1186.103,
146
- "eval_steps_per_second": 2.377,
147
- "step": 5856
 
 
 
 
 
 
 
148
  },
149
  {
150
  "epoch": 8.2,
151
- "learning_rate": 0.00075,
152
- "loss": 2.3838,
153
- "step": 6000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
154
  },
155
  {
156
  "epoch": 8.88,
157
- "learning_rate": 0.0008125,
158
- "loss": 2.3624,
159
- "step": 6500
 
160
  },
161
  {
162
  "epoch": 9.0,
163
- "eval_accuracy": 0.4844562265388169,
164
- "eval_loss": 2.5712339878082275,
165
- "eval_runtime": 3.9674,
166
- "eval_samples_per_second": 1131.98,
167
- "eval_steps_per_second": 2.268,
168
- "step": 6588
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
169
  },
170
  {
171
  "epoch": 9.56,
172
- "learning_rate": 0.0008750000000000001,
173
- "loss": 2.3246,
174
- "step": 7000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
175
  },
176
  {
177
  "epoch": 10.0,
178
- "eval_accuracy": 0.4867319941720712,
179
- "eval_loss": 2.555483818054199,
180
- "eval_runtime": 3.8239,
181
- "eval_samples_per_second": 1174.449,
182
- "eval_steps_per_second": 2.354,
183
- "step": 7320
184
  },
185
  {
186
  "epoch": 10.0,
187
- "step": 7320,
188
  "total_flos": 2273237316403200.0,
189
- "train_loss": 2.8862294931880763,
190
- "train_runtime": 610.8087,
191
- "train_samples_per_second": 766.934,
192
- "train_steps_per_second": 11.984
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.3569283485412598,
14
  "learning_rate": 6.25e-05,
15
+ "loss": 6.1959,
16
  "step": 500
17
  },
18
+ {
19
+ "epoch": 0.34,
20
+ "grad_norm": 0.8884565830230713,
21
+ "learning_rate": 0.000125,
22
+ "loss": 3.6302,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 0.51,
27
+ "grad_norm": 0.9340901374816895,
28
+ "learning_rate": 0.0001875,
29
+ "loss": 3.3395,
30
+ "step": 1500
31
+ },
32
+ {
33
+ "epoch": 0.68,
34
+ "grad_norm": 1.026479721069336,
35
+ "learning_rate": 0.00025,
36
+ "loss": 3.1647,
37
+ "step": 2000
38
+ },
39
+ {
40
+ "epoch": 0.85,
41
+ "grad_norm": 0.9679443836212158,
42
+ "learning_rate": 0.0003125,
43
+ "loss": 3.0568,
44
+ "step": 2500
45
+ },
46
  {
47
  "epoch": 1.0,
48
+ "eval_accuracy": 0.43760837510541645,
49
+ "eval_loss": 3.0170083045959473,
50
+ "eval_runtime": 3.3233,
51
+ "eval_samples_per_second": 1351.351,
52
+ "eval_steps_per_second": 10.832,
53
+ "step": 2928
54
+ },
55
+ {
56
+ "epoch": 1.02,
57
+ "grad_norm": 0.8825171589851379,
58
+ "learning_rate": 0.000375,
59
+ "loss": 2.9449,
60
+ "step": 3000
61
+ },
62
+ {
63
+ "epoch": 1.2,
64
+ "grad_norm": 0.9031845927238464,
65
+ "learning_rate": 0.00043750000000000006,
66
+ "loss": 2.8664,
67
+ "step": 3500
68
  },
69
  {
70
  "epoch": 1.37,
71
+ "grad_norm": 0.8770526051521301,
72
+ "learning_rate": 0.0005,
73
+ "loss": 2.8193,
74
+ "step": 4000
75
+ },
76
+ {
77
+ "epoch": 1.54,
78
+ "grad_norm": 0.8624604940414429,
79
+ "learning_rate": 0.0005625000000000001,
80
+ "loss": 2.7658,
81
+ "step": 4500
82
+ },
83
+ {
84
+ "epoch": 1.71,
85
+ "grad_norm": 0.8200684189796448,
86
+ "learning_rate": 0.000625,
87
+ "loss": 2.7358,
88
+ "step": 5000
89
+ },
90
+ {
91
+ "epoch": 1.88,
92
+ "grad_norm": 0.8737301230430603,
93
+ "learning_rate": 0.0006875,
94
+ "loss": 2.7079,
95
+ "step": 5500
96
  },
97
  {
98
  "epoch": 2.0,
99
+ "eval_accuracy": 0.4585023064501707,
100
+ "eval_loss": 2.7878565788269043,
101
+ "eval_runtime": 3.4124,
102
+ "eval_samples_per_second": 1316.077,
103
+ "eval_steps_per_second": 10.55,
104
+ "step": 5856
105
  },
106
  {
107
  "epoch": 2.05,
108
+ "grad_norm": 0.7897689342498779,
109
+ "learning_rate": 0.00075,
110
+ "loss": 2.6723,
111
+ "step": 6000
112
+ },
113
+ {
114
+ "epoch": 2.22,
115
+ "grad_norm": 0.8149366974830627,
116
+ "learning_rate": 0.0008125,
117
+ "loss": 2.6196,
118
+ "step": 6500
119
+ },
120
+ {
121
+ "epoch": 2.39,
122
+ "grad_norm": 0.684084951877594,
123
+ "learning_rate": 0.0008750000000000001,
124
+ "loss": 2.6148,
125
+ "step": 7000
126
+ },
127
+ {
128
+ "epoch": 2.56,
129
+ "grad_norm": 0.6549258828163147,
130
+ "learning_rate": 0.0009375,
131
+ "loss": 2.6107,
132
+ "step": 7500
133
  },
134
  {
135
  "epoch": 2.73,
136
+ "grad_norm": 0.670656681060791,
137
+ "learning_rate": 0.001,
138
+ "loss": 2.5927,
139
+ "step": 8000
140
+ },
141
+ {
142
+ "epoch": 2.9,
143
+ "grad_norm": 0.6147797703742981,
144
+ "learning_rate": 0.0010625,
145
+ "loss": 2.5856,
146
+ "step": 8500
147
  },
148
  {
149
  "epoch": 3.0,
150
+ "eval_accuracy": 0.47057018674268924,
151
+ "eval_loss": 2.6923789978027344,
152
+ "eval_runtime": 3.3929,
153
+ "eval_samples_per_second": 1323.657,
154
+ "eval_steps_per_second": 10.61,
155
+ "step": 8784
156
+ },
157
+ {
158
+ "epoch": 3.07,
159
+ "grad_norm": 0.6165093779563904,
160
+ "learning_rate": 0.0011250000000000001,
161
+ "loss": 2.5405,
162
+ "step": 9000
163
+ },
164
+ {
165
+ "epoch": 3.24,
166
+ "grad_norm": 0.5891627669334412,
167
+ "learning_rate": 0.0011875,
168
+ "loss": 2.513,
169
+ "step": 9500
170
  },
171
  {
172
  "epoch": 3.42,
173
+ "grad_norm": 0.5683926939964294,
174
+ "learning_rate": 0.00125,
175
+ "loss": 2.5239,
176
+ "step": 10000
177
+ },
178
+ {
179
+ "epoch": 3.59,
180
+ "grad_norm": 0.49277979135513306,
181
+ "learning_rate": 0.0013125,
182
+ "loss": 2.5101,
183
+ "step": 10500
184
+ },
185
+ {
186
+ "epoch": 3.76,
187
+ "grad_norm": 0.5084903836250305,
188
+ "learning_rate": 0.001375,
189
+ "loss": 2.5176,
190
+ "step": 11000
191
+ },
192
+ {
193
+ "epoch": 3.93,
194
+ "grad_norm": 0.4873659014701843,
195
+ "learning_rate": 0.0014375000000000002,
196
+ "loss": 2.5108,
197
+ "step": 11500
198
  },
199
  {
200
  "epoch": 4.0,
201
+ "eval_accuracy": 0.4762087604780865,
202
+ "eval_loss": 2.6403005123138428,
203
+ "eval_runtime": 3.4307,
204
+ "eval_samples_per_second": 1309.069,
205
+ "eval_steps_per_second": 10.494,
206
+ "step": 11712
207
  },
208
  {
209
  "epoch": 4.1,
210
+ "grad_norm": 0.47938045859336853,
211
+ "learning_rate": 0.0015,
212
+ "loss": 2.4624,
213
+ "step": 12000
214
+ },
215
+ {
216
+ "epoch": 4.27,
217
+ "grad_norm": 0.4903472065925598,
218
+ "learning_rate": 0.0015625,
219
+ "loss": 2.4398,
220
+ "step": 12500
221
+ },
222
+ {
223
+ "epoch": 4.44,
224
+ "grad_norm": 0.43617236614227295,
225
+ "learning_rate": 0.001625,
226
+ "loss": 2.4683,
227
+ "step": 13000
228
+ },
229
+ {
230
+ "epoch": 4.61,
231
+ "grad_norm": 0.40969905257225037,
232
+ "learning_rate": 0.0016875,
233
+ "loss": 2.4612,
234
+ "step": 13500
235
  },
236
  {
237
  "epoch": 4.78,
238
+ "grad_norm": 0.367087185382843,
239
+ "learning_rate": 0.0017500000000000003,
240
+ "loss": 2.4706,
241
+ "step": 14000
242
+ },
243
+ {
244
+ "epoch": 4.95,
245
+ "grad_norm": 0.37903687357902527,
246
+ "learning_rate": 0.0018124999999999999,
247
+ "loss": 2.4576,
248
+ "step": 14500
249
  },
250
  {
251
  "epoch": 5.0,
252
+ "eval_accuracy": 0.47962591850367403,
253
+ "eval_loss": 2.607229709625244,
254
+ "eval_runtime": 3.4265,
255
+ "eval_samples_per_second": 1310.668,
256
+ "eval_steps_per_second": 10.506,
257
+ "step": 14640
258
+ },
259
+ {
260
+ "epoch": 5.12,
261
+ "grad_norm": 0.38673949241638184,
262
+ "learning_rate": 0.001875,
263
+ "loss": 2.4,
264
+ "step": 15000
265
+ },
266
+ {
267
+ "epoch": 5.29,
268
+ "grad_norm": 0.38303470611572266,
269
+ "learning_rate": 0.0019375000000000002,
270
+ "loss": 2.3994,
271
+ "step": 15500
272
  },
273
  {
274
  "epoch": 5.46,
275
+ "grad_norm": 0.3417862057685852,
276
+ "learning_rate": 0.002,
277
+ "loss": 2.4168,
278
+ "step": 16000
279
+ },
280
+ {
281
+ "epoch": 5.64,
282
+ "grad_norm": 0.34288668632507324,
283
+ "learning_rate": 0.0020625,
284
+ "loss": 2.4167,
285
+ "step": 16500
286
+ },
287
+ {
288
+ "epoch": 5.81,
289
+ "grad_norm": 0.36294203996658325,
290
+ "learning_rate": 0.002125,
291
+ "loss": 2.4319,
292
+ "step": 17000
293
+ },
294
+ {
295
+ "epoch": 5.98,
296
+ "grad_norm": 0.3338480293750763,
297
+ "learning_rate": 0.0021874999999999998,
298
+ "loss": 2.423,
299
+ "step": 17500
300
  },
301
  {
302
  "epoch": 6.0,
303
+ "eval_accuracy": 0.48246449153775617,
304
+ "eval_loss": 2.5827646255493164,
305
+ "eval_runtime": 3.3721,
306
+ "eval_samples_per_second": 1331.828,
307
+ "eval_steps_per_second": 10.676,
308
+ "step": 17568
309
  },
310
  {
311
  "epoch": 6.15,
312
+ "grad_norm": 0.34943464398384094,
313
+ "learning_rate": 0.0022500000000000003,
314
+ "loss": 2.3545,
315
+ "step": 18000
316
+ },
317
+ {
318
+ "epoch": 6.32,
319
+ "grad_norm": 0.33251243829727173,
320
+ "learning_rate": 0.0023125000000000003,
321
+ "loss": 2.371,
322
+ "step": 18500
323
+ },
324
+ {
325
+ "epoch": 6.49,
326
+ "grad_norm": 0.36209550499916077,
327
+ "learning_rate": 0.002375,
328
+ "loss": 2.3805,
329
+ "step": 19000
330
+ },
331
+ {
332
+ "epoch": 6.66,
333
+ "grad_norm": 0.36579540371894836,
334
+ "learning_rate": 0.0024375,
335
+ "loss": 2.3914,
336
+ "step": 19500
337
  },
338
  {
339
  "epoch": 6.83,
340
+ "grad_norm": 0.3118489980697632,
341
+ "learning_rate": 0.0025,
342
+ "loss": 2.3921,
343
+ "step": 20000
344
  },
345
  {
346
  "epoch": 7.0,
347
+ "eval_accuracy": 0.48544858746364117,
348
+ "eval_loss": 2.566972017288208,
349
+ "eval_runtime": 3.3834,
350
+ "eval_samples_per_second": 1327.359,
351
+ "eval_steps_per_second": 10.64,
352
+ "step": 20496
353
+ },
354
+ {
355
+ "epoch": 7.0,
356
+ "grad_norm": 0.33852943778038025,
357
+ "learning_rate": 0.0025625,
358
+ "loss": 2.3805,
359
+ "step": 20500
360
+ },
361
+ {
362
+ "epoch": 7.17,
363
+ "grad_norm": 0.3410607874393463,
364
+ "learning_rate": 0.002625,
365
+ "loss": 2.3178,
366
+ "step": 21000
367
+ },
368
+ {
369
+ "epoch": 7.34,
370
+ "grad_norm": 0.3233848810195923,
371
+ "learning_rate": 0.0026875000000000002,
372
+ "loss": 2.3326,
373
+ "step": 21500
374
  },
375
  {
376
  "epoch": 7.51,
377
+ "grad_norm": 0.3061616122722626,
378
+ "learning_rate": 0.00275,
379
+ "loss": 2.3472,
380
+ "step": 22000
381
+ },
382
+ {
383
+ "epoch": 7.68,
384
+ "grad_norm": 0.3195803463459015,
385
+ "learning_rate": 0.0028125,
386
+ "loss": 2.3588,
387
+ "step": 22500
388
+ },
389
+ {
390
+ "epoch": 7.86,
391
+ "grad_norm": 0.37057217955589294,
392
+ "learning_rate": 0.0028750000000000004,
393
+ "loss": 2.3602,
394
+ "step": 23000
395
  },
396
  {
397
  "epoch": 8.0,
398
+ "eval_accuracy": 0.4870756385912683,
399
+ "eval_loss": 2.553264856338501,
400
+ "eval_runtime": 3.4098,
401
+ "eval_samples_per_second": 1317.103,
402
+ "eval_steps_per_second": 10.558,
403
+ "step": 23424
404
+ },
405
+ {
406
+ "epoch": 8.03,
407
+ "grad_norm": 0.31338265538215637,
408
+ "learning_rate": 0.0029375,
409
+ "loss": 2.3546,
410
+ "step": 23500
411
  },
412
  {
413
  "epoch": 8.2,
414
+ "grad_norm": 0.3508692979812622,
415
+ "learning_rate": 0.003,
416
+ "loss": 2.2832,
417
+ "step": 24000
418
+ },
419
+ {
420
+ "epoch": 8.37,
421
+ "grad_norm": 0.3429717421531677,
422
+ "learning_rate": 0.002715909090909091,
423
+ "loss": 2.3064,
424
+ "step": 24500
425
+ },
426
+ {
427
+ "epoch": 8.54,
428
+ "grad_norm": 0.3300691545009613,
429
+ "learning_rate": 0.0024318181818181817,
430
+ "loss": 2.3012,
431
+ "step": 25000
432
+ },
433
+ {
434
+ "epoch": 8.71,
435
+ "grad_norm": 0.31728777289390564,
436
+ "learning_rate": 0.002147727272727273,
437
+ "loss": 2.2953,
438
+ "step": 25500
439
  },
440
  {
441
  "epoch": 8.88,
442
+ "grad_norm": 0.29462775588035583,
443
+ "learning_rate": 0.0018636363636363638,
444
+ "loss": 2.2861,
445
+ "step": 26000
446
  },
447
  {
448
  "epoch": 9.0,
449
+ "eval_accuracy": 0.49417470110825323,
450
+ "eval_loss": 2.500990867614746,
451
+ "eval_runtime": 3.4136,
452
+ "eval_samples_per_second": 1315.633,
453
+ "eval_steps_per_second": 10.546,
454
+ "step": 26352
455
+ },
456
+ {
457
+ "epoch": 9.05,
458
+ "grad_norm": 0.29570701718330383,
459
+ "learning_rate": 0.0015795454545454546,
460
+ "loss": 2.2492,
461
+ "step": 26500
462
+ },
463
+ {
464
+ "epoch": 9.22,
465
+ "grad_norm": 0.3393871486186981,
466
+ "learning_rate": 0.0012954545454545456,
467
+ "loss": 2.1726,
468
+ "step": 27000
469
+ },
470
+ {
471
+ "epoch": 9.39,
472
+ "grad_norm": 0.32914644479751587,
473
+ "learning_rate": 0.0010113636363636364,
474
+ "loss": 2.1655,
475
+ "step": 27500
476
  },
477
  {
478
  "epoch": 9.56,
479
+ "grad_norm": 0.2837465703487396,
480
+ "learning_rate": 0.0007272727272727273,
481
+ "loss": 2.1501,
482
+ "step": 28000
483
+ },
484
+ {
485
+ "epoch": 9.73,
486
+ "grad_norm": 0.3447483479976654,
487
+ "learning_rate": 0.0004431818181818182,
488
+ "loss": 2.1465,
489
+ "step": 28500
490
+ },
491
+ {
492
+ "epoch": 9.9,
493
+ "grad_norm": 0.30058619379997253,
494
+ "learning_rate": 0.0001590909090909091,
495
+ "loss": 2.1381,
496
+ "step": 29000
497
  },
498
  {
499
  "epoch": 10.0,
500
+ "eval_accuracy": 0.49974139004167567,
501
+ "eval_loss": 2.4754340648651123,
502
+ "eval_runtime": 3.3618,
503
+ "eval_samples_per_second": 1335.893,
504
+ "eval_steps_per_second": 10.709,
505
+ "step": 29280
506
  },
507
  {
508
  "epoch": 10.0,
509
+ "step": 29280,
510
  "total_flos": 2273237316403200.0,
511
+ "train_loss": 2.566941033035028,
512
+ "train_runtime": 742.3412,
513
+ "train_samples_per_second": 631.044,
514
+ "train_steps_per_second": 39.443
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
  }