mikhail-panzo commited on
Commit
16780a1
1 Parent(s): 7c39dd8

Training in progress, step 2000, checkpoint

Browse files
checkpoint-2000/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1af5ba9426ba255036b3e6e2b13db0b1f45bd15f4b1d4bebcb310e0deaeedd92
3
  size 577789320
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:807f698e4bfe058beabc0d8d422b9755a6f22e7229dec51a6d0c8f6ac60204d5
3
  size 577789320
checkpoint-2000/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:43bf774e2ab05c18a2a9069be9f45fb6dab500375dab25f10ddd7cb33e96e118
3
  size 1155772233
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f3f1924b32b0863cf14e60a41931f84d05b99727d89e5caed4e45bcfd5fe93a2
3
  size 1155772233
checkpoint-2000/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c37db81b30d7d7e10151647a4b79b9dcfccc175f8e0549c1cad2bed3b088a70
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:024d0cf30a7dc2cd699fa88d8139043fbfd6dbc8be6ed6931372b1fb1049dcda
3
  size 14244
checkpoint-2000/trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "best_metric": 0.41223829984664917,
3
- "best_model_checkpoint": "mikhail-panzo/zlm-fil-ceb_b64_le5_s8000/checkpoint-1500",
4
- "epoch": 86.95652173913044,
5
  "eval_steps": 500,
6
  "global_step": 2000,
7
  "is_hyper_param_search": false,
@@ -9,322 +9,322 @@
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 2.1739130434782608,
13
- "grad_norm": 0.7357823252677917,
14
  "learning_rate": 2.5000000000000004e-07,
15
- "loss": 0.4055,
16
  "step": 50
17
  },
18
  {
19
- "epoch": 4.3478260869565215,
20
- "grad_norm": 0.7944665551185608,
21
  "learning_rate": 5.000000000000001e-07,
22
- "loss": 0.3988,
23
  "step": 100
24
  },
25
  {
26
- "epoch": 6.521739130434782,
27
- "grad_norm": 0.7226569652557373,
28
  "learning_rate": 7.5e-07,
29
- "loss": 0.4072,
30
  "step": 150
31
  },
32
  {
33
- "epoch": 8.695652173913043,
34
- "grad_norm": 0.7845238447189331,
35
  "learning_rate": 1.0000000000000002e-06,
36
- "loss": 0.4028,
37
  "step": 200
38
  },
39
  {
40
- "epoch": 10.869565217391305,
41
- "grad_norm": 0.9769566655158997,
42
  "learning_rate": 1.25e-06,
43
- "loss": 0.4066,
44
  "step": 250
45
  },
46
  {
47
- "epoch": 13.043478260869565,
48
- "grad_norm": 0.8634074330329895,
49
  "learning_rate": 1.5e-06,
50
- "loss": 0.405,
51
  "step": 300
52
  },
53
  {
54
- "epoch": 15.217391304347826,
55
- "grad_norm": 0.9029327034950256,
56
  "learning_rate": 1.75e-06,
57
- "loss": 0.4074,
58
  "step": 350
59
  },
60
  {
61
- "epoch": 17.391304347826086,
62
- "grad_norm": 0.7624074220657349,
63
  "learning_rate": 2.0000000000000003e-06,
64
- "loss": 0.406,
65
  "step": 400
66
  },
67
  {
68
- "epoch": 19.565217391304348,
69
- "grad_norm": 1.005196213722229,
70
  "learning_rate": 2.25e-06,
71
- "loss": 0.4052,
72
  "step": 450
73
  },
74
  {
75
- "epoch": 21.73913043478261,
76
- "grad_norm": 1.0276380777359009,
77
  "learning_rate": 2.5e-06,
78
- "loss": 0.4019,
79
  "step": 500
80
  },
81
  {
82
- "epoch": 21.73913043478261,
83
- "eval_loss": 0.4143177568912506,
84
- "eval_runtime": 6.4074,
85
- "eval_samples_per_second": 24.815,
86
- "eval_steps_per_second": 3.121,
87
  "step": 500
88
  },
