File size: 7,328 Bytes
0c1c611
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 16464,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.09110787172011661,
      "grad_norm": 0.5542775988578796,
      "learning_rate": 4.848153547133139e-05,
      "loss": 7.6369,
      "step": 500
    },
    {
      "epoch": 0.18221574344023322,
      "grad_norm": 0.6718146800994873,
      "learning_rate": 4.696307094266278e-05,
      "loss": 6.0229,
      "step": 1000
    },
    {
      "epoch": 0.27332361516034986,
      "grad_norm": 0.6730484366416931,
      "learning_rate": 4.5444606413994175e-05,
      "loss": 5.5062,
      "step": 1500
    },
    {
      "epoch": 0.36443148688046645,
      "grad_norm": 0.7009643912315369,
      "learning_rate": 4.3926141885325564e-05,
      "loss": 5.2064,
      "step": 2000
    },
    {
      "epoch": 0.4555393586005831,
      "grad_norm": 0.7892385721206665,
      "learning_rate": 4.240767735665695e-05,
      "loss": 4.9923,
      "step": 2500
    },
    {
      "epoch": 0.5466472303206997,
      "grad_norm": 0.840638279914856,
      "learning_rate": 4.088921282798834e-05,
      "loss": 4.8437,
      "step": 3000
    },
    {
      "epoch": 0.6377551020408163,
      "grad_norm": 0.8709360957145691,
      "learning_rate": 3.937074829931973e-05,
      "loss": 4.7268,
      "step": 3500
    },
    {
      "epoch": 0.7288629737609329,
      "grad_norm": 0.9013066291809082,
      "learning_rate": 3.785228377065112e-05,
      "loss": 4.6324,
      "step": 4000
    },
    {
      "epoch": 0.8199708454810496,
      "grad_norm": 0.9396289587020874,
      "learning_rate": 3.6333819241982507e-05,
      "loss": 4.556,
      "step": 4500
    },
    {
      "epoch": 0.9110787172011662,
      "grad_norm": 1.0571647882461548,
      "learning_rate": 3.4815354713313895e-05,
      "loss": 4.4846,
      "step": 5000
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.23895378723078822,
      "eval_loss": 4.422997951507568,
      "eval_runtime": 213.5442,
      "eval_samples_per_second": 89.185,
      "eval_steps_per_second": 2.791,
      "step": 5488
    },
    {
      "epoch": 1.0021865889212829,
      "grad_norm": 0.9928910732269287,
      "learning_rate": 3.329689018464529e-05,
      "loss": 4.4267,
      "step": 5500
    },
    {
      "epoch": 1.0932944606413995,
      "grad_norm": 1.0094648599624634,
      "learning_rate": 3.177842565597668e-05,
      "loss": 4.3598,
      "step": 6000
    },
    {
      "epoch": 1.184402332361516,
      "grad_norm": 1.1073040962219238,
      "learning_rate": 3.0259961127308068e-05,
      "loss": 4.3227,
      "step": 6500
    },
    {
      "epoch": 1.2755102040816326,
      "grad_norm": 1.0472743511199951,
      "learning_rate": 2.8741496598639456e-05,
      "loss": 4.2846,
      "step": 7000
    },
    {
      "epoch": 1.3666180758017492,
      "grad_norm": 1.0852700471878052,
      "learning_rate": 2.7223032069970845e-05,
      "loss": 4.2512,
      "step": 7500
    },
    {
      "epoch": 1.4577259475218658,
      "grad_norm": 1.0805084705352783,
      "learning_rate": 2.5704567541302237e-05,
      "loss": 4.2266,
      "step": 8000
    },
    {
      "epoch": 1.5488338192419824,
      "grad_norm": 1.1083316802978516,
      "learning_rate": 2.4186103012633625e-05,
      "loss": 4.1916,
      "step": 8500
    },
    {
      "epoch": 1.639941690962099,
      "grad_norm": 1.1683896780014038,
      "learning_rate": 2.2667638483965014e-05,
      "loss": 4.1699,
      "step": 9000
    },
    {
      "epoch": 1.7310495626822158,
      "grad_norm": 1.2223966121673584,
      "learning_rate": 2.1149173955296406e-05,
      "loss": 4.1517,
      "step": 9500
    },
    {
      "epoch": 1.8221574344023324,
      "grad_norm": 1.1384276151657104,
      "learning_rate": 1.9630709426627795e-05,
      "loss": 4.1228,
      "step": 10000
    },
    {
      "epoch": 1.913265306122449,
      "grad_norm": 1.2079554796218872,
      "learning_rate": 1.8112244897959187e-05,
      "loss": 4.1079,
      "step": 10500
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.27172573472254197,
      "eval_loss": 4.11556339263916,
      "eval_runtime": 213.5538,
      "eval_samples_per_second": 89.181,
      "eval_steps_per_second": 2.791,
      "step": 10976
    },
    {
      "epoch": 2.0043731778425657,
      "grad_norm": 1.2157140970230103,
      "learning_rate": 1.6593780369290575e-05,
      "loss": 4.0902,
      "step": 11000
    },
    {
      "epoch": 2.0954810495626823,
      "grad_norm": 1.2627109289169312,
      "learning_rate": 1.5075315840621965e-05,
      "loss": 4.0556,
      "step": 11500
    },
    {
      "epoch": 2.186588921282799,
      "grad_norm": 1.2118178606033325,
      "learning_rate": 1.3556851311953352e-05,
      "loss": 4.0461,
      "step": 12000
    },
    {
      "epoch": 2.2776967930029155,
      "grad_norm": 1.3033051490783691,
      "learning_rate": 1.2038386783284743e-05,
      "loss": 4.0326,
      "step": 12500
    },
    {
      "epoch": 2.368804664723032,
      "grad_norm": 1.2471331357955933,
      "learning_rate": 1.0519922254616133e-05,
      "loss": 4.0282,
      "step": 13000
    },
    {
      "epoch": 2.4599125364431487,
      "grad_norm": 1.3048267364501953,
      "learning_rate": 9.001457725947522e-06,
      "loss": 4.0197,
      "step": 13500
    },
    {
      "epoch": 2.5510204081632653,
      "grad_norm": 1.1969304084777832,
      "learning_rate": 7.482993197278912e-06,
      "loss": 4.0076,
      "step": 14000
    },
    {
      "epoch": 2.642128279883382,
      "grad_norm": 1.264858603477478,
      "learning_rate": 5.964528668610301e-06,
      "loss": 3.9948,
      "step": 14500
    },
    {
      "epoch": 2.7332361516034984,
      "grad_norm": 1.244125485420227,
      "learning_rate": 4.4460641399416915e-06,
      "loss": 3.9953,
      "step": 15000
    },
    {
      "epoch": 2.824344023323615,
      "grad_norm": 1.2413588762283325,
      "learning_rate": 2.9275996112730806e-06,
      "loss": 3.9837,
      "step": 15500
    },
    {
      "epoch": 2.9154518950437316,
      "grad_norm": 1.233183741569519,
      "learning_rate": 1.4091350826044704e-06,
      "loss": 3.9841,
      "step": 16000
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.2825701437173418,
      "eval_loss": 4.0273261070251465,
      "eval_runtime": 215.4599,
      "eval_samples_per_second": 88.392,
      "eval_steps_per_second": 2.766,
      "step": 16464
    },
    {
      "epoch": 3.0,
      "step": 16464,
      "total_flos": 1.44589665695957e+17,
      "train_loss": 4.469071887788319,
      "train_runtime": 13360.3551,
      "train_samples_per_second": 39.431,
      "train_steps_per_second": 1.232
    }
  ],
  "logging_steps": 500,
  "max_steps": 16464,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 1.44589665695957e+17,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}