Mueris commited on
Commit
8c249a4
·
verified ·
1 Parent(s): 9a737e7

Upload 4 files

Browse files
config.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BlipForConditionalGeneration"
4
+ ],
5
+ "image_text_hidden_size": 256,
6
+ "initializer_factor": 1.0,
7
+ "initializer_range": 0.02,
8
+ "label_smoothing": 0.0,
9
+ "logit_scale_init_value": 2.6592,
10
+ "model_type": "blip",
11
+ "projection_dim": 512,
12
+ "text_config": {
13
+ "_attn_implementation_autoset": true,
14
+ "attention_probs_dropout_prob": 0.0,
15
+ "encoder_hidden_size": 768,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 768,
19
+ "initializer_factor": 1.0,
20
+ "initializer_range": 0.02,
21
+ "intermediate_size": 3072,
22
+ "label_smoothing": 0.0,
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 512,
25
+ "model_type": "blip_text_model",
26
+ "num_attention_heads": 12,
27
+ "num_hidden_layers": 12,
28
+ "projection_dim": 768,
29
+ "torch_dtype": "float32",
30
+ "use_cache": true,
31
+ "vocab_size": 30524
32
+ },
33
+ "torch_dtype": "float32",
34
+ "transformers_version": "4.51.3",
35
+ "vision_config": {
36
+ "_attn_implementation_autoset": true,
37
+ "attention_dropout": 0.0,
38
+ "dropout": 0.0,
39
+ "hidden_act": "gelu",
40
+ "hidden_size": 768,
41
+ "image_size": 384,
42
+ "initializer_factor": 1.0,
43
+ "initializer_range": 0.02,
44
+ "intermediate_size": 3072,
45
+ "layer_norm_eps": 1e-05,
46
+ "model_type": "blip_vision_model",
47
+ "num_attention_heads": 12,
48
+ "num_channels": 3,
49
+ "num_hidden_layers": 12,
50
+ "patch_size": 16,
51
+ "projection_dim": 512,
52
+ "torch_dtype": "float32"
53
+ }
54
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 30522,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "4.51.3"
7
+ }
pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:472c0afe2b754d96f0968da203ce6b1a2f7c6627739a8e76f358090fadcc8556
3
+ size 989717056
trainer_state.json ADDED
@@ -0,0 +1,902 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 6.0,
6
+ "eval_steps": 500,
7
+ "global_step": 12498,
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.04800768122899664,
14
+ "grad_norm": 3.5230019092559814,
15
+ "learning_rate": 4.995343254920787e-05,
16
+ "loss": 3.834,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 0.09601536245799328,
21
+ "grad_norm": 2.2122299671173096,
22
+ "learning_rate": 4.990542486797888e-05,
23
+ "loss": 1.006,
24
+ "step": 200
25
+ },
26
+ {
27
+ "epoch": 0.14402304368698993,
28
+ "grad_norm": 1.6999797821044922,
29
+ "learning_rate": 4.985741718674988e-05,
30
+ "loss": 0.795,
31
+ "step": 300
32
+ },
33
+ {
34
+ "epoch": 0.19203072491598655,
35
+ "grad_norm": 1.8523478507995605,
36
+ "learning_rate": 4.980940950552089e-05,
37
+ "loss": 0.7485,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 0.2400384061449832,
42
+ "grad_norm": 1.6911518573760986,
43
+ "learning_rate": 4.9761401824291885e-05,
44
+ "loss": 0.7051,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.28804608737397985,
49
+ "grad_norm": 1.78285551071167,
50
+ "learning_rate": 4.971339414306289e-05,
51
+ "loss": 0.6518,
52
+ "step": 600
53
+ },
54
+ {
55
+ "epoch": 0.3360537686029765,
56
+ "grad_norm": 1.8025232553482056,
57
+ "learning_rate": 4.9665386461833895e-05,
58
+ "loss": 0.6429,
59
+ "step": 700
60
+ },
61
+ {
62
+ "epoch": 0.3840614498319731,
63
+ "grad_norm": 1.6609948873519897,
64
+ "learning_rate": 4.96173787806049e-05,
65
+ "loss": 0.6148,
66
+ "step": 800
67
+ },
68
+ {
69
+ "epoch": 0.4320691310609698,
70
+ "grad_norm": 1.7044763565063477,
71
+ "learning_rate": 4.9569371099375905e-05,
72
+ "loss": 0.6052,
73
+ "step": 900
74
+ },
75
+ {
76
+ "epoch": 0.4800768122899664,
77
+ "grad_norm": 1.742863655090332,
78
+ "learning_rate": 4.9521363418146903e-05,
79
+ "loss": 0.599,
80
+ "step": 1000
81
+ },
82
+ {
83
+ "epoch": 0.528084493518963,
84
+ "grad_norm": 1.6556651592254639,
85
+ "learning_rate": 4.947335573691791e-05,
86
+ "loss": 0.5772,
87
+ "step": 1100
88
+ },
89
+ {
90
+ "epoch": 0.5760921747479597,
91
+ "grad_norm": 1.578718662261963,
92
+ "learning_rate": 4.9425348055688914e-05,
93
+ "loss": 0.5617,
94
+ "step": 1200
95
+ },
96
+ {
97
+ "epoch": 0.6240998559769563,
98
+ "grad_norm": 1.6328682899475098,
99
+ "learning_rate": 4.937734037445992e-05,
100
+ "loss": 0.5566,
101
+ "step": 1300
102
+ },
103
+ {
104
+ "epoch": 0.672107537205953,
105
+ "grad_norm": 1.775172233581543,
106
+ "learning_rate": 4.932981277004321e-05,
107
+ "loss": 0.5474,
108
+ "step": 1400
109
+ },
110
+ {
111
+ "epoch": 0.7201152184349496,
112
+ "grad_norm": 1.4435912370681763,
113
+ "learning_rate": 4.928180508881422e-05,
114
+ "loss": 0.5342,
115
+ "step": 1500
116
+ },
117
+ {
118
+ "epoch": 0.7681228996639462,
119
+ "grad_norm": 1.2456103563308716,
120
+ "learning_rate": 4.9233797407585215e-05,
121
+ "loss": 0.5237,
122
+ "step": 1600
123
+ },
124
+ {
125
+ "epoch": 0.8161305808929429,
126
+ "grad_norm": 1.5610588788986206,
127
+ "learning_rate": 4.918578972635622e-05,
128
+ "loss": 0.5335,
129
+ "step": 1700
130
+ },
131
+ {
132
+ "epoch": 0.8641382621219396,
133
+ "grad_norm": 1.677467703819275,
134
+ "learning_rate": 4.9137782045127225e-05,
135
+ "loss": 0.5052,
136
+ "step": 1800
137
+ },
138
+ {
139
+ "epoch": 0.9121459433509361,
140
+ "grad_norm": 1.3189855813980103,
141
+ "learning_rate": 4.908977436389823e-05,
142
+ "loss": 0.4929,
143
+ "step": 1900
144
+ },
145
+ {
146
+ "epoch": 0.9601536245799328,
147
+ "grad_norm": 1.320538878440857,
148
+ "learning_rate": 4.904176668266923e-05,
149
+ "loss": 0.4973,
150
+ "step": 2000
151
+ },
152
+ {
153
+ "epoch": 1.0081613058089294,
154
+ "grad_norm": 1.2392332553863525,
155
+ "learning_rate": 4.8993759001440233e-05,
156
+ "loss": 0.4879,
157
+ "step": 2100
158
+ },
159
+ {
160
+ "epoch": 1.056168987037926,
161
+ "grad_norm": 1.2972139120101929,
162
+ "learning_rate": 4.894575132021124e-05,
163
+ "loss": 0.4339,
164
+ "step": 2200
165
+ },
166
+ {
167
+ "epoch": 1.1041766682669227,
168
+ "grad_norm": 1.763129711151123,
169
+ "learning_rate": 4.8897743638982243e-05,
170
+ "loss": 0.4215,
171
+ "step": 2300
172
+ },
173
+ {
174
+ "epoch": 1.1521843494959194,
175
+ "grad_norm": 1.8901598453521729,
176
+ "learning_rate": 4.884973595775325e-05,
177
+ "loss": 0.4386,
178
+ "step": 2400
179
+ },
180
+ {
181
+ "epoch": 1.2001920307249159,
182
+ "grad_norm": 1.1455730199813843,
183
+ "learning_rate": 4.880172827652425e-05,
184
+ "loss": 0.4414,
185
+ "step": 2500
186
+ },
187
+ {
188
+ "epoch": 1.2481997119539125,
189
+ "grad_norm": 1.3231408596038818,
190
+ "learning_rate": 4.875372059529525e-05,
191
+ "loss": 0.4232,
192
+ "step": 2600
193
+ },
194
+ {
195
+ "epoch": 1.2962073931829092,
196
+ "grad_norm": 1.544893503189087,
197
+ "learning_rate": 4.870571291406626e-05,
198
+ "loss": 0.4338,
199
+ "step": 2700
200
+ },
201
+ {
202
+ "epoch": 1.344215074411906,
203
+ "grad_norm": 1.4648122787475586,
204
+ "learning_rate": 4.8657705232837255e-05,
205
+ "loss": 0.4273,
206
+ "step": 2800
207
+ },
208
+ {
209
+ "epoch": 1.3922227556409026,
210
+ "grad_norm": 1.405002474784851,
211
+ "learning_rate": 4.860969755160826e-05,
212
+ "loss": 0.4337,
213
+ "step": 2900
214
+ },
215
+ {
216
+ "epoch": 1.4402304368698993,
217
+ "grad_norm": 1.2591880559921265,
218
+ "learning_rate": 4.8561689870379265e-05,
219
+ "loss": 0.4234,
220
+ "step": 3000
221
+ },
222
+ {
223
+ "epoch": 1.488238118098896,
224
+ "grad_norm": 1.422620415687561,
225
+ "learning_rate": 4.8513682189150264e-05,
226
+ "loss": 0.4239,
227
+ "step": 3100
228
+ },
229
+ {
230
+ "epoch": 1.5362457993278924,
231
+ "grad_norm": 1.585553526878357,
232
+ "learning_rate": 4.846567450792127e-05,
233
+ "loss": 0.4254,
234
+ "step": 3200
235
+ },
236
+ {
237
+ "epoch": 1.584253480556889,
238
+ "grad_norm": 1.1293108463287354,
239
+ "learning_rate": 4.8417666826692274e-05,
240
+ "loss": 0.422,
241
+ "step": 3300
242
+ },
243
+ {
244
+ "epoch": 1.6322611617858858,
245
+ "grad_norm": 1.265154242515564,
246
+ "learning_rate": 4.836965914546328e-05,
247
+ "loss": 0.4194,
248
+ "step": 3400
249
+ },
250
+ {
251
+ "epoch": 1.6802688430148822,
252
+ "grad_norm": 1.381218433380127,
253
+ "learning_rate": 4.832165146423428e-05,
254
+ "loss": 0.4321,
255
+ "step": 3500
256
+ },
257
+ {
258
+ "epoch": 1.7282765242438791,
259
+ "grad_norm": 1.3752981424331665,
260
+ "learning_rate": 4.827364378300528e-05,
261
+ "loss": 0.4151,
262
+ "step": 3600
263
+ },
264
+ {
265
+ "epoch": 1.7762842054728756,
266
+ "grad_norm": 1.1908366680145264,
267
+ "learning_rate": 4.822563610177629e-05,
268
+ "loss": 0.4203,
269
+ "step": 3700
270
+ },
271
+ {
272
+ "epoch": 1.8242918867018723,
273
+ "grad_norm": 1.2795414924621582,
274
+ "learning_rate": 4.8177628420547285e-05,
275
+ "loss": 0.414,
276
+ "step": 3800
277
+ },
278
+ {
279
+ "epoch": 1.872299567930869,
280
+ "grad_norm": 1.2664345502853394,
281
+ "learning_rate": 4.812962073931829e-05,
282
+ "loss": 0.4102,
283
+ "step": 3900
284
+ },
285
+ {
286
+ "epoch": 1.9203072491598656,
287
+ "grad_norm": 1.1444271802902222,
288
+ "learning_rate": 4.8081613058089295e-05,
289
+ "loss": 0.4066,
290
+ "step": 4000
291
+ },
292
+ {
293
+ "epoch": 1.9683149303888623,
294
+ "grad_norm": 1.233847737312317,
295
+ "learning_rate": 4.80336053768603e-05,
296
+ "loss": 0.4235,
297
+ "step": 4100
298
+ },
299
+ {
300
+ "epoch": 2.0163226116178588,
301
+ "grad_norm": 1.1085718870162964,
302
+ "learning_rate": 4.79855976956313e-05,
303
+ "loss": 0.3852,
304
+ "step": 4200
305
+ },
306
+ {
307
+ "epoch": 2.0643302928468557,
308
+ "grad_norm": 1.2250890731811523,
309
+ "learning_rate": 4.7937590014402304e-05,
310
+ "loss": 0.334,
311
+ "step": 4300
312
+ },
313
+ {
314
+ "epoch": 2.112337974075852,
315
+ "grad_norm": 1.3685556650161743,
316
+ "learning_rate": 4.788958233317331e-05,
317
+ "loss": 0.3273,
318
+ "step": 4400
319
+ },
320
+ {
321
+ "epoch": 2.160345655304849,
322
+ "grad_norm": 1.2295827865600586,
323
+ "learning_rate": 4.7841574651944314e-05,
324
+ "loss": 0.3332,
325
+ "step": 4500
326
+ },
327
+ {
328
+ "epoch": 2.2083533365338455,
329
+ "grad_norm": 1.3651865720748901,
330
+ "learning_rate": 4.779356697071531e-05,
331
+ "loss": 0.3355,
332
+ "step": 4600
333
+ },
334
+ {
335
+ "epoch": 2.256361017762842,
336
+ "grad_norm": 1.414611577987671,
337
+ "learning_rate": 4.774555928948632e-05,
338
+ "loss": 0.3347,
339
+ "step": 4700
340
+ },
341
+ {
342
+ "epoch": 2.304368698991839,
343
+ "grad_norm": 1.1170624494552612,
344
+ "learning_rate": 4.769755160825732e-05,
345
+ "loss": 0.3363,
346
+ "step": 4800
347
+ },
348
+ {
349
+ "epoch": 2.3523763802208353,
350
+ "grad_norm": 1.4784187078475952,
351
+ "learning_rate": 4.764954392702833e-05,
352
+ "loss": 0.3497,
353
+ "step": 4900
354
+ },
355
+ {
356
+ "epoch": 2.4003840614498317,
357
+ "grad_norm": 1.3222328424453735,
358
+ "learning_rate": 4.7601536245799325e-05,
359
+ "loss": 0.3319,
360
+ "step": 5000
361
+ },
362
+ {
363
+ "epoch": 2.4483917426788286,
364
+ "grad_norm": 1.4248158931732178,
365
+ "learning_rate": 4.755352856457033e-05,
366
+ "loss": 0.3456,
367
+ "step": 5100
368
+ },
369
+ {
370
+ "epoch": 2.496399423907825,
371
+ "grad_norm": 1.3521333932876587,
372
+ "learning_rate": 4.7505520883341336e-05,
373
+ "loss": 0.346,
374
+ "step": 5200
375
+ },
376
+ {
377
+ "epoch": 2.544407105136822,
378
+ "grad_norm": 1.3070430755615234,
379
+ "learning_rate": 4.745751320211234e-05,
380
+ "loss": 0.338,
381
+ "step": 5300
382
+ },
383
+ {
384
+ "epoch": 2.5924147863658185,
385
+ "grad_norm": 1.2667192220687866,
386
+ "learning_rate": 4.7409505520883346e-05,
387
+ "loss": 0.3508,
388
+ "step": 5400
389
+ },
390
+ {
391
+ "epoch": 2.6404224675948154,
392
+ "grad_norm": 1.404773473739624,
393
+ "learning_rate": 4.7361497839654344e-05,
394
+ "loss": 0.3376,
395
+ "step": 5500
396
+ },
397
+ {
398
+ "epoch": 2.688430148823812,
399
+ "grad_norm": 1.233578085899353,
400
+ "learning_rate": 4.731349015842535e-05,
401
+ "loss": 0.3432,
402
+ "step": 5600
403
+ },
404
+ {
405
+ "epoch": 2.7364378300528083,
406
+ "grad_norm": 1.2811566591262817,
407
+ "learning_rate": 4.7265482477196354e-05,
408
+ "loss": 0.3498,
409
+ "step": 5700
410
+ },
411
+ {
412
+ "epoch": 2.784445511281805,
413
+ "grad_norm": 1.1714744567871094,
414
+ "learning_rate": 4.721747479596736e-05,
415
+ "loss": 0.3451,
416
+ "step": 5800
417
+ },
418
+ {
419
+ "epoch": 2.8324531925108016,
420
+ "grad_norm": 1.1416313648223877,
421
+ "learning_rate": 4.716946711473836e-05,
422
+ "loss": 0.3427,
423
+ "step": 5900
424
+ },
425
+ {
426
+ "epoch": 2.8804608737397985,
427
+ "grad_norm": 1.3504301309585571,
428
+ "learning_rate": 4.712145943350936e-05,
429
+ "loss": 0.3477,
430
+ "step": 6000
431
+ },
432
+ {
433
+ "epoch": 2.928468554968795,
434
+ "grad_norm": 1.4579728841781616,
435
+ "learning_rate": 4.707345175228037e-05,
436
+ "loss": 0.3473,
437
+ "step": 6100
438
+ },
439
+ {
440
+ "epoch": 2.976476236197792,
441
+ "grad_norm": 1.1714892387390137,
442
+ "learning_rate": 4.702544407105137e-05,
443
+ "loss": 0.341,
444
+ "step": 6200
445
+ },
446
+ {
447
+ "epoch": 3.0244839174267883,
448
+ "grad_norm": 1.200076699256897,
449
+ "learning_rate": 4.697743638982237e-05,
450
+ "loss": 0.2989,
451
+ "step": 6300
452
+ },
453
+ {
454
+ "epoch": 3.072491598655785,
455
+ "grad_norm": 1.3686045408248901,
456
+ "learning_rate": 4.6929428708593376e-05,
457
+ "loss": 0.2606,
458
+ "step": 6400
459
+ },
460
+ {
461
+ "epoch": 3.1204992798847817,
462
+ "grad_norm": 1.2546569108963013,
463
+ "learning_rate": 4.688142102736438e-05,
464
+ "loss": 0.2635,
465
+ "step": 6500
466
+ },
467
+ {
468
+ "epoch": 3.168506961113778,
469
+ "grad_norm": 1.3534624576568604,
470
+ "learning_rate": 4.6833413346135386e-05,
471
+ "loss": 0.2606,
472
+ "step": 6600
473
+ },
474
+ {
475
+ "epoch": 3.2165146423427746,
476
+ "grad_norm": 1.4877707958221436,
477
+ "learning_rate": 4.678540566490639e-05,
478
+ "loss": 0.2685,
479
+ "step": 6700
480
+ },
481
+ {
482
+ "epoch": 3.2645223235717715,
483
+ "grad_norm": 1.4656774997711182,
484
+ "learning_rate": 4.673739798367739e-05,
485
+ "loss": 0.272,
486
+ "step": 6800
487
+ },
488
+ {
489
+ "epoch": 3.312530004800768,
490
+ "grad_norm": 1.24815034866333,
491
+ "learning_rate": 4.6689390302448394e-05,
492
+ "loss": 0.2703,
493
+ "step": 6900
494
+ },
495
+ {
496
+ "epoch": 3.360537686029765,
497
+ "grad_norm": 1.1822447776794434,
498
+ "learning_rate": 4.66413826212194e-05,
499
+ "loss": 0.2702,
500
+ "step": 7000
501
+ },
502
+ {
503
+ "epoch": 3.4085453672587613,
504
+ "grad_norm": 1.4170218706130981,
505
+ "learning_rate": 4.6593374939990404e-05,
506
+ "loss": 0.2716,
507
+ "step": 7100
508
+ },
509
+ {
510
+ "epoch": 3.4565530484877582,
511
+ "grad_norm": 1.360338568687439,
512
+ "learning_rate": 4.65453672587614e-05,
513
+ "loss": 0.2717,
514
+ "step": 7200
515
+ },
516
+ {
517
+ "epoch": 3.5045607297167547,
518
+ "grad_norm": 1.4744322299957275,
519
+ "learning_rate": 4.649735957753241e-05,
520
+ "loss": 0.2795,
521
+ "step": 7300
522
+ },
523
+ {
524
+ "epoch": 3.552568410945751,
525
+ "grad_norm": 1.5080569982528687,
526
+ "learning_rate": 4.644935189630341e-05,
527
+ "loss": 0.2753,
528
+ "step": 7400
529
+ },
530
+ {
531
+ "epoch": 3.600576092174748,
532
+ "grad_norm": 1.5707645416259766,
533
+ "learning_rate": 4.640134421507442e-05,
534
+ "loss": 0.278,
535
+ "step": 7500
536
+ },
537
+ {
538
+ "epoch": 3.6485837734037445,
539
+ "grad_norm": 1.4137930870056152,
540
+ "learning_rate": 4.6353336533845416e-05,
541
+ "loss": 0.2756,
542
+ "step": 7600
543
+ },
544
+ {
545
+ "epoch": 3.696591454632741,
546
+ "grad_norm": 1.4156709909439087,
547
+ "learning_rate": 4.630532885261642e-05,
548
+ "loss": 0.2841,
549
+ "step": 7700
550
+ },
551
+ {
552
+ "epoch": 3.744599135861738,
553
+ "grad_norm": 1.4720118045806885,
554
+ "learning_rate": 4.6257321171387426e-05,
555
+ "loss": 0.2767,
556
+ "step": 7800
557
+ },
558
+ {
559
+ "epoch": 3.7926068170907348,
560
+ "grad_norm": 1.2695891857147217,
561
+ "learning_rate": 4.620931349015843e-05,
562
+ "loss": 0.2762,
563
+ "step": 7900
564
+ },
565
+ {
566
+ "epoch": 3.8406144983197312,
567
+ "grad_norm": 1.439794659614563,
568
+ "learning_rate": 4.6161305808929436e-05,
569
+ "loss": 0.2799,
570
+ "step": 8000
571
+ },
572
+ {
573
+ "epoch": 3.8886221795487277,
574
+ "grad_norm": 1.3036625385284424,
575
+ "learning_rate": 4.611377820451272e-05,
576
+ "loss": 0.2728,
577
+ "step": 8100
578
+ },
579
+ {
580
+ "epoch": 3.9366298607777246,
581
+ "grad_norm": 1.3942623138427734,
582
+ "learning_rate": 4.606577052328373e-05,
583
+ "loss": 0.2779,
584
+ "step": 8200
585
+ },
586
+ {
587
+ "epoch": 3.984637542006721,
588
+ "grad_norm": 1.3402965068817139,
589
+ "learning_rate": 4.6017762842054726e-05,
590
+ "loss": 0.2759,
591
+ "step": 8300
592
+ },
593
+ {
594
+ "epoch": 4.0326452232357175,
595
+ "grad_norm": 1.2347018718719482,
596
+ "learning_rate": 4.596975516082573e-05,
597
+ "loss": 0.2239,
598
+ "step": 8400
599
+ },
600
+ {
601
+ "epoch": 4.080652904464714,
602
+ "grad_norm": 1.2590075731277466,
603
+ "learning_rate": 4.5922227556409024e-05,
604
+ "loss": 0.1964,
605
+ "step": 8500
606
+ },
607
+ {
608
+ "epoch": 4.128660585693711,
609
+ "grad_norm": 1.2734559774398804,
610
+ "learning_rate": 4.587421987518003e-05,
611
+ "loss": 0.2033,
612
+ "step": 8600
613
+ },
614
+ {
615
+ "epoch": 4.176668266922707,
616
+ "grad_norm": 1.3685240745544434,
617
+ "learning_rate": 4.5826212193951034e-05,
618
+ "loss": 0.2039,
619
+ "step": 8700
620
+ },
621
+ {
622
+ "epoch": 4.224675948151704,
623
+ "grad_norm": 1.2431707382202148,
624
+ "learning_rate": 4.577820451272204e-05,
625
+ "loss": 0.2031,
626
+ "step": 8800
627
+ },
628
+ {
629
+ "epoch": 4.272683629380701,
630
+ "grad_norm": 1.3057847023010254,
631
+ "learning_rate": 4.573019683149304e-05,
632
+ "loss": 0.1993,
633
+ "step": 8900
634
+ },
635
+ {
636
+ "epoch": 4.320691310609698,
637
+ "grad_norm": 1.3426510095596313,
638
+ "learning_rate": 4.568218915026404e-05,
639
+ "loss": 0.2103,
640
+ "step": 9000
641
+ },
642
+ {
643
+ "epoch": 4.368698991838694,
644
+ "grad_norm": 1.4993879795074463,
645
+ "learning_rate": 4.563418146903505e-05,
646
+ "loss": 0.2038,
647
+ "step": 9100
648
+ },
649
+ {
650
+ "epoch": 4.416706673067691,
651
+ "grad_norm": 1.1477452516555786,
652
+ "learning_rate": 4.558617378780605e-05,
653
+ "loss": 0.2087,
654
+ "step": 9200
655
+ },
656
+ {
657
+ "epoch": 4.464714354296688,
658
+ "grad_norm": 1.2744262218475342,
659
+ "learning_rate": 4.553816610657706e-05,
660
+ "loss": 0.2128,
661
+ "step": 9300
662
+ },
663
+ {
664
+ "epoch": 4.512722035525684,
665
+ "grad_norm": 1.2372483015060425,
666
+ "learning_rate": 4.5490158425348056e-05,
667
+ "loss": 0.2141,
668
+ "step": 9400
669
+ },
670
+ {
671
+ "epoch": 4.560729716754681,
672
+ "grad_norm": 1.5185002088546753,
673
+ "learning_rate": 4.544215074411906e-05,
674
+ "loss": 0.2163,
675
+ "step": 9500
676
+ },
677
+ {
678
+ "epoch": 4.608737397983678,
679
+ "grad_norm": 1.2318006753921509,
680
+ "learning_rate": 4.5394143062890066e-05,
681
+ "loss": 0.2142,
682
+ "step": 9600
683
+ },
684
+ {
685
+ "epoch": 4.656745079212674,
686
+ "grad_norm": 1.0812604427337646,
687
+ "learning_rate": 4.534613538166107e-05,
688
+ "loss": 0.2179,
689
+ "step": 9700
690
+ },
691
+ {
692
+ "epoch": 4.704752760441671,
693
+ "grad_norm": 1.460476279258728,
694
+ "learning_rate": 4.529812770043207e-05,
695
+ "loss": 0.2203,
696
+ "step": 9800
697
+ },
698
+ {
699
+ "epoch": 4.7527604416706675,
700
+ "grad_norm": 1.4925509691238403,
701
+ "learning_rate": 4.5250120019203074e-05,
702
+ "loss": 0.2167,
703
+ "step": 9900
704
+ },
705
+ {
706
+ "epoch": 4.8007681228996635,
707
+ "grad_norm": 1.4219658374786377,
708
+ "learning_rate": 4.520211233797408e-05,
709
+ "loss": 0.2163,
710
+ "step": 10000
711
+ },
712
+ {
713
+ "epoch": 4.84877580412866,
714
+ "grad_norm": 1.0717830657958984,
715
+ "learning_rate": 4.5154104656745084e-05,
716
+ "loss": 0.2138,
717
+ "step": 10100
718
+ },
719
+ {
720
+ "epoch": 4.896783485357657,
721
+ "grad_norm": 1.0514252185821533,
722
+ "learning_rate": 4.510609697551608e-05,
723
+ "loss": 0.2187,
724
+ "step": 10200
725
+ },
726
+ {
727
+ "epoch": 4.944791166586654,
728
+ "grad_norm": 1.1718051433563232,
729
+ "learning_rate": 4.505808929428709e-05,
730
+ "loss": 0.2179,
731
+ "step": 10300
732
+ },
733
+ {
734
+ "epoch": 4.99279884781565,
735
+ "grad_norm": 1.3955721855163574,
736
+ "learning_rate": 4.501008161305809e-05,
737
+ "loss": 0.2222,
738
+ "step": 10400
739
+ },
740
+ {
741
+ "epoch": 5.040806529044647,
742
+ "grad_norm": 1.0917434692382812,
743
+ "learning_rate": 4.49620739318291e-05,
744
+ "loss": 0.1556,
745
+ "step": 10500
746
+ },
747
+ {
748
+ "epoch": 5.088814210273644,
749
+ "grad_norm": 1.2517110109329224,
750
+ "learning_rate": 4.49140662506001e-05,
751
+ "loss": 0.1442,
752
+ "step": 10600
753
+ },
754
+ {
755
+ "epoch": 5.13682189150264,
756
+ "grad_norm": 1.4139866828918457,
757
+ "learning_rate": 4.48660585693711e-05,
758
+ "loss": 0.1456,
759
+ "step": 10700
760
+ },
761
+ {
762
+ "epoch": 5.184829572731637,
763
+ "grad_norm": 1.4004515409469604,
764
+ "learning_rate": 4.4818050888142106e-05,
765
+ "loss": 0.1475,
766
+ "step": 10800
767
+ },
768
+ {
769
+ "epoch": 5.232837253960634,
770
+ "grad_norm": 1.3576511144638062,
771
+ "learning_rate": 4.477004320691311e-05,
772
+ "loss": 0.1475,
773
+ "step": 10900
774
+ },
775
+ {
776
+ "epoch": 5.280844935189631,
777
+ "grad_norm": 1.2428905963897705,
778
+ "learning_rate": 4.4722035525684116e-05,
779
+ "loss": 0.1505,
780
+ "step": 11000
781
+ },
782
+ {
783
+ "epoch": 5.328852616418627,
784
+ "grad_norm": 1.577462077140808,
785
+ "learning_rate": 4.4674027844455114e-05,
786
+ "loss": 0.1533,
787
+ "step": 11100
788
+ },
789
+ {
790
+ "epoch": 5.376860297647624,
791
+ "grad_norm": 1.5575320720672607,
792
+ "learning_rate": 4.462602016322612e-05,
793
+ "loss": 0.1545,
794
+ "step": 11200
795
+ },
796
+ {
797
+ "epoch": 5.4248679788766205,
798
+ "grad_norm": 1.6392663717269897,
799
+ "learning_rate": 4.4578012481997124e-05,
800
+ "loss": 0.1535,
801
+ "step": 11300
802
+ },
803
+ {
804
+ "epoch": 5.4728756601056165,
805
+ "grad_norm": 1.5311086177825928,
806
+ "learning_rate": 4.453000480076813e-05,
807
+ "loss": 0.1579,
808
+ "step": 11400
809
+ },
810
+ {
811
+ "epoch": 5.5208833413346134,
812
+ "grad_norm": 1.2279739379882812,
813
+ "learning_rate": 4.448199711953913e-05,
814
+ "loss": 0.1616,
815
+ "step": 11500
816
+ },
817
+ {
818
+ "epoch": 5.56889102256361,
819
+ "grad_norm": 1.48116934299469,
820
+ "learning_rate": 4.443398943831013e-05,
821
+ "loss": 0.1603,
822
+ "step": 11600
823
+ },
824
+ {
825
+ "epoch": 5.616898703792607,
826
+ "grad_norm": 1.2573692798614502,
827
+ "learning_rate": 4.438598175708114e-05,
828
+ "loss": 0.1615,
829
+ "step": 11700
830
+ },
831
+ {
832
+ "epoch": 5.664906385021603,
833
+ "grad_norm": 1.3175160884857178,
834
+ "learning_rate": 4.433797407585214e-05,
835
+ "loss": 0.1618,
836
+ "step": 11800
837
+ },
838
+ {
839
+ "epoch": 5.7129140662506,
840
+ "grad_norm": 1.3634432554244995,
841
+ "learning_rate": 4.428996639462314e-05,
842
+ "loss": 0.1634,
843
+ "step": 11900
844
+ },
845
+ {
846
+ "epoch": 5.760921747479597,
847
+ "grad_norm": 1.4681994915008545,
848
+ "learning_rate": 4.4241958713394146e-05,
849
+ "loss": 0.1644,
850
+ "step": 12000
851
+ },
852
+ {
853
+ "epoch": 5.808929428708593,
854
+ "grad_norm": 2.1259870529174805,
855
+ "learning_rate": 4.419395103216515e-05,
856
+ "loss": 0.1637,
857
+ "step": 12100
858
+ },
859
+ {
860
+ "epoch": 5.85693710993759,
861
+ "grad_norm": 1.3507694005966187,
862
+ "learning_rate": 4.4145943350936156e-05,
863
+ "loss": 0.1676,
864
+ "step": 12200
865
+ },
866
+ {
867
+ "epoch": 5.904944791166587,
868
+ "grad_norm": 1.3994966745376587,
869
+ "learning_rate": 4.4097935669707155e-05,
870
+ "loss": 0.1682,
871
+ "step": 12300
872
+ },
873
+ {
874
+ "epoch": 5.952952472395584,
875
+ "grad_norm": 1.37490713596344,
876
+ "learning_rate": 4.404992798847816e-05,
877
+ "loss": 0.1632,
878
+ "step": 12400
879
+ }
880
+ ],
881
+ "logging_steps": 100,
882
+ "max_steps": 104150,
883
+ "num_input_tokens_seen": 0,
884
+ "num_train_epochs": 50,
885
+ "save_steps": 10000,
886
+ "stateful_callbacks": {
887
+ "TrainerControl": {
888
+ "args": {
889
+ "should_epoch_stop": false,
890
+ "should_evaluate": false,
891
+ "should_log": false,
892
+ "should_save": true,
893
+ "should_training_stop": false
894
+ },
895
+ "attributes": {}
896
+ }
897
+ },
898
+ "total_flos": 1.186514987204802e+20,
899
+ "train_batch_size": 16,
900
+ "trial_name": null,
901
+ "trial_params": null
902
+ }