89
  {
90
- "epoch": 23.91304347826087,
91
- "grad_norm": 0.9067140817642212,
92
  "learning_rate": 2.7500000000000004e-06,
93
- "loss": 0.4021,
94
  "step": 550
95
  },
96
  {
97
- "epoch": 26.08695652173913,
98
- "grad_norm": 0.8799042701721191,
99
  "learning_rate": 3e-06,
100
- "loss": 0.4071,
101
  "step": 600
102
  },
103
  {
104
- "epoch": 28.26086956521739,
105
- "grad_norm": 1.0451533794403076,
106
  "learning_rate": 3.2500000000000002e-06,
107
- "loss": 0.4146,
108
  "step": 650
109
  },
110
  {
111
- "epoch": 30.434782608695652,
112
- "grad_norm": 0.7998800277709961,
113
  "learning_rate": 3.5e-06,
114
- "loss": 0.4023,
115
  "step": 700
116
  },
117
  {
118
- "epoch": 32.608695652173914,
119
- "grad_norm": 1.0301792621612549,
120
  "learning_rate": 3.7500000000000005e-06,
121
- "loss": 0.4045,
122
  "step": 750
123
  },
124
  {
125
- "epoch": 34.78260869565217,
126
- "grad_norm": 0.6388185620307922,
127
  "learning_rate": 4.000000000000001e-06,
128
- "loss": 0.4045,
129
  "step": 800
130
  },
131
  {
132
- "epoch": 36.95652173913044,
133
- "grad_norm": 0.9676840901374817,
134
  "learning_rate": 4.25e-06,
135
- "loss": 0.4116,
136
  "step": 850
137
  },
138
  {
139
- "epoch": 39.130434782608695,
140
- "grad_norm": 0.8355514407157898,
141
  "learning_rate": 4.5e-06,
142
- "loss": 0.4013,
143
  "step": 900
144
  },
145
  {
146
- "epoch": 41.30434782608695,
147
- "grad_norm": 0.7534067630767822,
148
  "learning_rate": 4.75e-06,
149
- "loss": 0.4019,
150
  "step": 950
151
  },
152
  {
153
- "epoch": 43.47826086956522,
154
- "grad_norm": 0.9366117119789124,
155
  "learning_rate": 5e-06,
156
- "loss": 0.4064,
157
  "step": 1000
158
  },
159
  {
160
- "epoch": 43.47826086956522,
161
- "eval_loss": 0.41316837072372437,
162
- "eval_runtime": 6.5616,
163
- "eval_samples_per_second": 24.232,
164
- "eval_steps_per_second": 3.048,
165
  "step": 1000
166
  },
167
  {
168
- "epoch": 45.65217391304348,
169
- "grad_norm": 0.7025179266929626,
170
  "learning_rate": 5.2500000000000006e-06,
171
- "loss": 0.4061,
172
  "step": 1050
173
  },
174
  {
175
- "epoch": 47.82608695652174,
176
- "grad_norm": 0.9240646958351135,
177
  "learning_rate": 5.500000000000001e-06,
178
- "loss": 0.4061,
179
  "step": 1100
180
  },
181
  {
182
- "epoch": 50.0,
183
- "grad_norm": 0.9685391187667847,
184
  "learning_rate": 5.75e-06,
185
- "loss": 0.4063,
186
  "step": 1150
187
  },
188
  {
189
- "epoch": 52.17391304347826,
190
- "grad_norm": 0.8591094017028809,
191
  "learning_rate": 6e-06,
192
- "loss": 0.4013,
193
  "step": 1200
194
  },
195
  {
196
- "epoch": 54.34782608695652,
197
- "grad_norm": 0.9348228573799133,
198
  "learning_rate": 6.25e-06,
199
- "loss": 0.402,
200
  "step": 1250
201
  },
202
  {
203
- "epoch": 56.52173913043478,
204
- "grad_norm": 0.7908409237861633,
205
  "learning_rate": 6.5000000000000004e-06,
206
- "loss": 0.4046,
207
  "step": 1300
208
  },
