BilelDJ commited on
Commit
565dee2
1 Parent(s): 1e8c912

Training in progress, step 141

Browse files
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04d4c56c3c01ab6d32b9236ca3a5acaa749a3826d60565b3c5b9f0332f9afeec
3
  size 605156676
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7af941ae7c70d0f59f8e7ce775861aea3e9ba4983a483c8ffc10e6e14e547b30
3
  size 605156676
run-qx9gqtq5/checkpoint-141/config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "openai/clip-vit-base-patch32",
3
+ "architectures": [
4
+ "CLIPModel"
5
+ ],
6
+ "initializer_factor": 1.0,
7
+ "logit_scale_init_value": 2.6592,
8
+ "model_type": "clip",
9
+ "projection_dim": 512,
10
+ "text_config": {
11
+ "bos_token_id": 0,
12
+ "dropout": 0.0,
13
+ "eos_token_id": 2,
14
+ "model_type": "clip_text_model"
15
+ },
16
+ "torch_dtype": "float32",
17
+ "transformers_version": "4.42.0.dev0",
18
+ "vision_config": {
19
+ "dropout": 0.0,
20
+ "model_type": "clip_vision_model"
21
+ }
22
+ }
run-qx9gqtq5/checkpoint-141/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7af941ae7c70d0f59f8e7ce775861aea3e9ba4983a483c8ffc10e6e14e547b30
3
+ size 605156676
run-qx9gqtq5/checkpoint-141/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:208f59b014c2615f90d7f36d9d4175ac6de9134538c60a8c493966233fe9c666
3
+ size 1210551612
run-qx9gqtq5/checkpoint-141/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90840b2cfbd00f077db878a1cf083ee4df3c7d151bb1846cdc939d37fb68a456
3
+ size 14244
run-qx9gqtq5/checkpoint-141/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:163a97c22509e0550c2adadfa00a8798148bcdb46f07d4c64faa3a90734a5059
3
+ size 1064
run-qx9gqtq5/checkpoint-141/trainer_state.json ADDED
@@ -0,0 +1,745 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 3,
6
+ "global_step": 141,
7
+ "is_hyper_param_search": true,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.06382978723404255,
13
+ "grad_norm": 134.82960510253906,
14
+ "learning_rate": 3.4524225370179614e-06,
15
+ "loss": 2.0515,
16
+ "step": 3
17
+ },
18
+ {
19
+ "epoch": 0.06382978723404255,
20
+ "eval_loss": 1.1583281755447388,
21
+ "eval_runtime": 97.0528,
22
+ "eval_samples_per_second": 3.091,
23
+ "eval_steps_per_second": 0.103,
24
+ "step": 3
25
+ },
26
+ {
27
+ "epoch": 0.1276595744680851,
28
+ "grad_norm": 141.0943145751953,
29
+ "learning_rate": 3.377369873169745e-06,
30
+ "loss": 1.5004,
31
+ "step": 6
32
+ },
33
+ {
34
+ "epoch": 0.1276595744680851,
35
+ "eval_loss": 1.1162742376327515,
36
+ "eval_runtime": 60.2202,
37
+ "eval_samples_per_second": 4.982,
38
+ "eval_steps_per_second": 0.166,
39
+ "step": 6
40
+ },
41
+ {
42
+ "epoch": 0.19148936170212766,
43
+ "grad_norm": 134.9931640625,
44
+ "learning_rate": 3.3023172093215284e-06,
45
+ "loss": 1.5627,
46
+ "step": 9
47
+ },
48
+ {
49
+ "epoch": 0.19148936170212766,
50
+ "eval_loss": 1.0802254676818848,
51
+ "eval_runtime": 41.4969,
52
+ "eval_samples_per_second": 7.229,
53
+ "eval_steps_per_second": 0.241,
54
+ "step": 9
55
+ },
56
+ {
57
+ "epoch": 0.2553191489361702,
58
+ "grad_norm": 113.67478942871094,
59
+ "learning_rate": 3.227264545473312e-06,
60
+ "loss": 1.3625,
61
+ "step": 12
62
+ },
63
+ {
64
+ "epoch": 0.2553191489361702,
65
+ "eval_loss": 1.0606831312179565,
66
+ "eval_runtime": 41.3217,
67
+ "eval_samples_per_second": 7.26,
68
+ "eval_steps_per_second": 0.242,
69
+ "step": 12
70
+ },
71
+ {
72
+ "epoch": 0.3191489361702128,
73
+ "grad_norm": 105.70118713378906,
74
+ "learning_rate": 3.152211881625095e-06,
75
+ "loss": 1.6648,
76
+ "step": 15
77
+ },
78
+ {
79
+ "epoch": 0.3191489361702128,
80
+ "eval_loss": 1.0375689268112183,
81
+ "eval_runtime": 48.7659,
82
+ "eval_samples_per_second": 6.152,
83
+ "eval_steps_per_second": 0.205,
84
+ "step": 15
85
+ },
86
+ {
87
+ "epoch": 0.3829787234042553,
88
+ "grad_norm": 96.33596801757812,
89
+ "learning_rate": 3.077159217776879e-06,
90
+ "loss": 1.3305,
91
+ "step": 18
92
+ },
93
+ {
94
+ "epoch": 0.3829787234042553,
95
+ "eval_loss": 1.0117995738983154,
96
+ "eval_runtime": 41.5605,
97
+ "eval_samples_per_second": 7.218,
98
+ "eval_steps_per_second": 0.241,
99
+ "step": 18
100
+ },
101
+ {
102
+ "epoch": 0.44680851063829785,
103
+ "grad_norm": 105.89424133300781,
104
+ "learning_rate": 3.002106553928662e-06,
105
+ "loss": 1.2195,
106
+ "step": 21
107
+ },
108
+ {
109
+ "epoch": 0.44680851063829785,
110
+ "eval_loss": 0.9878906011581421,
111
+ "eval_runtime": 44.9672,
112
+ "eval_samples_per_second": 6.672,
113
+ "eval_steps_per_second": 0.222,
114
+ "step": 21
115
+ },
116
+ {
117
+ "epoch": 0.5106382978723404,
118
+ "grad_norm": 96.90633392333984,
119
+ "learning_rate": 2.9270538900804454e-06,
120
+ "loss": 1.4076,
121
+ "step": 24
122
+ },
123
+ {
124
+ "epoch": 0.5106382978723404,
125
+ "eval_loss": 0.9666847586631775,
126
+ "eval_runtime": 52.2922,
127
+ "eval_samples_per_second": 5.737,
128
+ "eval_steps_per_second": 0.191,
129
+ "step": 24
130
+ },
131
+ {
132
+ "epoch": 0.574468085106383,
133
+ "grad_norm": 101.37423706054688,
134
+ "learning_rate": 2.852001226232229e-06,
135
+ "loss": 1.5353,
136
+ "step": 27
137
+ },
138
+ {
139
+ "epoch": 0.574468085106383,
140
+ "eval_loss": 0.9519509077072144,
141
+ "eval_runtime": 41.7527,
142
+ "eval_samples_per_second": 7.185,
143
+ "eval_steps_per_second": 0.24,
144
+ "step": 27
145
+ },
146
+ {
147
+ "epoch": 0.6382978723404256,
148
+ "grad_norm": 92.97295379638672,
149
+ "learning_rate": 2.7769485623840124e-06,
150
+ "loss": 1.2954,
151
+ "step": 30
152
+ },
153
+ {
154
+ "epoch": 0.6382978723404256,
155
+ "eval_loss": 0.937315046787262,
156
+ "eval_runtime": 41.2373,
157
+ "eval_samples_per_second": 7.275,
158
+ "eval_steps_per_second": 0.242,
159
+ "step": 30
160
+ },
161
+ {
162
+ "epoch": 0.7021276595744681,
163
+ "grad_norm": 110.28973388671875,
164
+ "learning_rate": 2.701895898535796e-06,
165
+ "loss": 1.489,
166
+ "step": 33
167
+ },
168
+ {
169
+ "epoch": 0.7021276595744681,
170
+ "eval_loss": 0.9187784790992737,
171
+ "eval_runtime": 42.0237,
172
+ "eval_samples_per_second": 7.139,
173
+ "eval_steps_per_second": 0.238,
174
+ "step": 33
175
+ },
176
+ {
177
+ "epoch": 0.7659574468085106,
178
+ "grad_norm": 93.82958984375,
179
+ "learning_rate": 2.6268432346875793e-06,
180
+ "loss": 1.5987,
181
+ "step": 36
182
+ },
183
+ {
184
+ "epoch": 0.7659574468085106,
185
+ "eval_loss": 0.9012949466705322,
186
+ "eval_runtime": 41.6757,
187
+ "eval_samples_per_second": 7.198,
188
+ "eval_steps_per_second": 0.24,
189
+ "step": 36
190
+ },
191
+ {
192
+ "epoch": 0.8297872340425532,
193
+ "grad_norm": 74.33155822753906,
194
+ "learning_rate": 2.551790570839363e-06,
195
+ "loss": 1.3207,
196
+ "step": 39
197
+ },
198
+ {
199
+ "epoch": 0.8297872340425532,
200
+ "eval_loss": 0.8868340849876404,
201
+ "eval_runtime": 41.5644,
202
+ "eval_samples_per_second": 7.218,
203
+ "eval_steps_per_second": 0.241,
204
+ "step": 39
205
+ },
206
+ {
207
+ "epoch": 0.8936170212765957,
208
+ "grad_norm": 100.4666519165039,
209
+ "learning_rate": 2.4767379069911463e-06,
210
+ "loss": 1.3121,
211
+ "step": 42
212
+ },
213
+ {
214
+ "epoch": 0.8936170212765957,
215
+ "eval_loss": 0.8769957423210144,
216
+ "eval_runtime": 51.3924,
217
+ "eval_samples_per_second": 5.837,
218
+ "eval_steps_per_second": 0.195,
219
+ "step": 42
220
+ },
221
+ {
222
+ "epoch": 0.9574468085106383,
223
+ "grad_norm": 78.03990173339844,
224
+ "learning_rate": 2.4016852431429298e-06,
225
+ "loss": 1.198,
226
+ "step": 45
227
+ },
228
+ {
229
+ "epoch": 0.9574468085106383,
230
+ "eval_loss": 0.8720031976699829,
231
+ "eval_runtime": 41.262,
232
+ "eval_samples_per_second": 7.271,
233
+ "eval_steps_per_second": 0.242,
234
+ "step": 45
235
+ },
236
+ {
237
+ "epoch": 1.0212765957446808,
238
+ "grad_norm": 48.33544158935547,
239
+ "learning_rate": 2.326632579294713e-06,
240
+ "loss": 1.0786,
241
+ "step": 48
242
+ },
243
+ {
244
+ "epoch": 1.0212765957446808,
245
+ "eval_loss": 0.8655586838722229,
246
+ "eval_runtime": 41.5498,
247
+ "eval_samples_per_second": 7.22,
248
+ "eval_steps_per_second": 0.241,
249
+ "step": 48
250
+ },
251
+ {
252
+ "epoch": 1.0851063829787233,
253
+ "grad_norm": 35.39493179321289,
254
+ "learning_rate": 2.2515799154464967e-06,
255
+ "loss": 0.7367,
256
+ "step": 51
257
+ },
258
+ {
259
+ "epoch": 1.0851063829787233,
260
+ "eval_loss": 0.8586752414703369,
261
+ "eval_runtime": 47.8744,
262
+ "eval_samples_per_second": 6.266,
263
+ "eval_steps_per_second": 0.209,
264
+ "step": 51
265
+ },
266
+ {
267
+ "epoch": 1.148936170212766,
268
+ "grad_norm": 38.954620361328125,
269
+ "learning_rate": 2.17652725159828e-06,
270
+ "loss": 0.5906,
271
+ "step": 54
272
+ },
273
+ {
274
+ "epoch": 1.148936170212766,
275
+ "eval_loss": 0.854831874370575,
276
+ "eval_runtime": 41.7553,
277
+ "eval_samples_per_second": 7.185,
278
+ "eval_steps_per_second": 0.239,
279
+ "step": 54
280
+ },
281
+ {
282
+ "epoch": 1.2127659574468086,
283
+ "grad_norm": 61.59217071533203,
284
+ "learning_rate": 2.1014745877500633e-06,
285
+ "loss": 0.794,
286
+ "step": 57
287
+ },
288
+ {
289
+ "epoch": 1.2127659574468086,
290
+ "eval_loss": 0.8518243432044983,
291
+ "eval_runtime": 41.3107,
292
+ "eval_samples_per_second": 7.262,
293
+ "eval_steps_per_second": 0.242,
294
+ "step": 57
295
+ },
296
+ {
297
+ "epoch": 1.2765957446808511,
298
+ "grad_norm": 43.460899353027344,
299
+ "learning_rate": 2.026421923901847e-06,
300
+ "loss": 0.6699,
301
+ "step": 60
302
+ },
303
+ {
304
+ "epoch": 1.2765957446808511,
305
+ "eval_loss": 0.849520742893219,
306
+ "eval_runtime": 41.3487,
307
+ "eval_samples_per_second": 7.255,
308
+ "eval_steps_per_second": 0.242,
309
+ "step": 60
310
+ },
311
+ {
312
+ "epoch": 1.3404255319148937,
313
+ "grad_norm": 40.51012420654297,
314
+ "learning_rate": 1.9513692600536303e-06,
315
+ "loss": 0.5351,
316
+ "step": 63
317
+ },
318
+ {
319
+ "epoch": 1.3404255319148937,
320
+ "eval_loss": 0.8519204258918762,
321
+ "eval_runtime": 41.2821,
322
+ "eval_samples_per_second": 7.267,
323
+ "eval_steps_per_second": 0.242,
324
+ "step": 63
325
+ },
326
+ {
327
+ "epoch": 1.4042553191489362,
328
+ "grad_norm": 55.731903076171875,
329
+ "learning_rate": 1.8763165962054137e-06,
330
+ "loss": 0.6526,
331
+ "step": 66
332
+ },
333
+ {
334
+ "epoch": 1.4042553191489362,
335
+ "eval_loss": 0.85467129945755,
336
+ "eval_runtime": 41.1797,
337
+ "eval_samples_per_second": 7.285,
338
+ "eval_steps_per_second": 0.243,
339
+ "step": 66
340
+ },
341
+ {
342
+ "epoch": 1.4680851063829787,
343
+ "grad_norm": 55.70009231567383,
344
+ "learning_rate": 1.801263932357197e-06,
345
+ "loss": 0.7753,
346
+ "step": 69
347
+ },
348
+ {
349
+ "epoch": 1.4680851063829787,
350
+ "eval_loss": 0.8556016087532043,
351
+ "eval_runtime": 41.2215,
352
+ "eval_samples_per_second": 7.278,
353
+ "eval_steps_per_second": 0.243,
354
+ "step": 69
355
+ },
356
+ {
357
+ "epoch": 1.5319148936170213,
358
+ "grad_norm": 51.552974700927734,
359
+ "learning_rate": 1.7262112685089807e-06,
360
+ "loss": 0.6996,
361
+ "step": 72
362
+ },
363
+ {
364
+ "epoch": 1.5319148936170213,
365
+ "eval_loss": 0.8541069626808167,
366
+ "eval_runtime": 41.8446,
367
+ "eval_samples_per_second": 7.169,
368
+ "eval_steps_per_second": 0.239,
369
+ "step": 72
370
+ },
371
+ {
372
+ "epoch": 1.5957446808510638,
373
+ "grad_norm": 61.95724105834961,
374
+ "learning_rate": 1.6511586046607642e-06,
375
+ "loss": 0.8078,
376
+ "step": 75
377
+ },
378
+ {
379
+ "epoch": 1.5957446808510638,
380
+ "eval_loss": 0.8522682785987854,
381
+ "eval_runtime": 45.1234,
382
+ "eval_samples_per_second": 6.648,
383
+ "eval_steps_per_second": 0.222,
384
+ "step": 75
385
+ },
386
+ {
387
+ "epoch": 1.6595744680851063,
388
+ "grad_norm": 51.09059524536133,
389
+ "learning_rate": 1.5761059408125475e-06,
390
+ "loss": 0.6704,
391
+ "step": 78
392
+ },
393
+ {
394
+ "epoch": 1.6595744680851063,
395
+ "eval_loss": 0.8499072194099426,
396
+ "eval_runtime": 41.6072,
397
+ "eval_samples_per_second": 7.21,
398
+ "eval_steps_per_second": 0.24,
399
+ "step": 78
400
+ },
401
+ {
402
+ "epoch": 1.7234042553191489,
403
+ "grad_norm": 57.01670455932617,
404
+ "learning_rate": 1.501053276964331e-06,
405
+ "loss": 0.5976,
406
+ "step": 81
407
+ },
408
+ {
409
+ "epoch": 1.7234042553191489,
410
+ "eval_loss": 0.8474313616752625,
411
+ "eval_runtime": 41.405,
412
+ "eval_samples_per_second": 7.246,
413
+ "eval_steps_per_second": 0.242,
414
+ "step": 81
415
+ },
416
+ {
417
+ "epoch": 1.7872340425531914,
418
+ "grad_norm": 58.575130462646484,
419
+ "learning_rate": 1.4260006131161144e-06,
420
+ "loss": 0.6588,
421
+ "step": 84
422
+ },
423
+ {
424
+ "epoch": 1.7872340425531914,
425
+ "eval_loss": 0.8449164032936096,
426
+ "eval_runtime": 41.6188,
427
+ "eval_samples_per_second": 7.208,
428
+ "eval_steps_per_second": 0.24,
429
+ "step": 84
430
+ },
431
+ {
432
+ "epoch": 1.851063829787234,
433
+ "grad_norm": 60.291622161865234,
434
+ "learning_rate": 1.350947949267898e-06,
435
+ "loss": 0.7081,
436
+ "step": 87
437
+ },
438
+ {
439
+ "epoch": 1.851063829787234,
440
+ "eval_loss": 0.8420330882072449,
441
+ "eval_runtime": 41.9537,
442
+ "eval_samples_per_second": 7.151,
443
+ "eval_steps_per_second": 0.238,
444
+ "step": 87
445
+ },
446
+ {
447
+ "epoch": 1.9148936170212765,
448
+ "grad_norm": 50.23271942138672,
449
+ "learning_rate": 1.2758952854196814e-06,
450
+ "loss": 0.5632,
451
+ "step": 90
452
+ },
453
+ {
454
+ "epoch": 1.9148936170212765,
455
+ "eval_loss": 0.8388514518737793,
456
+ "eval_runtime": 41.6015,
457
+ "eval_samples_per_second": 7.211,
458
+ "eval_steps_per_second": 0.24,
459
+ "step": 90
460
+ },
461
+ {
462
+ "epoch": 1.978723404255319,
463
+ "grad_norm": 52.8415412902832,
464
+ "learning_rate": 1.2008426215714649e-06,
465
+ "loss": 0.6249,
466
+ "step": 93
467
+ },
468
+ {
469
+ "epoch": 1.978723404255319,
470
+ "eval_loss": 0.8353903293609619,
471
+ "eval_runtime": 41.6241,
472
+ "eval_samples_per_second": 7.207,
473
+ "eval_steps_per_second": 0.24,
474
+ "step": 93
475
+ },
476
+ {
477
+ "epoch": 2.0425531914893615,
478
+ "grad_norm": 55.50529098510742,
479
+ "learning_rate": 1.1257899577232484e-06,
480
+ "loss": 0.612,
481
+ "step": 96
482
+ },
483
+ {
484
+ "epoch": 2.0425531914893615,
485
+ "eval_loss": 0.8322966694831848,
486
+ "eval_runtime": 41.6827,
487
+ "eval_samples_per_second": 7.197,
488
+ "eval_steps_per_second": 0.24,
489
+ "step": 96
490
+ },
491
+ {
492
+ "epoch": 2.106382978723404,
493
+ "grad_norm": 33.535972595214844,
494
+ "learning_rate": 1.0507372938750316e-06,
495
+ "loss": 0.4171,
496
+ "step": 99
497
+ },
498
+ {
499
+ "epoch": 2.106382978723404,
500
+ "eval_loss": 0.8299477100372314,
501
+ "eval_runtime": 41.6454,
502
+ "eval_samples_per_second": 7.204,
503
+ "eval_steps_per_second": 0.24,
504
+ "step": 99
505
+ },
506
+ {
507
+ "epoch": 2.1702127659574466,
508
+ "grad_norm": 33.902915954589844,
509
+ "learning_rate": 9.756846300268151e-07,
510
+ "loss": 0.3473,
511
+ "step": 102
512
+ },
513
+ {
514
+ "epoch": 2.1702127659574466,
515
+ "eval_loss": 0.8301065564155579,
516
+ "eval_runtime": 41.3799,
517
+ "eval_samples_per_second": 7.25,
518
+ "eval_steps_per_second": 0.242,
519
+ "step": 102
520
+ },
521
+ {
522
+ "epoch": 2.2340425531914896,
523
+ "grad_norm": 54.96790313720703,
524
+ "learning_rate": 9.006319661785985e-07,
525
+ "loss": 0.4751,
526
+ "step": 105
527
+ },
528
+ {
529
+ "epoch": 2.2340425531914896,
530
+ "eval_loss": 0.8314597606658936,
531
+ "eval_runtime": 41.4767,
532
+ "eval_samples_per_second": 7.233,
533
+ "eval_steps_per_second": 0.241,
534
+ "step": 105
535
+ },
536
+ {
537
+ "epoch": 2.297872340425532,
538
+ "grad_norm": 43.61612319946289,
539
+ "learning_rate": 8.255793023303821e-07,
540
+ "loss": 0.4088,
541
+ "step": 108
542
+ },
543
+ {
544
+ "epoch": 2.297872340425532,
545
+ "eval_loss": 0.8334099054336548,
546
+ "eval_runtime": 60.1786,
547
+ "eval_samples_per_second": 4.985,
548
+ "eval_steps_per_second": 0.166,
549
+ "step": 108
550
+ },
551
+ {
552
+ "epoch": 2.3617021276595747,
553
+ "grad_norm": 41.3808708190918,
554
+ "learning_rate": 7.505266384821655e-07,
555
+ "loss": 0.3174,
556
+ "step": 111
557
+ },
558
+ {
559
+ "epoch": 2.3617021276595747,
560
+ "eval_loss": 0.8363153338432312,
561
+ "eval_runtime": 41.6562,
562
+ "eval_samples_per_second": 7.202,
563
+ "eval_steps_per_second": 0.24,
564
+ "step": 111
565
+ },
566
+ {
567
+ "epoch": 2.425531914893617,
568
+ "grad_norm": 39.414100646972656,
569
+ "learning_rate": 6.75473974633949e-07,
570
+ "loss": 0.3267,
571
+ "step": 114
572
+ },
573
+ {
574
+ "epoch": 2.425531914893617,
575
+ "eval_loss": 0.8386973142623901,
576
+ "eval_runtime": 41.5441,
577
+ "eval_samples_per_second": 7.221,
578
+ "eval_steps_per_second": 0.241,
579
+ "step": 114
580
+ },
581
+ {
582
+ "epoch": 2.4893617021276597,
583
+ "grad_norm": 41.55443572998047,
584
+ "learning_rate": 6.004213107857324e-07,
585
+ "loss": 0.3142,
586
+ "step": 117
587
+ },
588
+ {
589
+ "epoch": 2.4893617021276597,
590
+ "eval_loss": 0.8412825465202332,
591
+ "eval_runtime": 41.7129,
592
+ "eval_samples_per_second": 7.192,
593
+ "eval_steps_per_second": 0.24,
594
+ "step": 117
595
+ },
596
+ {
597
+ "epoch": 2.5531914893617023,
598
+ "grad_norm": 45.54193115234375,
599
+ "learning_rate": 5.253686469375158e-07,
600
+ "loss": 0.4253,
601
+ "step": 120
602
+ },
603
+ {
604
+ "epoch": 2.5531914893617023,
605
+ "eval_loss": 0.8432453274726868,
606
+ "eval_runtime": 41.5872,
607
+ "eval_samples_per_second": 7.214,
608
+ "eval_steps_per_second": 0.24,
609
+ "step": 120
610
+ },
611
+ {
612
+ "epoch": 2.617021276595745,
613
+ "grad_norm": 58.93795394897461,
614
+ "learning_rate": 4.5031598308929925e-07,
615
+ "loss": 0.4269,
616
+ "step": 123
617
+ },
618
+ {
619
+ "epoch": 2.617021276595745,
620
+ "eval_loss": 0.8454075455665588,
621
+ "eval_runtime": 41.7045,
622
+ "eval_samples_per_second": 7.193,
623
+ "eval_steps_per_second": 0.24,
624
+ "step": 123
625
+ },
626
+ {
627
+ "epoch": 2.6808510638297873,
628
+ "grad_norm": 30.54688835144043,
629
+ "learning_rate": 3.7526331924108274e-07,
630
+ "loss": 0.2765,
631
+ "step": 126
632
+ },
633
+ {
634
+ "epoch": 2.6808510638297873,
635
+ "eval_loss": 0.8469324707984924,
636
+ "eval_runtime": 41.6321,
637
+ "eval_samples_per_second": 7.206,
638
+ "eval_steps_per_second": 0.24,
639
+ "step": 126
640
+ },
641
+ {
642
+ "epoch": 2.74468085106383,
643
+ "grad_norm": 46.656986236572266,
644
+ "learning_rate": 3.002106553928662e-07,
645
+ "loss": 0.3071,
646
+ "step": 129
647
+ },
648
+ {
649
+ "epoch": 2.74468085106383,
650
+ "eval_loss": 0.848118782043457,
651
+ "eval_runtime": 41.3133,
652
+ "eval_samples_per_second": 7.262,
653
+ "eval_steps_per_second": 0.242,
654
+ "step": 129
655
+ },
656
+ {
657
+ "epoch": 2.8085106382978724,
658
+ "grad_norm": 41.179161071777344,
659
+ "learning_rate": 2.2515799154464963e-07,
660
+ "loss": 0.3277,
661
+ "step": 132
662
+ },
663
+ {
664
+ "epoch": 2.8085106382978724,
665
+ "eval_loss": 0.8486995100975037,
666
+ "eval_runtime": 41.4208,
667
+ "eval_samples_per_second": 7.243,
668
+ "eval_steps_per_second": 0.241,
669
+ "step": 132
670
+ },
671
+ {
672
+ "epoch": 2.872340425531915,
673
+ "grad_norm": 35.64809036254883,
674
+ "learning_rate": 1.501053276964331e-07,
675
+ "loss": 0.3206,
676
+ "step": 135
677
+ },
678
+ {
679
+ "epoch": 2.872340425531915,
680
+ "eval_loss": 0.8487841486930847,
681
+ "eval_runtime": 41.4922,
682
+ "eval_samples_per_second": 7.23,
683
+ "eval_steps_per_second": 0.241,
684
+ "step": 135
685
+ },
686
+ {
687
+ "epoch": 2.9361702127659575,
688
+ "grad_norm": 39.29479217529297,
689
+ "learning_rate": 7.505266384821656e-08,
690
+ "loss": 0.4095,
691
+ "step": 138
692
+ },
693
+ {
694
+ "epoch": 2.9361702127659575,
695
+ "eval_loss": 0.848578929901123,
696
+ "eval_runtime": 41.4984,
697
+ "eval_samples_per_second": 7.229,
698
+ "eval_steps_per_second": 0.241,
699
+ "step": 138
700
+ },
701
+ {
702
+ "epoch": 3.0,
703
+ "grad_norm": 37.54413604736328,
704
+ "learning_rate": 0.0,
705
+ "loss": 0.2987,
706
+ "step": 141
707
+ },
708
+ {
709
+ "epoch": 3.0,
710
+ "eval_loss": 0.8485209941864014,
711
+ "eval_runtime": 41.5472,
712
+ "eval_samples_per_second": 7.221,
713
+ "eval_steps_per_second": 0.241,
714
+ "step": 141
715
+ }
716
+ ],
717
+ "logging_steps": 3,
718
+ "max_steps": 141,
719
+ "num_input_tokens_seen": 0,
720
+ "num_train_epochs": 3,
721
+ "save_steps": 500,
722
+ "stateful_callbacks": {
723
+ "TrainerControl": {
724
+ "args": {
725
+ "should_epoch_stop": false,
726
+ "should_evaluate": false,
727
+ "should_log": false,
728
+ "should_save": true,
729
+ "should_training_stop": true
730
+ },
731
+ "attributes": {}
732
+ }
733
+ },
734
+ "total_flos": 523328480700102.0,
735
+ "train_batch_size": 64,
736
+ "trial_name": null,
737
+ "trial_params": {
738
+ "_wandb": {},
739
+ "assignments": {},
740
+ "decay": 0.1,
741
+ "learning_rate": 3.527475200866178e-06,
742
+ "metric": "eval/loss",
743
+ "per_device_train_batch_size": 64
744
+ }
745
+ }
run-qx9gqtq5/checkpoint-141/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ffe3953d97093514d4a6a5d7fe29c2ca1b676f67e792cd5ccf81d7e5cbf3040
3
+ size 5112
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:12f53ae926fcf51f8fe6702a1a01fbe955cf0435dc5c5228f5dc3f0a507d214d
3
  size 5112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ffe3953d97093514d4a6a5d7fe29c2ca1b676f67e792cd5ccf81d7e5cbf3040
3
  size 5112