File size: 8,362 Bytes
f039cc3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 220,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.022779043280182234,
      "grad_norm": 1.0761960744857788,
      "learning_rate": 2.181818181818182e-06,
      "loss": 1.4131,
      "step": 5
    },
    {
      "epoch": 0.04555808656036447,
      "grad_norm": 0.7671697735786438,
      "learning_rate": 4.90909090909091e-06,
      "loss": 1.3993,
      "step": 10
    },
    {
      "epoch": 0.0683371298405467,
      "grad_norm": 0.5325023531913757,
      "learning_rate": 7.636363636363636e-06,
      "loss": 1.3575,
      "step": 15
    },
    {
      "epoch": 0.09111617312072894,
      "grad_norm": 0.5349851846694946,
      "learning_rate": 1.0363636363636364e-05,
      "loss": 1.3727,
      "step": 20
    },
    {
      "epoch": 0.11389521640091116,
      "grad_norm": 0.5694962739944458,
      "learning_rate": 1.309090909090909e-05,
      "loss": 1.3057,
      "step": 25
    },
    {
      "epoch": 0.1366742596810934,
      "grad_norm": 0.541211724281311,
      "learning_rate": 1.5818181818181818e-05,
      "loss": 1.2344,
      "step": 30
    },
    {
      "epoch": 0.15945330296127563,
      "grad_norm": 0.5115331411361694,
      "learning_rate": 1.8545454545454545e-05,
      "loss": 1.3365,
      "step": 35
    },
    {
      "epoch": 0.18223234624145787,
      "grad_norm": 0.46041619777679443,
      "learning_rate": 2.1272727272727273e-05,
      "loss": 1.2517,
      "step": 40
    },
    {
      "epoch": 0.20501138952164008,
      "grad_norm": 0.4588950276374817,
      "learning_rate": 2.4e-05,
      "loss": 1.233,
      "step": 45
    },
    {
      "epoch": 0.22779043280182232,
      "grad_norm": 0.571067750453949,
      "learning_rate": 2.6727272727272728e-05,
      "loss": 1.1614,
      "step": 50
    },
    {
      "epoch": 0.2505694760820046,
      "grad_norm": 0.47105151414871216,
      "learning_rate": 2.9454545454545456e-05,
      "loss": 1.1789,
      "step": 55
    },
    {
      "epoch": 0.2733485193621868,
      "grad_norm": 0.5167170763015747,
      "learning_rate": 2.9998915465986464e-05,
      "loss": 1.1674,
      "step": 60
    },
    {
      "epoch": 0.296127562642369,
      "grad_norm": 0.5363845229148865,
      "learning_rate": 2.999450981533838e-05,
      "loss": 1.1646,
      "step": 65
    },
    {
      "epoch": 0.31890660592255127,
      "grad_norm": 0.5026624798774719,
      "learning_rate": 2.9986716259340288e-05,
      "loss": 1.1268,
      "step": 70
    },
    {
      "epoch": 0.3416856492027335,
      "grad_norm": 0.46400442719459534,
      "learning_rate": 2.9975536558892034e-05,
      "loss": 1.0501,
      "step": 75
    },
    {
      "epoch": 0.36446469248291574,
      "grad_norm": 0.6948593258857727,
      "learning_rate": 2.9960973239969295e-05,
      "loss": 1.0567,
      "step": 80
    },
    {
      "epoch": 0.38724373576309795,
      "grad_norm": 0.6658697724342346,
      "learning_rate": 2.9943029593052822e-05,
      "loss": 1.1199,
      "step": 85
    },
    {
      "epoch": 0.41002277904328016,
      "grad_norm": 0.5796182155609131,
      "learning_rate": 2.992170967238502e-05,
      "loss": 1.0407,
      "step": 90
    },
    {
      "epoch": 0.4328018223234624,
      "grad_norm": 0.5862748622894287,
      "learning_rate": 2.9897018295053883e-05,
      "loss": 1.0524,
      "step": 95
    },
    {
      "epoch": 0.45558086560364464,
      "grad_norm": 0.714149534702301,
      "learning_rate": 2.9868961039904628e-05,
      "loss": 1.0367,
      "step": 100
    },
    {
      "epoch": 0.4783599088838269,
      "grad_norm": 0.7176491618156433,
      "learning_rate": 2.983754424627919e-05,
      "loss": 1.0013,
      "step": 105
    },
    {
      "epoch": 0.5011389521640092,
      "grad_norm": 0.6981073617935181,
      "learning_rate": 2.9802775012583884e-05,
      "loss": 1.0079,
      "step": 110
    },
    {
      "epoch": 0.5239179954441914,
      "grad_norm": 0.5984258651733398,
      "learning_rate": 2.9764661194685583e-05,
      "loss": 0.975,
      "step": 115
    },
    {
      "epoch": 0.5466970387243736,
      "grad_norm": 0.7986994981765747,
      "learning_rate": 2.972321140413672e-05,
      "loss": 0.9417,
      "step": 120
    },
    {
      "epoch": 0.5694760820045558,
      "grad_norm": 0.6552079916000366,
      "learning_rate": 2.9678435006229585e-05,
      "loss": 0.9208,
      "step": 125
    },
    {
      "epoch": 0.592255125284738,
      "grad_norm": 0.7615071535110474,
      "learning_rate": 2.9630342117880293e-05,
      "loss": 0.9208,
      "step": 130
    },
    {
      "epoch": 0.6150341685649203,
      "grad_norm": 0.73598313331604,
      "learning_rate": 2.957894360534295e-05,
      "loss": 0.9459,
      "step": 135
    },
    {
      "epoch": 0.6378132118451025,
      "grad_norm": 0.9029515981674194,
      "learning_rate": 2.9524251081754475e-05,
      "loss": 0.9338,
      "step": 140
    },
    {
      "epoch": 0.6605922551252847,
      "grad_norm": 0.8563511967658997,
      "learning_rate": 2.9466276904510713e-05,
      "loss": 0.8437,
      "step": 145
    },
    {
      "epoch": 0.683371298405467,
      "grad_norm": 0.8966438174247742,
      "learning_rate": 2.9405034172474363e-05,
      "loss": 0.8995,
      "step": 150
    },
    {
      "epoch": 0.7061503416856492,
      "grad_norm": 0.9085351228713989,
      "learning_rate": 2.9340536723015367e-05,
      "loss": 0.8988,
      "step": 155
    },
    {
      "epoch": 0.7289293849658315,
      "grad_norm": 0.920749843120575,
      "learning_rate": 2.927279912888447e-05,
      "loss": 0.8457,
      "step": 160
    },
    {
      "epoch": 0.7517084282460137,
      "grad_norm": 0.844048023223877,
      "learning_rate": 2.920183669492061e-05,
      "loss": 0.8699,
      "step": 165
    },
    {
      "epoch": 0.7744874715261959,
      "grad_norm": 1.1362037658691406,
      "learning_rate": 2.9127665454592872e-05,
      "loss": 0.8561,
      "step": 170
    },
    {
      "epoch": 0.7972665148063781,
      "grad_norm": 0.9490585923194885,
      "learning_rate": 2.9050302166377858e-05,
      "loss": 0.8016,
      "step": 175
    },
    {
      "epoch": 0.8200455580865603,
      "grad_norm": 1.047890543937683,
      "learning_rate": 2.896976430997323e-05,
      "loss": 0.8047,
      "step": 180
    },
    {
      "epoch": 0.8428246013667426,
      "grad_norm": 1.252584457397461,
      "learning_rate": 2.8886070082348268e-05,
      "loss": 0.8087,
      "step": 185
    },
    {
      "epoch": 0.8656036446469249,
      "grad_norm": 1.0744786262512207,
      "learning_rate": 2.879923839363242e-05,
      "loss": 0.7823,
      "step": 190
    },
    {
      "epoch": 0.8883826879271071,
      "grad_norm": 0.9743952751159668,
      "learning_rate": 2.870928886284267e-05,
      "loss": 0.727,
      "step": 195
    },
    {
      "epoch": 0.9111617312072893,
      "grad_norm": 1.029543399810791,
      "learning_rate": 2.8616241813450755e-05,
      "loss": 0.7648,
      "step": 200
    },
    {
      "epoch": 0.9339407744874715,
      "grad_norm": 1.031134009361267,
      "learning_rate": 2.852011826879125e-05,
      "loss": 0.7444,
      "step": 205
    },
    {
      "epoch": 0.9567198177676538,
      "grad_norm": 1.1572555303573608,
      "learning_rate": 2.8420939947311454e-05,
      "loss": 0.7408,
      "step": 210
    },
    {
      "epoch": 0.979498861047836,
      "grad_norm": 1.0397342443466187,
      "learning_rate": 2.8318729257664265e-05,
      "loss": 0.7015,
      "step": 215
    },
    {
      "epoch": 1.0,
      "grad_norm": 1.4852145910263062,
      "learning_rate": 2.821350929364512e-05,
      "loss": 0.6917,
      "step": 220
    }
  ],
  "logging_steps": 5,
  "max_steps": 1100,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 2000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 3.180798646723543e+17,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}