209
  {
210
- "epoch": 58.69565217391305,
211
- "grad_norm": 1.109017014503479,
212
  "learning_rate": 6.750000000000001e-06,
213
- "loss": 0.3995,
214
  "step": 1350
215
  },
216
  {
217
- "epoch": 60.869565217391305,
218
- "grad_norm": 0.8529478311538696,
219
  "learning_rate": 7e-06,
220
- "loss": 0.4018,
221
  "step": 1400
222
  },
223
  {
224
- "epoch": 63.04347826086956,
225
- "grad_norm": 0.9947476387023926,
226
  "learning_rate": 7.25e-06,
227
- "loss": 0.401,
228
  "step": 1450
229
  },
230
  {
231
- "epoch": 65.21739130434783,
232
- "grad_norm": 1.2476813793182373,
233
  "learning_rate": 7.500000000000001e-06,
234
- "loss": 0.4034,
235
  "step": 1500
236
  },
237
  {
238
- "epoch": 65.21739130434783,
239
- "eval_loss": 0.41223829984664917,
240
- "eval_runtime": 6.5846,
241
- "eval_samples_per_second": 24.147,
242
- "eval_steps_per_second": 3.037,
243
  "step": 1500
244
  },
245
  {
246
- "epoch": 67.3913043478261,
247
- "grad_norm": 0.8658013343811035,
248
  "learning_rate": 7.75e-06,
249
- "loss": 0.4007,
250
  "step": 1550
251
  },
252
  {
253
- "epoch": 69.56521739130434,
254
- "grad_norm": 0.7932788133621216,
255
  "learning_rate": 8.000000000000001e-06,
256
- "loss": 0.4054,
257
  "step": 1600
258
  },
259
  {
260
- "epoch": 71.73913043478261,
261
- "grad_norm": 2.33347225189209,
262
  "learning_rate": 8.25e-06,
263
- "loss": 0.4075,
264
  "step": 1650
265
  },
266
  {
267
- "epoch": 73.91304347826087,
268
- "grad_norm": 1.0095164775848389,
269
  "learning_rate": 8.5e-06,
270
- "loss": 0.3982,
271
  "step": 1700
272
  },
273
  {
274
- "epoch": 76.08695652173913,
275
- "grad_norm": 0.8895902633666992,
276
  "learning_rate": 8.750000000000001e-06,
277
- "loss": 0.4018,
278
  "step": 1750
279
  },
280
  {
281
- "epoch": 78.26086956521739,
282
- "grad_norm": 1.2252330780029297,
283
  "learning_rate": 9e-06,
284
- "loss": 0.4024,
285
  "step": 1800
286
  },
287
  {
288
- "epoch": 80.43478260869566,
289
- "grad_norm": 0.7731898427009583,
290
  "learning_rate": 9.250000000000001e-06,
291
- "loss": 0.3994,
292
  "step": 1850
293
  },
294
  {
295
- "epoch": 82.6086956521739,
296
- "grad_norm": 1.047607660293579,
297
  "learning_rate": 9.5e-06,
298
- "loss": 0.4013,
299
  "step": 1900
300
  },
301
  {
302
- "epoch": 84.78260869565217,
303
- "grad_norm": 0.9284490346908569,
304
  "learning_rate": 9.75e-06,
305
- "loss": 0.3986,
306
  "step": 1950
307
  },
308
  {
309
- "epoch": 86.95652173913044,
310
- "grad_norm": 4.725491046905518,
311
  "learning_rate": 1e-05,
312
- "loss": 0.3985,
313
  "step": 2000
314
  },
315
  {
316
- "epoch": 86.95652173913044,
317
- "eval_loss": 0.4147759974002838,
318
- "eval_runtime": 6.5475,
319
- "eval_samples_per_second": 24.284,
320
- "eval_steps_per_second": 3.055,
321
  "step": 2000
322
  }
323
  ],
324
  "logging_steps": 50,
325
  "max_steps": 8000,
326
  "num_input_tokens_seen": 0,
327
- "num_train_epochs": 348,
328
  "save_steps": 500,
329
  "stateful_callbacks": {
330
  "TrainerControl": {
@@ -338,8 +338,8 @@
338
  "attributes": {}
339
  }
340
  },
