jasoneden commited on
Commit
33ed0e7
1 Parent(s): f6a3d56

Upload 8 files

Browse files
.gitattributes CHANGED
@@ -31,3 +31,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
31
  *.zip filter=lfs diff=lfs merge=lfs -text
32
  *.zst filter=lfs diff=lfs merge=lfs -text
33
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
31
  *.zip filter=lfs diff=lfs merge=lfs -text
32
  *.zst filter=lfs diff=lfs merge=lfs -text
33
  *tfevents* filter=lfs diff=lfs merge=lfs -text
34
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,3 +1,54 @@
1
  ---
2
  license: bigscience-bloom-rail-1.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: bigscience-bloom-rail-1.0
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - squad_v2
7
+ model-index:
8
+ - name: debug_bloom_squad
9
+ results: []
10
  ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # debug_bloom_squad
16
+
17
+ This model is a fine-tuned version of [bigscience/bloom-560m](https://huggingface.co/bigscience/bloom-560m) on the squad_v2 dataset.
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - learning_rate: 3e-05
37
+ - train_batch_size: 6
38
+ - eval_batch_size: 8
39
+ - seed: 42
40
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
+ - lr_scheduler_type: linear
42
+ - num_epochs: 2.0
43
+ - mixed_precision_training: Native AMP
44
+
45
+ ### Training results
46
+
47
+
48
+
49
+ ### Framework versions
50
+
51
+ - Transformers 4.24.0.dev0
52
+ - Pytorch 1.12.1+cu102
53
+ - Datasets 2.6.1
54
+ - Tokenizers 0.13.1
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "train_loss": 3.5304283500368654,
4
+ "train_runtime": 15323.5402,
5
+ "train_samples": 131854,
6
+ "train_samples_per_second": 17.209,
7
+ "train_steps_per_second": 2.868
8
+ }
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bigscience/bloom-560m",
3
+ "apply_residual_connection_post_layernorm": false,
4
+ "architectures": [
5
+ "BloomForQuestionAnswering"
6
+ ],
7
+ "attention_dropout": 0.0,
8
+ "attention_softmax_in_fp32": true,
9
+ "bias_dropout_fusion": true,
10
+ "bos_token_id": 1,
11
+ "eos_token_id": 2,
12
+ "hidden_dropout": 0.0,
13
+ "hidden_size": 1024,
14
+ "initializer_range": 0.02,
15
+ "layer_norm_epsilon": 1e-05,
16
+ "masked_softmax_fusion": true,
17
+ "model_type": "bloom",
18
+ "n_head": 16,
19
+ "n_inner": null,
20
+ "n_layer": 24,
21
+ "offset_alibi": 100,
22
+ "pad_token_id": 3,
23
+ "pretraining_tp": 1,
24
+ "skip_bias_add": true,
25
+ "skip_bias_add_qkv": false,
26
+ "slow_but_exact": false,
27
+ "torch_dtype": "float32",
28
+ "transformers_version": "4.24.0.dev0",
29
+ "unk_token_id": 0,
30
+ "use_cache": true,
31
+ "vocab_size": 250880
32
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "eos_token": "</s>",
4
+ "pad_token": "<pad>",
5
+ "unk_token": "<unk>"
6
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f6efc66e73f1fd69da4f436e48befb519fdff3fe18910850c1d41bd862293a5
3
+ size 14500443
tokenizer_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "name_or_path": "bigscience/bloom-560m",
6
+ "pad_token": "<pad>",
7
+ "padding_side": "left",
8
+ "special_tokens_map_file": null,
9
+ "tokenizer_class": "BloomTokenizer",
10
+ "unk_token": "<unk>"
11
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "train_loss": 3.5304283500368654,
4
+ "train_runtime": 15323.5402,
5
+ "train_samples": 131854,
6
+ "train_samples_per_second": 17.209,
7
+ "train_steps_per_second": 2.868
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,547 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.0,
5
+ "global_step": 43952,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.02,
12
+ "learning_rate": 2.9665544230069168e-05,
13
+ "loss": 8.2605,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.05,
18
+ "learning_rate": 2.9324262832180562e-05,
19
+ "loss": 4.2073,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.07,
24
+ "learning_rate": 2.8982981434291954e-05,
25
+ "loss": 4.0271,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.09,
30
+ "learning_rate": 2.864170003640335e-05,
31
+ "loss": 3.8021,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.11,
36
+ "learning_rate": 2.8300418638514747e-05,
37
+ "loss": 3.7889,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 0.14,
42
+ "learning_rate": 2.795913724062614e-05,
43
+ "loss": 3.9639,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.16,
48
+ "learning_rate": 2.7617855842737534e-05,
49
+ "loss": 3.7584,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 0.18,
54
+ "learning_rate": 2.7276574444848925e-05,
55
+ "loss": 3.9555,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.2,
60
+ "learning_rate": 2.6935975609756097e-05,
61
+ "loss": 3.9079,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 0.23,
66
+ "learning_rate": 2.6594694211867496e-05,
67
+ "loss": 3.9956,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 0.25,
72
+ "learning_rate": 2.6253412813978887e-05,
73
+ "loss": 3.1935,
74
+ "step": 5500
75
+ },
76
+ {
77
+ "epoch": 0.27,
78
+ "learning_rate": 2.5912131416090282e-05,
79
+ "loss": 3.772,
80
+ "step": 6000
81
+ },
82
+ {
83
+ "epoch": 0.3,
84
+ "learning_rate": 2.5572215143793228e-05,
85
+ "loss": 5.1608,
86
+ "step": 6500
87
+ },
88
+ {
89
+ "epoch": 0.32,
90
+ "learning_rate": 2.5230933745904623e-05,
91
+ "loss": 4.1891,
92
+ "step": 7000
93
+ },
94
+ {
95
+ "epoch": 0.34,
96
+ "learning_rate": 2.4889652348016018e-05,
97
+ "loss": 3.9175,
98
+ "step": 7500
99
+ },
100
+ {
101
+ "epoch": 0.36,
102
+ "learning_rate": 2.4548370950127413e-05,
103
+ "loss": 3.9636,
104
+ "step": 8000
105
+ },
106
+ {
107
+ "epoch": 0.39,
108
+ "learning_rate": 2.4207089552238807e-05,
109
+ "loss": 4.1169,
110
+ "step": 8500
111
+ },
112
+ {
113
+ "epoch": 0.41,
114
+ "learning_rate": 2.38658081543502e-05,
115
+ "loss": 3.9023,
116
+ "step": 9000
117
+ },
118
+ {
119
+ "epoch": 0.43,
120
+ "learning_rate": 2.3524526756461594e-05,
121
+ "loss": 3.8476,
122
+ "step": 9500
123
+ },
124
+ {
125
+ "epoch": 0.46,
126
+ "learning_rate": 2.3183245358572992e-05,
127
+ "loss": 3.7756,
128
+ "step": 10000
129
+ },
130
+ {
131
+ "epoch": 0.48,
132
+ "learning_rate": 2.2841963960684384e-05,
133
+ "loss": 4.0936,
134
+ "step": 10500
135
+ },
136
+ {
137
+ "epoch": 0.5,
138
+ "learning_rate": 2.250068256279578e-05,
139
+ "loss": 3.7001,
140
+ "step": 11000
141
+ },
142
+ {
143
+ "epoch": 0.52,
144
+ "learning_rate": 2.215940116490717e-05,
145
+ "loss": 4.269,
146
+ "step": 11500
147
+ },
148
+ {
149
+ "epoch": 0.55,
150
+ "learning_rate": 2.1818119767018565e-05,
151
+ "loss": 3.8019,
152
+ "step": 12000
153
+ },
154
+ {
155
+ "epoch": 0.57,
156
+ "learning_rate": 2.147752093192574e-05,
157
+ "loss": 3.8787,
158
+ "step": 12500
159
+ },
160
+ {
161
+ "epoch": 0.59,
162
+ "learning_rate": 2.113692209683291e-05,
163
+ "loss": 3.9497,
164
+ "step": 13000
165
+ },
166
+ {
167
+ "epoch": 0.61,
168
+ "learning_rate": 2.0795640698944304e-05,
169
+ "loss": 3.8454,
170
+ "step": 13500
171
+ },
172
+ {
173
+ "epoch": 0.64,
174
+ "learning_rate": 2.0454359301055696e-05,
175
+ "loss": 3.824,
176
+ "step": 14000
177
+ },
178
+ {
179
+ "epoch": 0.66,
180
+ "learning_rate": 2.011307790316709e-05,
181
+ "loss": 3.9265,
182
+ "step": 14500
183
+ },
184
+ {
185
+ "epoch": 0.68,
186
+ "learning_rate": 1.977179650527849e-05,
187
+ "loss": 3.9985,
188
+ "step": 15000
189
+ },
190
+ {
191
+ "epoch": 0.71,
192
+ "learning_rate": 1.943051510738988e-05,
193
+ "loss": 3.7808,
194
+ "step": 15500
195
+ },
196
+ {
197
+ "epoch": 0.73,
198
+ "learning_rate": 1.9089233709501275e-05,
199
+ "loss": 4.0853,
200
+ "step": 16000
201
+ },
202
+ {
203
+ "epoch": 0.75,
204
+ "learning_rate": 1.8747952311612667e-05,
205
+ "loss": 4.1285,
206
+ "step": 16500
207
+ },
208
+ {
209
+ "epoch": 0.77,
210
+ "learning_rate": 1.840735347651984e-05,
211
+ "loss": 3.4963,
212
+ "step": 17000
213
+ },
214
+ {
215
+ "epoch": 0.8,
216
+ "learning_rate": 1.8066072078631237e-05,
217
+ "loss": 3.8048,
218
+ "step": 17500
219
+ },
220
+ {
221
+ "epoch": 0.82,
222
+ "learning_rate": 1.772479068074263e-05,
223
+ "loss": 3.5536,
224
+ "step": 18000
225
+ },
226
+ {
227
+ "epoch": 0.84,
228
+ "learning_rate": 1.7383509282854024e-05,
229
+ "loss": 3.6811,
230
+ "step": 18500
231
+ },
232
+ {
233
+ "epoch": 0.86,
234
+ "learning_rate": 1.7042910447761196e-05,
235
+ "loss": 3.6341,
236
+ "step": 19000
237
+ },
238
+ {
239
+ "epoch": 0.89,
240
+ "learning_rate": 1.6701629049872587e-05,
241
+ "loss": 3.5169,
242
+ "step": 19500
243
+ },
244
+ {
245
+ "epoch": 0.91,
246
+ "learning_rate": 1.6360347651983986e-05,
247
+ "loss": 3.9027,
248
+ "step": 20000
249
+ },
250
+ {
251
+ "epoch": 0.93,
252
+ "learning_rate": 1.6019066254095377e-05,
253
+ "loss": 3.9672,
254
+ "step": 20500
255
+ },
256
+ {
257
+ "epoch": 0.96,
258
+ "learning_rate": 1.567846741900255e-05,
259
+ "loss": 3.6631,
260
+ "step": 21000
261
+ },
262
+ {
263
+ "epoch": 0.98,
264
+ "learning_rate": 1.5337186021113944e-05,
265
+ "loss": 3.4308,
266
+ "step": 21500
267
+ },
268
+ {
269
+ "epoch": 1.0,
270
+ "learning_rate": 1.4995904623225337e-05,
271
+ "loss": 3.4298,
272
+ "step": 22000
273
+ },
274
+ {
275
+ "epoch": 1.02,
276
+ "learning_rate": 1.465462322533673e-05,
277
+ "loss": 3.5444,
278
+ "step": 22500
279
+ },
280
+ {
281
+ "epoch": 1.05,
282
+ "learning_rate": 1.4313341827448125e-05,
283
+ "loss": 3.1415,
284
+ "step": 23000
285
+ },
286
+ {
287
+ "epoch": 1.07,
288
+ "learning_rate": 1.397206042955952e-05,
289
+ "loss": 3.1462,
290
+ "step": 23500
291
+ },
292
+ {
293
+ "epoch": 1.09,
294
+ "learning_rate": 1.3630779031670913e-05,
295
+ "loss": 3.0745,
296
+ "step": 24000
297
+ },
298
+ {
299
+ "epoch": 1.11,
300
+ "learning_rate": 1.3289497633782308e-05,
301
+ "loss": 2.8795,
302
+ "step": 24500
303
+ },
304
+ {
305
+ "epoch": 1.14,
306
+ "learning_rate": 1.294889879868948e-05,
307
+ "loss": 2.9813,
308
+ "step": 25000
309
+ },
310
+ {
311
+ "epoch": 1.16,
312
+ "learning_rate": 1.2607617400800874e-05,
313
+ "loss": 2.9452,
314
+ "step": 25500
315
+ },
316
+ {
317
+ "epoch": 1.18,
318
+ "learning_rate": 1.2266336002912269e-05,
319
+ "loss": 3.1609,
320
+ "step": 26000
321
+ },
322
+ {
323
+ "epoch": 1.21,
324
+ "learning_rate": 1.1925054605023662e-05,
325
+ "loss": 3.0651,
326
+ "step": 26500
327
+ },
328
+ {
329
+ "epoch": 1.23,
330
+ "learning_rate": 1.1584455769930834e-05,
331
+ "loss": 3.1032,
332
+ "step": 27000
333
+ },
334
+ {
335
+ "epoch": 1.25,
336
+ "learning_rate": 1.1243174372042229e-05,
337
+ "loss": 3.1298,
338
+ "step": 27500
339
+ },
340
+ {
341
+ "epoch": 1.27,
342
+ "learning_rate": 1.0901892974153622e-05,
343
+ "loss": 3.033,
344
+ "step": 28000
345
+ },
346
+ {
347
+ "epoch": 1.3,
348
+ "learning_rate": 1.0560611576265017e-05,
349
+ "loss": 3.0212,
350
+ "step": 28500
351
+ },
352
+ {
353
+ "epoch": 1.32,
354
+ "learning_rate": 1.0220012741172187e-05,
355
+ "loss": 2.9057,
356
+ "step": 29000
357
+ },
358
+ {
359
+ "epoch": 1.34,
360
+ "learning_rate": 9.878731343283582e-06,
361
+ "loss": 3.1331,
362
+ "step": 29500
363
+ },
364
+ {
365
+ "epoch": 1.37,
366
+ "learning_rate": 9.537449945394977e-06,
367
+ "loss": 3.1259,
368
+ "step": 30000
369
+ },
370
+ {
371
+ "epoch": 1.39,
372
+ "learning_rate": 9.19616854750637e-06,
373
+ "loss": 2.7155,
374
+ "step": 30500
375
+ },
376
+ {
377
+ "epoch": 1.41,
378
+ "learning_rate": 8.855569712413542e-06,
379
+ "loss": 2.9427,
380
+ "step": 31000
381
+ },
382
+ {
383
+ "epoch": 1.43,
384
+ "learning_rate": 8.514288314524936e-06,
385
+ "loss": 2.9428,
386
+ "step": 31500
387
+ },
388
+ {
389
+ "epoch": 1.46,
390
+ "learning_rate": 8.17300691663633e-06,
391
+ "loss": 3.1678,
392
+ "step": 32000
393
+ },
394
+ {
395
+ "epoch": 1.48,
396
+ "learning_rate": 7.831725518747725e-06,
397
+ "loss": 3.4137,
398
+ "step": 32500
399
+ },
400
+ {
401
+ "epoch": 1.5,
402
+ "learning_rate": 7.491126683654897e-06,
403
+ "loss": 2.9195,
404
+ "step": 33000
405
+ },
406
+ {
407
+ "epoch": 1.52,
408
+ "learning_rate": 7.149845285766291e-06,
409
+ "loss": 3.3734,
410
+ "step": 33500
411
+ },
412
+ {
413
+ "epoch": 1.55,
414
+ "learning_rate": 6.808563887877685e-06,
415
+ "loss": 3.2046,
416
+ "step": 34000
417
+ },
418
+ {
419
+ "epoch": 1.57,
420
+ "learning_rate": 6.467282489989079e-06,
421
+ "loss": 3.1923,
422
+ "step": 34500
423
+ },
424
+ {
425
+ "epoch": 1.59,
426
+ "learning_rate": 6.126683654896251e-06,
427
+ "loss": 3.0387,
428
+ "step": 35000
429
+ },
430
+ {
431
+ "epoch": 1.62,
432
+ "learning_rate": 5.785402257007645e-06,
433
+ "loss": 3.0879,
434
+ "step": 35500
435
+ },
436
+ {
437
+ "epoch": 1.64,
438
+ "learning_rate": 5.444120859119039e-06,
439
+ "loss": 3.0943,
440
+ "step": 36000
441
+ },
442
+ {
443
+ "epoch": 1.66,
444
+ "learning_rate": 5.102839461230433e-06,
445
+ "loss": 3.1384,
446
+ "step": 36500
447
+ },
448
+ {
449
+ "epoch": 1.68,
450
+ "learning_rate": 4.762240626137604e-06,
451
+ "loss": 2.8415,
452
+ "step": 37000
453
+ },
454
+ {
455
+ "epoch": 1.71,
456
+ "learning_rate": 4.420959228248999e-06,
457
+ "loss": 3.0139,
458
+ "step": 37500
459
+ },
460
+ {
461
+ "epoch": 1.73,
462
+ "learning_rate": 4.079677830360393e-06,
463
+ "loss": 3.1161,
464
+ "step": 38000
465
+ },
466
+ {
467
+ "epoch": 1.75,
468
+ "learning_rate": 3.7383964324717874e-06,
469
+ "loss": 3.2882,
470
+ "step": 38500
471
+ },
472
+ {
473
+ "epoch": 1.77,
474
+ "learning_rate": 3.3977975973789587e-06,
475
+ "loss": 3.0338,
476
+ "step": 39000
477
+ },
478
+ {
479
+ "epoch": 1.8,
480
+ "learning_rate": 3.056516199490353e-06,
481
+ "loss": 2.876,
482
+ "step": 39500
483
+ },
484
+ {
485
+ "epoch": 1.82,
486
+ "learning_rate": 2.7152348016017477e-06,
487
+ "loss": 3.0888,
488
+ "step": 40000
489
+ },
490
+ {
491
+ "epoch": 1.84,
492
+ "learning_rate": 2.3739534037131413e-06,
493
+ "loss": 3.1206,
494
+ "step": 40500
495
+ },
496
+ {
497
+ "epoch": 1.87,
498
+ "learning_rate": 2.0326720058245358e-06,
499
+ "loss": 3.0954,
500
+ "step": 41000
501
+ },
502
+ {
503
+ "epoch": 1.89,
504
+ "learning_rate": 1.6920731707317074e-06,
505
+ "loss": 3.4009,
506
+ "step": 41500
507
+ },
508
+ {
509
+ "epoch": 1.91,
510
+ "learning_rate": 1.3507917728431015e-06,
511
+ "loss": 3.1574,
512
+ "step": 42000
513
+ },
514
+ {
515
+ "epoch": 1.93,
516
+ "learning_rate": 1.009510374954496e-06,
517
+ "loss": 2.8075,
518
+ "step": 42500
519
+ },
520
+ {
521
+ "epoch": 1.96,
522
+ "learning_rate": 6.682289770658902e-07,
523
+ "loss": 3.1462,
524
+ "step": 43000
525
+ },
526
+ {
527
+ "epoch": 1.98,
528
+ "learning_rate": 3.2763014197306156e-07,
529
+ "loss": 3.0345,
530
+ "step": 43500
531
+ },
532
+ {
533
+ "epoch": 2.0,
534
+ "step": 43952,
535
+ "total_flos": 1.836818419405271e+17,
536
+ "train_loss": 3.5304283500368654,
537
+ "train_runtime": 15323.5402,
538
+ "train_samples_per_second": 17.209,
539
+ "train_steps_per_second": 2.868
540
+ }
541
+ ],
542
+ "max_steps": 43952,
543
+ "num_train_epochs": 2,
544
+ "total_flos": 1.836818419405271e+17,
545
+ "trial_name": null,
546
+ "trial_params": null
547
+ }