File size: 6,976 Bytes
6f51c81
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "best_metric": 0.1676628440618515,
  "best_model_checkpoint": "saves/Qwen1.5-1.8B/WordProblem/checkpoint-9000",
  "epoch": 0.9999725884707108,
  "eval_steps": 1500,
  "global_step": 9120,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.03289383514706285,
      "grad_norm": 4.8125,
      "learning_rate": 3e-05,
      "loss": 0.7143,
      "step": 300
    },
    {
      "epoch": 0.0657876702941257,
      "grad_norm": 3.5,
      "learning_rate": 4.998339850669331e-05,
      "loss": 0.2219,
      "step": 600
    },
    {
      "epoch": 0.09868150544118856,
      "grad_norm": 4.40625,
      "learning_rate": 4.9734816848192624e-05,
      "loss": 0.2074,
      "step": 900
    },
    {
      "epoch": 0.1315753405882514,
      "grad_norm": 4.03125,
      "learning_rate": 4.9190839785031474e-05,
      "loss": 0.1906,
      "step": 1200
    },
    {
      "epoch": 0.16446917573531428,
      "grad_norm": 3.78125,
      "learning_rate": 4.835796376008569e-05,
      "loss": 0.1923,
      "step": 1500
    },
    {
      "epoch": 0.16446917573531428,
      "eval_loss": 0.1849033087491989,
      "eval_runtime": 79.6037,
      "eval_samples_per_second": 37.423,
      "eval_steps_per_second": 9.359,
      "step": 1500
    },
    {
      "epoch": 0.19736301088237712,
      "grad_norm": 4.15625,
      "learning_rate": 4.7246135390382216e-05,
      "loss": 0.1839,
      "step": 1800
    },
    {
      "epoch": 0.23025684602944,
      "grad_norm": 4.5,
      "learning_rate": 4.586863267968384e-05,
      "loss": 0.1938,
      "step": 2100
    },
    {
      "epoch": 0.2631506811765028,
      "grad_norm": 3.1875,
      "learning_rate": 4.4241906446007296e-05,
      "loss": 0.1863,
      "step": 2400
    },
    {
      "epoch": 0.2960445163235657,
      "grad_norm": 2.71875,
      "learning_rate": 4.238538385782601e-05,
      "loss": 0.1797,
      "step": 2700
    },
    {
      "epoch": 0.32893835147062855,
      "grad_norm": 3.65625,
      "learning_rate": 4.032123642522486e-05,
      "loss": 0.176,
      "step": 3000
    },
    {
      "epoch": 0.32893835147062855,
      "eval_loss": 0.1760552078485489,
      "eval_runtime": 79.6333,
      "eval_samples_per_second": 37.409,
      "eval_steps_per_second": 9.355,
      "step": 3000
    },
    {
      "epoch": 0.3618321866176914,
      "grad_norm": 4.0,
      "learning_rate": 3.8074115216771435e-05,
      "loss": 0.1791,
      "step": 3300
    },
    {
      "epoch": 0.39472602176475424,
      "grad_norm": 3.90625,
      "learning_rate": 3.567085646427478e-05,
      "loss": 0.1808,
      "step": 3600
    },
    {
      "epoch": 0.4276198569118171,
      "grad_norm": 3.421875,
      "learning_rate": 3.3140161071244915e-05,
      "loss": 0.1805,
      "step": 3900
    },
    {
      "epoch": 0.46051369205888,
      "grad_norm": 2.640625,
      "learning_rate": 3.05122518525215e-05,
      "loss": 0.1738,
      "step": 4200
    },
    {
      "epoch": 0.49340752720594283,
      "grad_norm": 4.5,
      "learning_rate": 2.781851259848554e-05,
      "loss": 0.1736,
      "step": 4500
    },
    {
      "epoch": 0.49340752720594283,
      "eval_loss": 0.17090687155723572,
      "eval_runtime": 79.6329,
      "eval_samples_per_second": 37.409,
      "eval_steps_per_second": 9.355,
      "step": 4500
    },
    {
      "epoch": 0.5263013623530056,
      "grad_norm": 3.578125,
      "learning_rate": 2.509111327432736e-05,
      "loss": 0.1709,
      "step": 4800
    },
    {
      "epoch": 0.5591951975000685,
      "grad_norm": 3.515625,
      "learning_rate": 2.236262583042668e-05,
      "loss": 0.1775,
      "step": 5100
    },
    {
      "epoch": 0.5920890326471314,
      "grad_norm": 4.9375,
      "learning_rate": 1.966563521202681e-05,
      "loss": 0.1759,
      "step": 5400
    },
    {
      "epoch": 0.6249828677941942,
      "grad_norm": 3.640625,
      "learning_rate": 1.7032350213717874e-05,
      "loss": 0.1754,
      "step": 5700
    },
    {
      "epoch": 0.6578767029412571,
      "grad_norm": 3.578125,
      "learning_rate": 1.4494218826096939e-05,
      "loss": 0.1688,
      "step": 6000
    },
    {
      "epoch": 0.6578767029412571,
      "eval_loss": 0.16823573410511017,
      "eval_runtime": 79.6163,
      "eval_samples_per_second": 37.417,
      "eval_steps_per_second": 9.357,
      "step": 6000
    },
    {
      "epoch": 0.6907705380883199,
      "grad_norm": 3.6875,
      "learning_rate": 1.2081552668325321e-05,
      "loss": 0.1707,
      "step": 6300
    },
    {
      "epoch": 0.7236643732353828,
      "grad_norm": 4.0625,
      "learning_rate": 9.82316499179518e-06,
      "loss": 0.171,
      "step": 6600
    },
    {
      "epoch": 0.7565582083824457,
      "grad_norm": 3.984375,
      "learning_rate": 7.74602657804425e-06,
      "loss": 0.1702,
      "step": 6900
    },
    {
      "epoch": 0.7894520435295085,
      "grad_norm": 3.421875,
      "learning_rate": 5.874943640356082e-06,
      "loss": 0.1718,
      "step": 7200
    },
    {
      "epoch": 0.8223458786765714,
      "grad_norm": 4.1875,
      "learning_rate": 4.232261575703861e-06,
      "loss": 0.1689,
      "step": 7500
    },
    {
      "epoch": 0.8223458786765714,
      "eval_loss": 0.16773280501365662,
      "eval_runtime": 79.6198,
      "eval_samples_per_second": 37.415,
      "eval_steps_per_second": 9.357,
      "step": 7500
    },
    {
      "epoch": 0.8552397138236342,
      "grad_norm": 2.703125,
      "learning_rate": 2.83759810497852e-06,
      "loss": 0.1692,
      "step": 7800
    },
    {
      "epoch": 0.8881335489706971,
      "grad_norm": 4.0625,
      "learning_rate": 1.70760898847247e-06,
      "loss": 0.1787,
      "step": 8100
    },
    {
      "epoch": 0.92102738411776,
      "grad_norm": 4.71875,
      "learning_rate": 8.557891145603042e-07,
      "loss": 0.1733,
      "step": 8400
    },
    {
      "epoch": 0.9539212192648228,
      "grad_norm": 3.6875,
      "learning_rate": 2.923113370737779e-07,
      "loss": 0.1741,
      "step": 8700
    },
    {
      "epoch": 0.9868150544118857,
      "grad_norm": 4.3125,
      "learning_rate": 2.3904986054812396e-08,
      "loss": 0.168,
      "step": 9000
    },
    {
      "epoch": 0.9868150544118857,
      "eval_loss": 0.1676628440618515,
      "eval_runtime": 79.7085,
      "eval_samples_per_second": 37.374,
      "eval_steps_per_second": 9.347,
      "step": 9000
    },
    {
      "epoch": 0.9999725884707108,
      "step": 9120,
      "total_flos": 3.6976201313039155e+17,
      "train_loss": 0.19690959160788019,
      "train_runtime": 20589.0374,
      "train_samples_per_second": 7.087,
      "train_steps_per_second": 0.443
    }
  ],
  "logging_steps": 300,
  "max_steps": 9120,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 3000,
  "total_flos": 3.6976201313039155e+17,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}