341
- "total_flos": 2.960719045637443e+16,
342
- "train_batch_size": 64,
343
  "trial_name": null,
344
  "trial_params": null
345
  }
 
1
  {
2
+ "best_metric": 0.39878609776496887,
3
+ "best_model_checkpoint": "mikhail-panzo/zlm-fil-ceb_b64_le5_s8000/checkpoint-2000",
4
+ "epoch": 79.20792079207921,
5
  "eval_steps": 500,
6
  "global_step": 2000,
7
  "is_hyper_param_search": false,
 
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
+ "epoch": 1.9801980198019802,
13
+ "grad_norm": 1.2683665752410889,
14
  "learning_rate": 2.5000000000000004e-07,
15
+ "loss": 0.4932,
16
  "step": 50
17
  },
18
  {
19
+ "epoch": 3.9603960396039604,
20
+ "grad_norm": 1.1266463994979858,
21
  "learning_rate": 5.000000000000001e-07,
22
+ "loss": 0.4898,
23
  "step": 100
24
  },
25
  {
26
+ "epoch": 5.9405940594059405,
27
+ "grad_norm": 1.1800968647003174,
28
  "learning_rate": 7.5e-07,
29
+ "loss": 0.4839,
30
  "step": 150
31
  },
32
  {
33
+ "epoch": 7.920792079207921,
34
+ "grad_norm": 0.8235568404197693,
35
  "learning_rate": 1.0000000000000002e-06,
36
+ "loss": 0.4785,
37
  "step": 200
38
  },
39
  {
40
+ "epoch": 9.900990099009901,
41
+ "grad_norm": 1.313211441040039,
42
  "learning_rate": 1.25e-06,
43
+ "loss": 0.4767,
44
  "step": 250
45
  },
46
  {
47
+ "epoch": 11.881188118811881,
48
+ "grad_norm": 0.7831560373306274,
49
  "learning_rate": 1.5e-06,
50
+ "loss": 0.4681,
51
  "step": 300
52
  },
53
  {
54
+ "epoch": 13.861386138613861,
55
+ "grad_norm": 0.7987237572669983,
56
  "learning_rate": 1.75e-06,
57
+ "loss": 0.4658,
58
  "step": 350
59
  },
60
  {
61
+ "epoch": 15.841584158415841,
62
+ "grad_norm": 0.7143813371658325,
63
  "learning_rate": 2.0000000000000003e-06,
64
+ "loss": 0.4627,
65
  "step": 400
66
  },
67
  {
68
+ "epoch": 17.821782178217823,
69
+ "grad_norm": 0.8037531971931458,
70
  "learning_rate": 2.25e-06,
71
+ "loss": 0.461,
72
  "step": 450
73
  },
74
  {
75
+ "epoch": 19.801980198019802,
76
+ "grad_norm": 0.79031902551651,
77
  "learning_rate": 2.5e-06,
78
+ "loss": 0.4592,
79
  "step": 500
80
  },
81
  {
82
+ "epoch": 19.801980198019802,
83
+ "eval_loss": 0.42528408765792847,
84
+ "eval_runtime": 7.8982,
85
+ "eval_samples_per_second": 22.79,
86
+ "eval_steps_per_second": 2.912,
87
  "step": 500
88
  },
89
  {
90
+ "epoch": 21.782178217821784,
91
+ "grad_norm": 0.7966288328170776,
92
  "learning_rate": 2.7500000000000004e-06,
93
+ "loss": 0.4551,
94
  "step": 550
95
  },
96
  {
97
+ "epoch": 23.762376237623762,
98
+ "grad_norm": 0.8126183152198792,
99
  "learning_rate": 3e-06,
100
+ "loss": 0.4542,
101
  "step": 600
102
  },
103
  {
104
+ "epoch": 25.742574257425744,
105
+ "grad_norm": 0.6602805852890015,
106
  "learning_rate": 3.2500000000000002e-06,
107
+ "loss": 0.4535,
108
  "step": 650
109
  },
