Attila1011 commited on
Commit
0a8aace
·
verified ·
1 Parent(s): e8ad0f5

Upload folder using huggingface_hub

Browse files
checkpoints-v2.5-new/checkpoint-21651/ema.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d34b0fb85701de16c04f8fd97b9cde9cc0d4640633b0ad92d2a58bee5e9cf5b6
3
+ size 550088
checkpoints-v2.5-new/checkpoint-21651/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2335ec8c671566bb5721c2a100a66275cc6739a13265bd9b79af71cbf7c9ddca
3
+ size 19318904
checkpoints-v2.5-new/checkpoint-21651/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c70933be3c05787310b03d258dd07afaae638226f0abd4d931812d125df5adc
3
+ size 1175115
checkpoints-v2.5-new/checkpoint-21651/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5e46d1e941b1e6770d70eb593b1d56f6673c57d4816ab6f614779a84f1a34c0
3
+ size 14645
checkpoints-v2.5-new/checkpoint-21651/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9645151424e5d1ef2c6d53d37ad4fc952ba7c45a02f85af465a7e8d4c6e613f5
3
+ size 1383
checkpoints-v2.5-new/checkpoint-21651/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c22ca73b4d580b1007c96d857f375acf2f3df919e1dd4b9b2e67e1edd0d3120
3
+ size 1465
checkpoints-v2.5-new/checkpoint-21651/trainer_state.json ADDED
@@ -0,0 +1,1147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 1024,
7
+ "global_step": 21651,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.011823934229365849,
14
+ "grad_norm": 0.24459639191627502,
15
+ "learning_rate": 0.000498046875,
16
+ "loss": 1.7330865859985352,
17
+ "step": 256
18
+ },
19
+ {
20
+ "epoch": 0.023647868458731697,
21
+ "grad_norm": 0.15448841452598572,
22
+ "learning_rate": 0.000998046875,
23
+ "loss": 1.177037000656128,
24
+ "step": 512
25
+ },
26
+ {
27
+ "epoch": 0.03547180268809755,
28
+ "grad_norm": 0.17487554252147675,
29
+ "learning_rate": 0.000999640996023194,
30
+ "loss": 1.0753283500671387,
31
+ "step": 768
32
+ },
33
+ {
34
+ "epoch": 0.047295736917463395,
35
+ "grad_norm": 0.12926897406578064,
36
+ "learning_rate": 0.0009985588674043958,
37
+ "loss": 1.0295459032058716,
38
+ "step": 1024
39
+ },
40
+ {
41
+ "epoch": 0.047295736917463395,
42
+ "eval_cos_loss": 0.2935629785741301,
43
+ "eval_loss": 0.9888965673914784,
44
+ "eval_mse_loss": 0.9888965673914784,
45
+ "flow/cos_sim": 0.7064370480302262,
46
+ "flow/improvement_ratio": 0.47962706260485194,
47
+ "flow/mag_ratio_mean": 0.7049914837972214,
48
+ "flow/mag_ratio_std": 0.14915283825503636,
49
+ "step": 1024
50
+ },
51
+ {
52
+ "epoch": 0.047295736917463395,
53
+ "eval_cos_loss": 0.2935629785741301,
54
+ "eval_loss": 0.9888965673914784,
55
+ "eval_mse_loss": 0.9888965673914784,
56
+ "eval_runtime": 38.2456,
57
+ "eval_samples_per_second": 731.927,
58
+ "eval_steps_per_second": 11.452,
59
+ "flow/cos_sim": 0.7064370480302262,
60
+ "flow/improvement_ratio": 0.47962706260485194,
61
+ "flow/mag_ratio_mean": 0.7049914837972214,
62
+ "flow/mag_ratio_std": 0.14915283825503636,
63
+ "step": 1024
64
+ },
65
+ {
66
+ "epoch": 0.05911967114682925,
67
+ "grad_norm": 0.11577294021844864,
68
+ "learning_rate": 0.0009967551747861387,
69
+ "loss": 1.0038137435913086,
70
+ "step": 1280
71
+ },
72
+ {
73
+ "epoch": 0.0709436053761951,
74
+ "grad_norm": 0.09389752149581909,
75
+ "learning_rate": 0.000994232528651847,
76
+ "loss": 0.9830185174942017,
77
+ "step": 1536
78
+ },
79
+ {
80
+ "epoch": 0.08276753960556095,
81
+ "grad_norm": 0.29331570863723755,
82
+ "learning_rate": 0.0009909945800260092,
83
+ "loss": 0.9637426137924194,
84
+ "step": 1792
85
+ },
86
+ {
87
+ "epoch": 0.09459147383492679,
88
+ "grad_norm": 0.2393321692943573,
89
+ "learning_rate": 0.0009870460151900522,
90
+ "loss": 0.9512394666671753,
91
+ "step": 2048
92
+ },
93
+ {
94
+ "epoch": 0.09459147383492679,
95
+ "eval_cos_loss": 0.266197642541095,
96
+ "eval_loss": 0.9073396895301941,
97
+ "eval_mse_loss": 0.9073396895301941,
98
+ "flow/cos_sim": 0.7338023810353997,
99
+ "flow/improvement_ratio": 0.4802392233316212,
100
+ "flow/mag_ratio_mean": 0.7257989417744554,
101
+ "flow/mag_ratio_std": 0.1568604064552479,
102
+ "step": 2048
103
+ },
104
+ {
105
+ "epoch": 0.09459147383492679,
106
+ "eval_cos_loss": 0.266197642541095,
107
+ "eval_loss": 0.9073396895301941,
108
+ "eval_mse_loss": 0.9073396895301941,
109
+ "eval_runtime": 38.2623,
110
+ "eval_samples_per_second": 731.607,
111
+ "eval_steps_per_second": 11.447,
112
+ "flow/cos_sim": 0.7338023810353997,
113
+ "flow/improvement_ratio": 0.4802392233316212,
114
+ "flow/mag_ratio_mean": 0.7257989417744554,
115
+ "flow/mag_ratio_std": 0.1568604064552479,
116
+ "step": 2048
117
+ },
118
+ {
119
+ "epoch": 0.10641540806429264,
120
+ "grad_norm": 0.26130712032318115,
121
+ "learning_rate": 0.0009823925488998885,
122
+ "loss": 0.9370531439781189,
123
+ "step": 2304
124
+ },
125
+ {
126
+ "epoch": 0.1182393422936585,
127
+ "grad_norm": 0.49878937005996704,
128
+ "learning_rate": 0.0009770409161149525,
129
+ "loss": 0.9286154508590698,
130
+ "step": 2560
131
+ },
132
+ {
133
+ "epoch": 0.13006327652302435,
134
+ "grad_norm": 0.1462121456861496,
135
+ "learning_rate": 0.0009709988622506973,
136
+ "loss": 0.9175565838813782,
137
+ "step": 2816
138
+ },
139
+ {
140
+ "epoch": 0.1418872107523902,
141
+ "grad_norm": 0.49823057651519775,
142
+ "learning_rate": 0.000964275131968659,
143
+ "loss": 0.909263014793396,
144
+ "step": 3072
145
+ },
146
+ {
147
+ "epoch": 0.1418872107523902,
148
+ "eval_cos_loss": 0.25432631951761026,
149
+ "eval_loss": 0.8673899229530875,
150
+ "eval_mse_loss": 0.8673899229530875,
151
+ "flow/cos_sim": 0.7456736926618777,
152
+ "flow/improvement_ratio": 0.4867378263713018,
153
+ "flow/mag_ratio_mean": 0.741756375793997,
154
+ "flow/mag_ratio_std": 0.16158553437419135,
155
+ "step": 3072
156
+ },
157
+ {
158
+ "epoch": 0.1418872107523902,
159
+ "eval_cos_loss": 0.25432631951761026,
160
+ "eval_loss": 0.8673899229530875,
161
+ "eval_mse_loss": 0.8673899229530875,
162
+ "eval_runtime": 38.3091,
163
+ "eval_samples_per_second": 730.713,
164
+ "eval_steps_per_second": 11.433,
165
+ "flow/cos_sim": 0.7456736926618777,
166
+ "flow/improvement_ratio": 0.4867378263713018,
167
+ "flow/mag_ratio_mean": 0.741756375793997,
168
+ "flow/mag_ratio_std": 0.16158553437419135,
169
+ "step": 3072
170
+ },
171
+ {
172
+ "epoch": 0.15371114498175603,
173
+ "grad_norm": 0.2198517918586731,
174
+ "learning_rate": 0.0009568794565203123,
175
+ "loss": 0.9027122855186462,
176
+ "step": 3328
177
+ },
178
+ {
179
+ "epoch": 0.1655350792111219,
180
+ "grad_norm": 0.4902757406234741,
181
+ "learning_rate": 0.0009488225396630347,
182
+ "loss": 0.8999761939048767,
183
+ "step": 3584
184
+ },
185
+ {
186
+ "epoch": 0.17735901344048774,
187
+ "grad_norm": 0.7366757988929749,
188
+ "learning_rate": 0.0009401160421685646,
189
+ "loss": 0.8889837861061096,
190
+ "step": 3840
191
+ },
192
+ {
193
+ "epoch": 0.18918294766985358,
194
+ "grad_norm": 0.6776683926582336,
195
+ "learning_rate": 0.0009307725649463714,
196
+ "loss": 0.8910129070281982,
197
+ "step": 4096
198
+ },
199
+ {
200
+ "epoch": 0.18918294766985358,
201
+ "eval_cos_loss": 0.24676692009516502,
202
+ "eval_loss": 0.8415617307299348,
203
+ "eval_mse_loss": 0.8415617307299348,
204
+ "flow/cos_sim": 0.7532330962348747,
205
+ "flow/improvement_ratio": 0.47731589372843913,
206
+ "flow/mag_ratio_mean": 0.7478010703439582,
207
+ "flow/mag_ratio_std": 0.16910598910154273,
208
+ "step": 4096
209
+ },
210
+ {
211
+ "epoch": 0.18918294766985358,
212
+ "eval_cos_loss": 0.24676692009516502,
213
+ "eval_loss": 0.8415617307299348,
214
+ "eval_mse_loss": 0.8415617307299348,
215
+ "eval_runtime": 38.8097,
216
+ "eval_samples_per_second": 721.288,
217
+ "eval_steps_per_second": 11.286,
218
+ "flow/cos_sim": 0.7532330962348747,
219
+ "flow/improvement_ratio": 0.47731589372843913,
220
+ "flow/mag_ratio_mean": 0.7478010703439582,
221
+ "flow/mag_ratio_std": 0.16910598910154273,
222
+ "step": 4096
223
+ },
224
+ {
225
+ "epoch": 0.20100688189921945,
226
+ "grad_norm": 0.12204498797655106,
227
+ "learning_rate": 0.0009208056308063659,
228
+ "loss": 0.8853418827056885,
229
+ "step": 4352
230
+ },
231
+ {
232
+ "epoch": 0.2128308161285853,
233
+ "grad_norm": 0.3798384368419647,
234
+ "learning_rate": 0.0009102296648873445,
235
+ "loss": 0.8796222805976868,
236
+ "step": 4608
237
+ },
238
+ {
239
+ "epoch": 0.22465475035795113,
240
+ "grad_norm": 0.34957996010780334,
241
+ "learning_rate": 0.0008990599737794927,
242
+ "loss": 0.8780807852745056,
243
+ "step": 4864
244
+ },
245
+ {
246
+ "epoch": 0.236478684587317,
247
+ "grad_norm": 0.13670794665813446,
248
+ "learning_rate": 0.0008873127233711644,
249
+ "loss": 0.8707529902458191,
250
+ "step": 5120
251
+ },
252
+ {
253
+ "epoch": 0.236478684587317,
254
+ "eval_cos_loss": 0.241777848503361,
255
+ "eval_loss": 0.8265683972127905,
256
+ "eval_mse_loss": 0.8265683972127905,
257
+ "flow/cos_sim": 0.7582221518368482,
258
+ "flow/improvement_ratio": 0.4770067110317483,
259
+ "flow/mag_ratio_mean": 0.7527041480149308,
260
+ "flow/mag_ratio_std": 0.16668922120713753,
261
+ "step": 5120
262
+ },
263
+ {
264
+ "epoch": 0.236478684587317,
265
+ "eval_cos_loss": 0.241777848503361,
266
+ "eval_loss": 0.8265683972127905,
267
+ "eval_mse_loss": 0.8265683972127905,
268
+ "eval_runtime": 38.3628,
269
+ "eval_samples_per_second": 729.691,
270
+ "eval_steps_per_second": 11.417,
271
+ "flow/cos_sim": 0.7582221518368482,
272
+ "flow/improvement_ratio": 0.4770067110317483,
273
+ "flow/mag_ratio_mean": 0.7527041480149308,
274
+ "flow/mag_ratio_std": 0.16668922120713753,
275
+ "step": 5120
276
+ },
277
+ {
278
+ "epoch": 0.24830261881668284,
279
+ "grad_norm": 0.31755343079566956,
280
+ "learning_rate": 0.0008750049154520011,
281
+ "loss": 0.8690189719200134,
282
+ "step": 5376
283
+ },
284
+ {
285
+ "epoch": 0.2601265530460487,
286
+ "grad_norm": 0.1687377691268921,
287
+ "learning_rate": 0.0008621543631062487,
288
+ "loss": 0.8690802454948425,
289
+ "step": 5632
290
+ },
291
+ {
292
+ "epoch": 0.27195048727541454,
293
+ "grad_norm": 0.13510888814926147,
294
+ "learning_rate": 0.0008487796649318904,
295
+ "loss": 0.8649267554283142,
296
+ "step": 5888
297
+ },
298
+ {
299
+ "epoch": 0.2837744215047804,
300
+ "grad_norm": 0.45792385935783386,
301
+ "learning_rate": 0.0008349001781229053,
302
+ "loss": 0.8642867803573608,
303
+ "step": 6144
304
+ },
305
+ {
306
+ "epoch": 0.2837744215047804,
307
+ "eval_cos_loss": 0.24099888016371968,
308
+ "eval_loss": 0.8225860546713006,
309
+ "eval_mse_loss": 0.8225860546713006,
310
+ "flow/cos_sim": 0.7590011426302941,
311
+ "flow/improvement_ratio": 0.4821603450600959,
312
+ "flow/mag_ratio_mean": 0.7519274261444127,
313
+ "flow/mag_ratio_std": 0.17307638475611875,
314
+ "step": 6144
315
+ },
316
+ {
317
+ "epoch": 0.2837744215047804,
318
+ "eval_cos_loss": 0.24099888016371968,
319
+ "eval_loss": 0.8225860546713006,
320
+ "eval_mse_loss": 0.8225860546713006,
321
+ "eval_runtime": 38.4064,
322
+ "eval_samples_per_second": 728.863,
323
+ "eval_steps_per_second": 11.404,
324
+ "flow/cos_sim": 0.7590011426302941,
325
+ "flow/improvement_ratio": 0.4821603450600959,
326
+ "flow/mag_ratio_mean": 0.7519274261444127,
327
+ "flow/mag_ratio_std": 0.17307638475611875,
328
+ "step": 6144
329
+ },
330
+ {
331
+ "epoch": 0.2955983557341462,
332
+ "grad_norm": 0.5180349946022034,
333
+ "learning_rate": 0.0008205359904536107,
334
+ "loss": 0.8578495979309082,
335
+ "step": 6400
336
+ },
337
+ {
338
+ "epoch": 0.30742228996351206,
339
+ "grad_norm": 1.0647313594818115,
340
+ "learning_rate": 0.0008057078912056363,
341
+ "loss": 0.8586434721946716,
342
+ "step": 6656
343
+ },
344
+ {
345
+ "epoch": 0.3192462241928779,
346
+ "grad_norm": 0.5194692015647888,
347
+ "learning_rate": 0.0007904978240451256,
348
+ "loss": 0.8578391671180725,
349
+ "step": 6912
350
+ },
351
+ {
352
+ "epoch": 0.3310701584222438,
353
+ "grad_norm": 0.643442690372467,
354
+ "learning_rate": 0.0007748085222964866,
355
+ "loss": 0.8566486835479736,
356
+ "step": 7168
357
+ },
358
+ {
359
+ "epoch": 0.3310701584222438,
360
+ "eval_cos_loss": 0.2396061379955784,
361
+ "eval_loss": 0.8176589157755516,
362
+ "eval_mse_loss": 0.8176589157755516,
363
+ "flow/cos_sim": 0.7603938767354782,
364
+ "flow/improvement_ratio": 0.48190105151912394,
365
+ "flow/mag_ratio_mean": 0.7536703851669346,
366
+ "flow/mag_ratio_std": 0.17139297301910783,
367
+ "step": 7168
368
+ },
369
+ {
370
+ "epoch": 0.3310701584222438,
371
+ "eval_cos_loss": 0.2396061379955784,
372
+ "eval_loss": 0.8176589157755516,
373
+ "eval_mse_loss": 0.8176589157755516,
374
+ "eval_runtime": 38.5206,
375
+ "eval_samples_per_second": 726.702,
376
+ "eval_steps_per_second": 11.371,
377
+ "flow/cos_sim": 0.7603938767354782,
378
+ "flow/improvement_ratio": 0.48190105151912394,
379
+ "flow/mag_ratio_mean": 0.7536703851669346,
380
+ "flow/mag_ratio_std": 0.17139297301910783,
381
+ "step": 7168
382
+ },
383
+ {
384
+ "epoch": 0.34289409265160964,
385
+ "grad_norm": 1.1865595579147339,
386
+ "learning_rate": 0.0007587214903109633,
387
+ "loss": 0.85623699426651,
388
+ "step": 7424
389
+ },
390
+ {
391
+ "epoch": 0.3547180268809755,
392
+ "grad_norm": 0.2217409312725067,
393
+ "learning_rate": 0.0007422600108419029,
394
+ "loss": 0.85430508852005,
395
+ "step": 7680
396
+ },
397
+ {
398
+ "epoch": 0.3665419611103413,
399
+ "grad_norm": 0.2605246603488922,
400
+ "learning_rate": 0.0007254479085803078,
401
+ "loss": 0.8524338603019714,
402
+ "step": 7936
403
+ },
404
+ {
405
+ "epoch": 0.37836589533970716,
406
+ "grad_norm": 0.5839515924453735,
407
+ "learning_rate": 0.000708309515673374,
408
+ "loss": 0.8523181080818176,
409
+ "step": 8192
410
+ },
411
+ {
412
+ "epoch": 0.37836589533970716,
413
+ "eval_cos_loss": 0.23637595976868722,
414
+ "eval_loss": 0.807374326197524,
415
+ "eval_mse_loss": 0.807374326197524,
416
+ "flow/cos_sim": 0.7636240394148108,
417
+ "flow/improvement_ratio": 0.4805144328777104,
418
+ "flow/mag_ratio_mean": 0.7592539517846826,
419
+ "flow/mag_ratio_std": 0.17078373482510378,
420
+ "step": 8192
421
+ },
422
+ {
423
+ "epoch": 0.37836589533970716,
424
+ "eval_cos_loss": 0.23637595976868722,
425
+ "eval_loss": 0.807374326197524,
426
+ "eval_mse_loss": 0.807374326197524,
427
+ "eval_runtime": 38.4795,
428
+ "eval_samples_per_second": 727.479,
429
+ "eval_steps_per_second": 11.383,
430
+ "flow/cos_sim": 0.7636240394148108,
431
+ "flow/improvement_ratio": 0.4805144328777104,
432
+ "flow/mag_ratio_mean": 0.7592539517846826,
433
+ "flow/mag_ratio_std": 0.17078373482510378,
434
+ "step": 8192
435
+ },
436
+ {
437
+ "epoch": 0.390189829569073,
438
+ "grad_norm": 0.1544925421476364,
439
+ "learning_rate": 0.0006908696365085842,
440
+ "loss": 0.8480651378631592,
441
+ "step": 8448
442
+ },
443
+ {
444
+ "epoch": 0.4020137637984389,
445
+ "grad_norm": 0.6336784958839417,
446
+ "learning_rate": 0.0006731535118143318,
447
+ "loss": 0.8499810695648193,
448
+ "step": 8704
449
+ },
450
+ {
451
+ "epoch": 0.41383769802780473,
452
+ "grad_norm": 0.22286705672740936,
453
+ "learning_rate": 0.0006551867821290267,
454
+ "loss": 0.8474311232566833,
455
+ "step": 8960
456
+ },
457
+ {
458
+ "epoch": 0.4256616322571706,
459
+ "grad_norm": 0.2539626359939575,
460
+ "learning_rate": 0.0006369954506915572,
461
+ "loss": 0.8453187346458435,
462
+ "step": 9216
463
+ },
464
+ {
465
+ "epoch": 0.4256616322571706,
466
+ "eval_cos_loss": 0.23540513982919797,
467
+ "eval_loss": 0.8038723715091949,
468
+ "eval_mse_loss": 0.8038723715091949,
469
+ "flow/cos_sim": 0.7645948911638565,
470
+ "flow/improvement_ratio": 0.47903556278034976,
471
+ "flow/mag_ratio_mean": 0.7639505939396549,
472
+ "flow/mag_ratio_std": 0.17141265415301607,
473
+ "step": 9216
474
+ },
475
+ {
476
+ "epoch": 0.4256616322571706,
477
+ "eval_cos_loss": 0.23540513982919797,
478
+ "eval_loss": 0.8038723715091949,
479
+ "eval_mse_loss": 0.8038723715091949,
480
+ "eval_runtime": 38.6914,
481
+ "eval_samples_per_second": 723.493,
482
+ "eval_steps_per_second": 11.32,
483
+ "flow/cos_sim": 0.7645948911638565,
484
+ "flow/improvement_ratio": 0.47903556278034976,
485
+ "flow/mag_ratio_mean": 0.7639505939396549,
486
+ "flow/mag_ratio_std": 0.17141265415301607,
487
+ "step": 9216
488
+ },
489
+ {
490
+ "epoch": 0.4374855664865364,
491
+ "grad_norm": 2.060523748397827,
492
+ "learning_rate": 0.0006186058458068149,
493
+ "loss": 0.846603512763977,
494
+ "step": 9472
495
+ },
496
+ {
497
+ "epoch": 0.44930950071590225,
498
+ "grad_norm": 0.36982041597366333,
499
+ "learning_rate": 0.0006000445827407526,
500
+ "loss": 0.845413088798523,
501
+ "step": 9728
502
+ },
503
+ {
504
+ "epoch": 0.4611334349452681,
505
+ "grad_norm": 0.1367645412683487,
506
+ "learning_rate": 0.0005813385252001232,
507
+ "loss": 0.8436699509620667,
508
+ "step": 9984
509
+ },
510
+ {
511
+ "epoch": 0.472957369174634,
512
+ "grad_norm": 0.7542423605918884,
513
+ "learning_rate": 0.00056251474645265,
514
+ "loss": 0.8420372009277344,
515
+ "step": 10240
516
+ },
517
+ {
518
+ "epoch": 0.472957369174634,
519
+ "eval_cos_loss": 0.23470993282179856,
520
+ "eval_loss": 0.8019931023795855,
521
+ "eval_mse_loss": 0.8019931023795855,
522
+ "flow/cos_sim": 0.7652900771463298,
523
+ "flow/improvement_ratio": 0.4823292998010165,
524
+ "flow/mag_ratio_mean": 0.7609158404613738,
525
+ "flow/mag_ratio_std": 0.1730088302397837,
526
+ "step": 10240
527
+ },
528
+ {
529
+ "epoch": 0.472957369174634,
530
+ "eval_cos_loss": 0.23470993282179856,
531
+ "eval_loss": 0.8019931023795855,
532
+ "eval_mse_loss": 0.8019931023795855,
533
+ "eval_runtime": 38.4706,
534
+ "eval_samples_per_second": 727.647,
535
+ "eval_steps_per_second": 11.385,
536
+ "flow/cos_sim": 0.7652900771463298,
537
+ "flow/improvement_ratio": 0.4823292998010165,
538
+ "flow/mag_ratio_mean": 0.7609158404613738,
539
+ "flow/mag_ratio_std": 0.1730088302397837,
540
+ "step": 10240
541
+ },
542
+ {
543
+ "epoch": 0.48478130340399983,
544
+ "grad_norm": 0.14835897088050842,
545
+ "learning_rate": 0.0005436745145807214,
546
+ "loss": 0.8407944440841675,
547
+ "step": 10496
548
+ },
549
+ {
550
+ "epoch": 0.49660523763336567,
551
+ "grad_norm": 1.4828170537948608,
552
+ "learning_rate": 0.0005246973484120841,
553
+ "loss": 0.83956378698349,
554
+ "step": 10752
555
+ },
556
+ {
557
+ "epoch": 0.5084291718627315,
558
+ "grad_norm": 0.2746009826660156,
559
+ "learning_rate": 0.0005056844377834413,
560
+ "loss": 0.8389896154403687,
561
+ "step": 11008
562
+ },
563
+ {
564
+ "epoch": 0.5202531060920974,
565
+ "grad_norm": 0.28171437978744507,
566
+ "learning_rate": 0.0004866633000708374,
567
+ "loss": 0.8387643098831177,
568
+ "step": 11264
569
+ },
570
+ {
571
+ "epoch": 0.5202531060920974,
572
+ "eval_cos_loss": 0.2334500370572691,
573
+ "eval_loss": 0.7986243916973131,
574
+ "eval_mse_loss": 0.7986243916973131,
575
+ "flow/cos_sim": 0.7665499975420025,
576
+ "flow/improvement_ratio": 0.4849324461804133,
577
+ "flow/mag_ratio_mean": 0.7622648357800698,
578
+ "flow/mag_ratio_std": 0.1729429620381904,
579
+ "step": 11264
580
+ },
581
+ {
582
+ "epoch": 0.5202531060920974,
583
+ "eval_cos_loss": 0.2334500370572691,
584
+ "eval_loss": 0.7986243916973131,
585
+ "eval_mse_loss": 0.7986243916973131,
586
+ "eval_runtime": 38.5864,
587
+ "eval_samples_per_second": 725.464,
588
+ "eval_steps_per_second": 11.351,
589
+ "flow/cos_sim": 0.7665499975420025,
590
+ "flow/improvement_ratio": 0.4849324461804133,
591
+ "flow/mag_ratio_mean": 0.7622648357800698,
592
+ "flow/mag_ratio_std": 0.1729429620381904,
593
+ "step": 11264
594
+ },
595
+ {
596
+ "epoch": 0.5320770403214632,
597
+ "grad_norm": 0.3698814809322357,
598
+ "learning_rate": 0.00046766146455737116,
599
+ "loss": 0.8367224335670471,
600
+ "step": 11520
601
+ },
602
+ {
603
+ "epoch": 0.5439009745508291,
604
+ "grad_norm": 0.24135378003120422,
605
+ "learning_rate": 0.00044870643259007823,
606
+ "loss": 0.8399287462234497,
607
+ "step": 11776
608
+ },
609
+ {
610
+ "epoch": 0.5557249087801949,
611
+ "grad_norm": 0.8356995582580566,
612
+ "learning_rate": 0.000429825637777245,
613
+ "loss": 0.8389950394630432,
614
+ "step": 12032
615
+ },
616
+ {
617
+ "epoch": 0.5675488430095608,
618
+ "grad_norm": 0.4132654666900635,
619
+ "learning_rate": 0.00041104640628376166,
620
+ "loss": 0.835192084312439,
621
+ "step": 12288
622
+ },
623
+ {
624
+ "epoch": 0.5675488430095608,
625
+ "eval_cos_loss": 0.2332091824104797,
626
+ "eval_loss": 0.7972118620458803,
627
+ "eval_mse_loss": 0.7972118620458803,
628
+ "flow/cos_sim": 0.7667908230328668,
629
+ "flow/improvement_ratio": 0.48442385131365634,
630
+ "flow/mag_ratio_mean": 0.7578917393401333,
631
+ "flow/mag_ratio_std": 0.1718250041887096,
632
+ "step": 12288
633
+ },
634
+ {
635
+ "epoch": 0.5675488430095608,
636
+ "eval_cos_loss": 0.2332091824104797,
637
+ "eval_loss": 0.7972118620458803,
638
+ "eval_mse_loss": 0.7972118620458803,
639
+ "eval_runtime": 38.6674,
640
+ "eval_samples_per_second": 723.943,
641
+ "eval_steps_per_second": 11.327,
642
+ "flow/cos_sim": 0.7667908230328668,
643
+ "flow/improvement_ratio": 0.48442385131365634,
644
+ "flow/mag_ratio_mean": 0.7578917393401333,
645
+ "flow/mag_ratio_std": 0.1718250041887096,
646
+ "step": 12288
647
+ },
648
+ {
649
+ "epoch": 0.5793727772389267,
650
+ "grad_norm": 0.7286463379859924,
651
+ "learning_rate": 0.0003924684852772684,
652
+ "loss": 0.8363797664642334,
653
+ "step": 12544
654
+ },
655
+ {
656
+ "epoch": 0.5911967114682924,
657
+ "grad_norm": 0.34544652700424194,
658
+ "learning_rate": 0.00037397307103411104,
659
+ "loss": 0.8373527526855469,
660
+ "step": 12800
661
+ },
662
+ {
663
+ "epoch": 0.6030206456976583,
664
+ "grad_norm": 0.17035049200057983,
665
+ "learning_rate": 0.0003556600555022182,
666
+ "loss": 0.838341474533081,
667
+ "step": 13056
668
+ },
669
+ {
670
+ "epoch": 0.6148445799270241,
671
+ "grad_norm": 0.27102863788604736,
672
+ "learning_rate": 0.00033755594309981104,
673
+ "loss": 0.8337681889533997,
674
+ "step": 13312
675
+ },
676
+ {
677
+ "epoch": 0.6148445799270241,
678
+ "eval_cos_loss": 0.23153494631863075,
679
+ "eval_loss": 0.7926306961333915,
680
+ "eval_mse_loss": 0.7926306961333915,
681
+ "flow/cos_sim": 0.768465078856847,
682
+ "flow/improvement_ratio": 0.4786990506055693,
683
+ "flow/mag_ratio_mean": 0.766311283389183,
684
+ "flow/mag_ratio_std": 0.17224226364670278,
685
+ "step": 13312
686
+ },
687
+ {
688
+ "epoch": 0.6148445799270241,
689
+ "eval_cos_loss": 0.23153494631863075,
690
+ "eval_loss": 0.7926306961333915,
691
+ "eval_mse_loss": 0.7926306961333915,
692
+ "eval_runtime": 38.757,
693
+ "eval_samples_per_second": 722.269,
694
+ "eval_steps_per_second": 11.301,
695
+ "flow/cos_sim": 0.768465078856847,
696
+ "flow/improvement_ratio": 0.4786990506055693,
697
+ "flow/mag_ratio_mean": 0.766311283389183,
698
+ "flow/mag_ratio_std": 0.17224226364670278,
699
+ "step": 13312
700
+ },
701
+ {
702
+ "epoch": 0.62666851415639,
703
+ "grad_norm": 0.19754841923713684,
704
+ "learning_rate": 0.00031975624571791317,
705
+ "loss": 0.8308400511741638,
706
+ "step": 13568
707
+ },
708
+ {
709
+ "epoch": 0.6384924483857558,
710
+ "grad_norm": 0.12013949453830719,
711
+ "learning_rate": 0.00030214713628576984,
712
+ "loss": 0.8329613208770752,
713
+ "step": 13824
714
+ },
715
+ {
716
+ "epoch": 0.6503163826151217,
717
+ "grad_norm": 0.37349933385849,
718
+ "learning_rate": 0.00028482437920433144,
719
+ "loss": 0.8327583074569702,
720
+ "step": 14080
721
+ },
722
+ {
723
+ "epoch": 0.6621403168444876,
724
+ "grad_norm": 0.22871997952461243,
725
+ "learning_rate": 0.00026781304569125866,
726
+ "loss": 0.8310943245887756,
727
+ "step": 14336
728
+ },
729
+ {
730
+ "epoch": 0.6621403168444876,
731
+ "eval_cos_loss": 0.23119079505335793,
732
+ "eval_loss": 0.7905431563995745,
733
+ "eval_mse_loss": 0.7905431563995745,
734
+ "flow/cos_sim": 0.7688092066817087,
735
+ "flow/improvement_ratio": 0.48085032192539406,
736
+ "flow/mag_ratio_mean": 0.7671621094827783,
737
+ "flow/mag_ratio_std": 0.17147603141118403,
738
+ "step": 14336
739
+ },
740
+ {
741
+ "epoch": 0.6621403168444876,
742
+ "eval_cos_loss": 0.23119079505335793,
743
+ "eval_loss": 0.7905431563995745,
744
+ "eval_mse_loss": 0.7905431563995745,
745
+ "eval_runtime": 38.6687,
746
+ "eval_samples_per_second": 723.919,
747
+ "eval_steps_per_second": 11.327,
748
+ "flow/cos_sim": 0.7688092066817087,
749
+ "flow/improvement_ratio": 0.48085032192539406,
750
+ "flow/mag_ratio_mean": 0.7671621094827783,
751
+ "flow/mag_ratio_std": 0.17147603141118403,
752
+ "step": 14336
753
+ },
754
+ {
755
+ "epoch": 0.6739642510738534,
756
+ "grad_norm": 0.2850095331668854,
757
+ "learning_rate": 0.0002511377562410373,
758
+ "loss": 0.831950843334198,
759
+ "step": 14592
760
+ },
761
+ {
762
+ "epoch": 0.6857881853032193,
763
+ "grad_norm": 0.1518682837486267,
764
+ "learning_rate": 0.00023482264499174915,
765
+ "loss": 0.8332844376564026,
766
+ "step": 14848
767
+ },
768
+ {
769
+ "epoch": 0.6976121195325851,
770
+ "grad_norm": 0.27970460057258606,
771
+ "learning_rate": 0.00021889132479574586,
772
+ "loss": 0.8316826820373535,
773
+ "step": 15104
774
+ },
775
+ {
776
+ "epoch": 0.709436053761951,
777
+ "grad_norm": 0.2190929353237152,
778
+ "learning_rate": 0.0002033668530447801,
779
+ "loss": 0.8302494883537292,
780
+ "step": 15360
781
+ },
782
+ {
783
+ "epoch": 0.709436053761951,
784
+ "eval_cos_loss": 0.2316437272448518,
785
+ "eval_loss": 0.7923038827774187,
786
+ "eval_mse_loss": 0.7923038827774187,
787
+ "flow/cos_sim": 0.7683562772459092,
788
+ "flow/improvement_ratio": 0.48893266535240765,
789
+ "flow/mag_ratio_mean": 0.7614389026546042,
790
+ "flow/mag_ratio_std": 0.1729467008347925,
791
+ "step": 15360
792
+ },
793
+ {
794
+ "epoch": 0.709436053761951,
795
+ "eval_cos_loss": 0.2316437272448518,
796
+ "eval_loss": 0.7923038827774187,
797
+ "eval_mse_loss": 0.7923038827774187,
798
+ "eval_runtime": 39.0693,
799
+ "eval_samples_per_second": 716.497,
800
+ "eval_steps_per_second": 11.211,
801
+ "flow/cos_sim": 0.7683562772459092,
802
+ "flow/improvement_ratio": 0.48893266535240765,
803
+ "flow/mag_ratio_mean": 0.7614389026546042,
804
+ "flow/mag_ratio_std": 0.1729467008347925,
805
+ "step": 15360
806
+ },
807
+ {
808
+ "epoch": 0.7212599879913169,
809
+ "grad_norm": 0.529835045337677,
810
+ "learning_rate": 0.0001882716982990524,
811
+ "loss": 0.8327122330665588,
812
+ "step": 15616
813
+ },
814
+ {
815
+ "epoch": 0.7330839222206826,
816
+ "grad_norm": 0.25720837712287903,
817
+ "learning_rate": 0.00017362770776847765,
818
+ "loss": 0.8322436213493347,
819
+ "step": 15872
820
+ },
821
+ {
822
+ "epoch": 0.7449078564500485,
823
+ "grad_norm": 0.47202157974243164,
824
+ "learning_rate": 0.00015945607569322633,
825
+ "loss": 0.8282320499420166,
826
+ "step": 16128
827
+ },
828
+ {
829
+ "epoch": 0.7567317906794143,
830
+ "grad_norm": 0.3377237021923065,
831
+ "learning_rate": 0.00014577731266931444,
832
+ "loss": 0.8291305303573608,
833
+ "step": 16384
834
+ },
835
+ {
836
+ "epoch": 0.7567317906794143,
837
+ "eval_cos_loss": 0.2321365722986661,
838
+ "eval_loss": 0.7941420237495475,
839
+ "eval_mse_loss": 0.7941420237495475,
840
+ "flow/cos_sim": 0.7678634562448824,
841
+ "flow/improvement_ratio": 0.4825173763379659,
842
+ "flow/mag_ratio_mean": 0.7654842053918534,
843
+ "flow/mag_ratio_std": 0.17167809105490986,
844
+ "step": 16384
845
+ },
846
+ {
847
+ "epoch": 0.7567317906794143,
848
+ "eval_cos_loss": 0.2321365722986661,
849
+ "eval_loss": 0.7941420237495475,
850
+ "eval_mse_loss": 0.7941420237495475,
851
+ "eval_runtime": 38.5956,
852
+ "eval_samples_per_second": 725.29,
853
+ "eval_steps_per_second": 11.348,
854
+ "flow/cos_sim": 0.7678634562448824,
855
+ "flow/improvement_ratio": 0.4825173763379659,
856
+ "flow/mag_ratio_mean": 0.7654842053918534,
857
+ "flow/mag_ratio_std": 0.17167809105490986,
858
+ "step": 16384
859
+ },
860
+ {
861
+ "epoch": 0.7685557249087802,
862
+ "grad_norm": 0.3533153533935547,
863
+ "learning_rate": 0.00013261121596362468,
864
+ "loss": 0.8273910284042358,
865
+ "step": 16640
866
+ },
867
+ {
868
+ "epoch": 0.780379659138146,
869
+ "grad_norm": 0.14819054305553436,
870
+ "learning_rate": 0.00011997684086133532,
871
+ "loss": 0.8282073140144348,
872
+ "step": 16896
873
+ },
874
+ {
875
+ "epoch": 0.7922035933675119,
876
+ "grad_norm": 0.3358698785305023,
877
+ "learning_rate": 0.000107892473087213,
878
+ "loss": 0.8298404812812805,
879
+ "step": 17152
880
+ },
881
+ {
882
+ "epoch": 0.8040275275968778,
883
+ "grad_norm": 0.2873646914958954,
884
+ "learning_rate": 9.637560234069376e-05,
885
+ "loss": 0.8292073607444763,
886
+ "step": 17408
887
+ },
888
+ {
889
+ "epoch": 0.8040275275968778,
890
+ "eval_cos_loss": 0.2309515728662003,
891
+ "eval_loss": 0.790707334671935,
892
+ "eval_mse_loss": 0.790707334671935,
893
+ "flow/cos_sim": 0.7690484462535545,
894
+ "flow/improvement_ratio": 0.4813669728089685,
895
+ "flow/mag_ratio_mean": 0.7638134028269276,
896
+ "flow/mag_ratio_std": 0.17316651834200505,
897
+ "step": 17408
898
+ },
899
+ {
900
+ "epoch": 0.8040275275968778,
901
+ "eval_cos_loss": 0.2309515728662003,
902
+ "eval_loss": 0.790707334671935,
903
+ "eval_mse_loss": 0.790707334671935,
904
+ "eval_runtime": 38.6828,
905
+ "eval_samples_per_second": 723.655,
906
+ "eval_steps_per_second": 11.323,
907
+ "flow/cos_sim": 0.7690484462535545,
908
+ "flow/improvement_ratio": 0.4813669728089685,
909
+ "flow/mag_ratio_mean": 0.7638134028269276,
910
+ "flow/mag_ratio_std": 0.17316651834200505,
911
+ "step": 17408
912
+ },
913
+ {
914
+ "epoch": 0.8158514618262436,
915
+ "grad_norm": 0.09706299751996994,
916
+ "learning_rate": 8.544289698304941e-05,
917
+ "loss": 0.8297677040100098,
918
+ "step": 17664
919
+ },
920
+ {
921
+ "epoch": 0.8276753960556095,
922
+ "grad_norm": 0.41707268357276917,
923
+ "learning_rate": 7.511017991327612e-05,
924
+ "loss": 0.8265313506126404,
925
+ "step": 17920
926
+ },
927
+ {
928
+ "epoch": 0.8394993302849753,
929
+ "grad_norm": 0.20580926537513733,
930
+ "learning_rate": 6.539240566762194e-05,
931
+ "loss": 0.8272239565849304,
932
+ "step": 18176
933
+ },
934
+ {
935
+ "epoch": 0.8513232645143411,
936
+ "grad_norm": 0.174299955368042,
937
+ "learning_rate": 5.630363877589212e-05,
938
+ "loss": 0.8269455432891846,
939
+ "step": 18432
940
+ },
941
+ {
942
+ "epoch": 0.8513232645143411,
943
+ "eval_cos_loss": 0.2312889698229424,
944
+ "eval_loss": 0.7905569756956405,
945
+ "eval_mse_loss": 0.7905569756956405,
946
+ "flow/cos_sim": 0.7687110510318791,
947
+ "flow/improvement_ratio": 0.4794755227965851,
948
+ "flow/mag_ratio_mean": 0.7669512829551958,
949
+ "flow/mag_ratio_std": 0.17198007941654284,
950
+ "step": 18432
951
+ },
952
+ {
953
+ "epoch": 0.8513232645143411,
954
+ "eval_cos_loss": 0.2312889698229424,
955
+ "eval_loss": 0.7905569756956405,
956
+ "eval_mse_loss": 0.7905569756956405,
957
+ "eval_runtime": 38.7452,
958
+ "eval_samples_per_second": 722.49,
959
+ "eval_steps_per_second": 11.305,
960
+ "flow/cos_sim": 0.7687110510318791,
961
+ "flow/improvement_ratio": 0.4794755227965851,
962
+ "flow/mag_ratio_mean": 0.7669512829551958,
963
+ "flow/mag_ratio_std": 0.17198007941654284,
964
+ "step": 18432
965
+ },
966
+ {
967
+ "epoch": 0.8631471987437069,
968
+ "grad_norm": 0.21769863367080688,
969
+ "learning_rate": 4.7857033405864005e-05,
970
+ "loss": 0.8271985650062561,
971
+ "step": 18688
972
+ },
973
+ {
974
+ "epoch": 0.8749711329730728,
975
+ "grad_norm": 0.1875995397567749,
976
+ "learning_rate": 4.006481432526532e-05,
977
+ "loss": 0.8238598704338074,
978
+ "step": 18944
979
+ },
980
+ {
981
+ "epoch": 0.8867950672024387,
982
+ "grad_norm": 0.14308157563209534,
983
+ "learning_rate": 3.2938259208874446e-05,
984
+ "loss": 0.8277314901351929,
985
+ "step": 19200
986
+ },
987
+ {
988
+ "epoch": 0.8986190014318045,
989
+ "grad_norm": 0.23388312757015228,
990
+ "learning_rate": 2.648768231634713e-05,
991
+ "loss": 0.8289337158203125,
992
+ "step": 19456
993
+ },
994
+ {
995
+ "epoch": 0.8986190014318045,
996
+ "eval_cos_loss": 0.23167473305714184,
997
+ "eval_loss": 0.7916786500580235,
998
+ "eval_mse_loss": 0.7916786500580235,
999
+ "flow/cos_sim": 0.7683252998410839,
1000
+ "flow/improvement_ratio": 0.4856329362278115,
1001
+ "flow/mag_ratio_mean": 0.7651582228266485,
1002
+ "flow/mag_ratio_std": 0.17314033819251953,
1003
+ "step": 19456
1004
+ },
1005
+ {
1006
+ "epoch": 0.8986190014318045,
1007
+ "eval_cos_loss": 0.23167473305714184,
1008
+ "eval_loss": 0.7916786500580235,
1009
+ "eval_mse_loss": 0.7916786500580235,
1010
+ "eval_runtime": 37.8543,
1011
+ "eval_samples_per_second": 739.492,
1012
+ "eval_steps_per_second": 11.571,
1013
+ "flow/cos_sim": 0.7683252998410839,
1014
+ "flow/improvement_ratio": 0.4856329362278115,
1015
+ "flow/mag_ratio_mean": 0.7651582228266485,
1016
+ "flow/mag_ratio_std": 0.17314033819251953,
1017
+ "step": 19456
1018
+ },
1019
+ {
1020
+ "epoch": 0.9104429356611704,
1021
+ "grad_norm": 0.1731036752462387,
1022
+ "learning_rate": 2.074359572060447e-05,
1023
+ "loss": 0.8261044025421143,
1024
+ "step": 19712
1025
+ },
1026
+ {
1027
+ "epoch": 0.9222668698905362,
1028
+ "grad_norm": 0.1453009396791458,
1029
+ "learning_rate": 1.5669266631345104e-05,
1030
+ "loss": 0.825700044631958,
1031
+ "step": 19968
1032
+ },
1033
+ {
1034
+ "epoch": 0.9340908041199021,
1035
+ "grad_norm": 0.15222381055355072,
1036
+ "learning_rate": 1.1295909170270401e-05,
1037
+ "loss": 0.8293823003768921,
1038
+ "step": 20224
1039
+ },
1040
+ {
1041
+ "epoch": 0.945914738349268,
1042
+ "grad_norm": 0.33283916115760803,
1043
+ "learning_rate": 7.629852895446598e-06,
1044
+ "loss": 0.8306663632392883,
1045
+ "step": 20480
1046
+ },
1047
+ {
1048
+ "epoch": 0.945914738349268,
1049
+ "eval_cos_loss": 0.2297409219012413,
1050
+ "eval_loss": 0.7868784807588411,
1051
+ "eval_mse_loss": 0.7868784807588411,
1052
+ "flow/cos_sim": 0.770259097422639,
1053
+ "flow/improvement_ratio": 0.4786264787525891,
1054
+ "flow/mag_ratio_mean": 0.7675387896087071,
1055
+ "flow/mag_ratio_std": 0.1731719597833886,
1056
+ "step": 20480
1057
+ },
1058
+ {
1059
+ "epoch": 0.945914738349268,
1060
+ "eval_cos_loss": 0.2297409219012413,
1061
+ "eval_loss": 0.7868784807588411,
1062
+ "eval_mse_loss": 0.7868784807588411,
1063
+ "eval_runtime": 38.1354,
1064
+ "eval_samples_per_second": 734.043,
1065
+ "eval_steps_per_second": 11.485,
1066
+ "flow/cos_sim": 0.770259097422639,
1067
+ "flow/improvement_ratio": 0.4786264787525891,
1068
+ "flow/mag_ratio_mean": 0.7675387896087071,
1069
+ "flow/mag_ratio_std": 0.1731719597833886,
1070
+ "step": 20480
1071
+ },
1072
+ {
1073
+ "epoch": 0.9577386725786338,
1074
+ "grad_norm": 0.12771452963352203,
1075
+ "learning_rate": 4.676403688286757e-06,
1076
+ "loss": 0.826269268989563,
1077
+ "step": 20736
1078
+ },
1079
+ {
1080
+ "epoch": 0.9695626068079997,
1081
+ "grad_norm": 0.18342143297195435,
1082
+ "learning_rate": 2.4398360743514426e-06,
1083
+ "loss": 0.8270809054374695,
1084
+ "step": 20992
1085
+ },
1086
+ {
1087
+ "epoch": 0.9813865410373654,
1088
+ "grad_norm": 0.17212283611297607,
1089
+ "learning_rate": 9.233870368297592e-07,
1090
+ "loss": 0.8276194930076599,
1091
+ "step": 21248
1092
+ },
1093
+ {
1094
+ "epoch": 0.9932104752667313,
1095
+ "grad_norm": 0.3044182062149048,
1096
+ "learning_rate": 1.2925133165525572e-07,
1097
+ "loss": 0.8276101350784302,
1098
+ "step": 21504
1099
+ },
1100
+ {
1101
+ "epoch": 0.9932104752667313,
1102
+ "eval_cos_loss": 0.22986769162492665,
1103
+ "eval_loss": 0.7868886709757591,
1104
+ "eval_mse_loss": 0.7868886709757591,
1105
+ "flow/cos_sim": 0.7701323260999706,
1106
+ "flow/improvement_ratio": 0.4870183017428063,
1107
+ "flow/mag_ratio_mean": 0.7669129322652948,
1108
+ "flow/mag_ratio_std": 0.17209734478497615,
1109
+ "step": 21504
1110
+ },
1111
+ {
1112
+ "epoch": 0.9932104752667313,
1113
+ "eval_cos_loss": 0.22986769162492665,
1114
+ "eval_loss": 0.7868886709757591,
1115
+ "eval_mse_loss": 0.7868886709757591,
1116
+ "eval_runtime": 38.0306,
1117
+ "eval_samples_per_second": 736.065,
1118
+ "eval_steps_per_second": 11.517,
1119
+ "flow/cos_sim": 0.7701323260999706,
1120
+ "flow/improvement_ratio": 0.4870183017428063,
1121
+ "flow/mag_ratio_mean": 0.7669129322652948,
1122
+ "flow/mag_ratio_std": 0.17209734478497615,
1123
+ "step": 21504
1124
+ }
1125
+ ],
1126
+ "logging_steps": 256,
1127
+ "max_steps": 21651,
1128
+ "num_input_tokens_seen": 0,
1129
+ "num_train_epochs": 1,
1130
+ "save_steps": 1024,
1131
+ "stateful_callbacks": {
1132
+ "TrainerControl": {
1133
+ "args": {
1134
+ "should_epoch_stop": false,
1135
+ "should_evaluate": false,
1136
+ "should_log": false,
1137
+ "should_save": true,
1138
+ "should_training_stop": true
1139
+ },
1140
+ "attributes": {}
1141
+ }
1142
+ },
1143
+ "total_flos": 0.0,
1144
+ "train_batch_size": 64,
1145
+ "trial_name": null,
1146
+ "trial_params": null
1147
+ }
checkpoints-v2.5-new/checkpoint-21651/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03a8c5803c6e5663005c9dc7a4c98c213c8267649ccec42581475db390cc8017
3
+ size 5137