110
  {
111
+ "epoch": 27.722772277227723,
112
+ "grad_norm": 0.7177000641822815,
113
  "learning_rate": 3.5e-06,
114
+ "loss": 0.4484,
115
  "step": 700
116
  },
117
  {
118
+ "epoch": 29.702970297029704,
119
+ "grad_norm": 0.7912996411323547,
120
  "learning_rate": 3.7500000000000005e-06,
121
+ "loss": 0.4486,
122
  "step": 750
123
  },
124
  {
125
+ "epoch": 31.683168316831683,
126
+ "grad_norm": 0.6251747608184814,
127
  "learning_rate": 4.000000000000001e-06,
128
+ "loss": 0.4463,
129
  "step": 800
130
  },
131
  {
132
+ "epoch": 33.663366336633665,
133
+ "grad_norm": 1.003780722618103,
134
  "learning_rate": 4.25e-06,
135
+ "loss": 0.4474,
136
  "step": 850
137
  },
138
  {
139
+ "epoch": 35.64356435643565,
140
+ "grad_norm": 0.742171585559845,
141
  "learning_rate": 4.5e-06,
142
+ "loss": 0.4434,
143
  "step": 900
144
  },
145
  {
146
+ "epoch": 37.62376237623762,
147
+ "grad_norm": 0.6034647822380066,
148
  "learning_rate": 4.75e-06,
149
+ "loss": 0.4425,
150
  "step": 950
151
  },
152
  {
153
+ "epoch": 39.603960396039604,
154
+ "grad_norm": 0.6990635395050049,
155
  "learning_rate": 5e-06,
156
+ "loss": 0.4381,
157
  "step": 1000
158
  },
159
  {
160
+ "epoch": 39.603960396039604,
161
+ "eval_loss": 0.41002073884010315,
162
+ "eval_runtime": 6.7782,
163
+ "eval_samples_per_second": 26.556,
164
+ "eval_steps_per_second": 3.393,
165
  "step": 1000
166
  },
167
  {
168
+ "epoch": 41.584158415841586,
169
+ "grad_norm": 0.9420008659362793,
170
  "learning_rate": 5.2500000000000006e-06,
171
+ "loss": 0.4421,
172
  "step": 1050
173
  },
174
  {
175
+ "epoch": 43.56435643564357,
176
+ "grad_norm": 0.7143107652664185,
177
  "learning_rate": 5.500000000000001e-06,
178
+ "loss": 0.4389,
179
  "step": 1100
180
  },
181
  {
182
+ "epoch": 45.54455445544554,
183
+ "grad_norm": 0.7934616804122925,
184
  "learning_rate": 5.75e-06,
185
+ "loss": 0.4369,
186
  "step": 1150
187
  },
188
  {
189
+ "epoch": 47.524752475247524,
190
+ "grad_norm": 0.9682563543319702,
191
  "learning_rate": 6e-06,
192
+ "loss": 0.4355,
193
  "step": 1200
194
  },
195
  {
196
+ "epoch": 49.504950495049506,
197
+ "grad_norm": 0.7880070209503174,
198
  "learning_rate": 6.25e-06,
199
+ "loss": 0.4359,
200
  "step": 1250
201
  },
202
  {
203
+ "epoch": 51.48514851485149,
204
+ "grad_norm": 0.8443828821182251,
205
  "learning_rate": 6.5000000000000004e-06,
206
+ "loss": 0.4324,
207
  "step": 1300
208
  },
209
  {
210
+ "epoch": 53.46534653465346,
211
+ "grad_norm": 0.724743127822876,
212
  "learning_rate": 6.750000000000001e-06,
213
+ "loss": 0.4348,
214
  "step": 1350
215
  },
216
  {
217
+ "epoch": 55.445544554455445,
218
+ "grad_norm": 0.7557791471481323,
219
  "learning_rate": 7e-06,
220
+ "loss": 0.433,
221
  "step": 1400
222
  },
223
  {
224
+ "epoch": 57.42574257425743,
225
+ "grad_norm": 0.7936097383499146,
226
  "learning_rate": 7.25e-06,
227
+ "loss": 0.4273,
228
  "step": 1450
229
  },
230
  {
231
+ "epoch": 59.40594059405941,
232
+ "grad_norm": 0.7041624784469604,
233
  "learning_rate": 7.500000000000001e-06,
234
+ "loss": 0.4281,
235
  "step": 1500
236
  },
237
  {
238
+ "epoch": 59.40594059405941,
239
+ "eval_loss": 0.40218469500541687,
240
+ "eval_runtime": 7.4309,
241
+ "eval_samples_per_second": 24.223,
242
+ "eval_steps_per_second": 3.095,
243
  "step": 1500
244
  },
245
  {
246
+ "epoch": 61.386138613861384,
247
+ "grad_norm": 0.9194286465644836,
248
  "learning_rate": 7.75e-06,
249
+ "loss": 0.4275,
250
  "step": 1550
251
  },
252
  {
253
+ "epoch": 63.366336633663366,
254
+ "grad_norm": 0.7239477634429932,
255
  "learning_rate": 8.000000000000001e-06,
256
+ "loss": 0.4245,
257
  "step": 1600
258
  },
259
  {
260
+ "epoch": 65.34653465346534,
261
+ "grad_norm": 0.7235706448554993,
262
  "learning_rate": 8.25e-06,
263
+ "loss": 0.4248,
264
  "step": 1650
265
  },
266
  {
267
+ "epoch": 67.32673267326733,
268
+ "grad_norm": 0.7281640768051147,
269
  "learning_rate": 8.5e-06,
270
+ "loss": 0.425,
271
  "step": 1700
272
  },
273
  {
274
+ "epoch": 69.3069306930693,
275
+ "grad_norm": 1.011294960975647,
276
  "learning_rate": 8.750000000000001e-06,
277
+ "loss": 0.4238,
278
  "step": 1750
279
  },
280
  {
281
+ "epoch": 71.2871287128713,
282
+ "grad_norm": 1.1249107122421265,
283
  "learning_rate": 9e-06,
284
+ "loss": 0.4203,
285
  "step": 1800
286
  },
287
  {
288
+ "epoch": 73.26732673267327,
289
+ "grad_norm": 0.7369679808616638,
290
  "learning_rate": 9.250000000000001e-06,
291
+ "loss": 0.422,
292
  "step": 1850
293
  },
294
  {
295
+ "epoch": 75.24752475247524,
296
+ "grad_norm": 0.704367995262146,
297
  "learning_rate": 9.5e-06,
298
+ "loss": 0.4201,
299
  "step": 1900
300
  },
301
  {
302
+ "epoch": 77.22772277227723,
303
+ "grad_norm": 0.8307437896728516,
304
  "learning_rate": 9.75e-06,
305
+ "loss": 0.4203,
306
  "step": 1950
307
  },
308
  {
309
+ "epoch": 79.20792079207921,
310
+ "grad_norm": 0.7827133536338806,
311
  "learning_rate": 1e-05,
312
+ "loss": 0.4195,
313
  "step": 2000
314
  },
315
  {
316
+ "epoch": 79.20792079207921,
317
+ "eval_loss": 0.39878609776496887,
318
+ "eval_runtime": 8.1579,
319
+ "eval_samples_per_second": 22.064,
320
+ "eval_steps_per_second": 2.819,
321
  "step": 2000
322
  }
323
  ],
324
  "logging_steps": 50,
325
  "max_steps": 8000,
326
  "num_input_tokens_seen": 0,
327
+ "num_train_epochs": 320,
328
  "save_steps": 500,
329
  "stateful_callbacks": {
330
  "TrainerControl": {
 
338
  "attributes": {}
339
  }
340
  },
341
+ "total_flos": 2.161155031107257e+16,
342
+ "train_batch_size": 16,
343
  "trial_name": null,
344
  "trial_params": null
345
  }
checkpoint-2000/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f95de2b194fe7f16ec10fbf1b93d83e04c25498d479793b9a5748c1d338bfbd0
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e97537a0fb344678f807f30d02bd561fb4b5c2162a68338251550dd3ef50d20
3
  size 5304