Hailey Schoelkopf commited on
Commit
5050b09
1 Parent(s): 2d8f9e7

Push to HF hub

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
.gitignore ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint-15000/
2
+ events.out.tfevents.1663038340.vesna.703613.2
3
+ events.out.tfevents.1662992784.vesna.703613.0
4
+ 1662992784.275189/
5
+ checkpoint-5000/
6
+ checkpoint-20000/
7
+ eval_results.json
8
+ word_embeddings_layernorm.pt
9
+ word_embeddings.pt
10
+ train_results.json
11
+ checkpoint-25000/
12
+ all_results.json
13
+ checkpoint-10000/
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bigscience/bloom-560m",
3
+ "adapters": {
4
+ "adapters": {},
5
+ "config_map": {},
6
+ "fusion_config_map": {},
7
+ "fusions": {}
8
+ },
9
+ "apply_residual_connection_post_layernorm": false,
10
+ "architectures": [
11
+ "BloomForCausalLM"
12
+ ],
13
+ "attention_dropout": 0.0,
14
+ "attention_softmax_in_fp32": true,
15
+ "bias_dropout_fusion": true,
16
+ "bos_token_id": 1,
17
+ "eos_token_id": 2,
18
+ "hidden_dropout": 0.0,
19
+ "hidden_size": 1024,
20
+ "initializer_range": 0.02,
21
+ "layer_norm_epsilon": 1e-05,
22
+ "masked_softmax_fusion": true,
23
+ "model_type": "bloom",
24
+ "n_head": 16,
25
+ "n_inner": null,
26
+ "n_layer": 24,
27
+ "offset_alibi": 100,
28
+ "pad_token_id": 3,
29
+ "pretraining_tp": 1,
30
+ "seq_length": 2048,
31
+ "skip_bias_add": true,
32
+ "skip_bias_add_qkv": false,
33
+ "slow_but_exact": false,
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.20.0.dev0",
36
+ "unk_token_id": 0,
37
+ "use_cache": true,
38
+ "vocab_size": 250880
39
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a47f3efc641d356f99e7da022403c8ec5c69b65f550c74aca7cd6aaa6e77b7e1
3
+ size 2236955191
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "pad_token": "<pad>"}
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 @@
 
 
1
+ {"unk_token": "<unk>", "bos_token": "<s>", "eos_token": "</s>", "pad_token": "<pad>", "add_prefix_space": false, "name_or_path": "bigscience/bloom-560m", "special_tokens_map_file": null, "padding_side": "left", "tokenizer_class": "BloomTokenizer"}
trainer_state.json ADDED
@@ -0,0 +1,365 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.6942951679229736,
3
+ "best_model_checkpoint": "/home/lily/nos6/bloom-adapt/exps_ru/bigscience/bloom-560m_ru_adpt_continual-pretrain_original-frozen_100_000samples/checkpoint-25000",
4
+ "epoch": 1.462073805485701,
5
+ "global_step": 25000,
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.03,
12
+ "learning_rate": 9.8e-05,
13
+ "loss": 2.6927,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.06,
18
+ "learning_rate": 9.6e-05,
19
+ "loss": 2.3553,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.09,
24
+ "learning_rate": 9.4e-05,
25
+ "loss": 2.2227,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.12,
30
+ "learning_rate": 9.200000000000001e-05,
31
+ "loss": 2.1588,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.15,
36
+ "learning_rate": 9e-05,
37
+ "loss": 2.1143,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 0.18,
42
+ "learning_rate": 8.800000000000001e-05,
43
+ "loss": 2.0865,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.2,
48
+ "learning_rate": 8.6e-05,
49
+ "loss": 2.034,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 0.23,
54
+ "learning_rate": 8.4e-05,
55
+ "loss": 2.0258,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.26,
60
+ "learning_rate": 8.2e-05,
61
+ "loss": 2.001,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 0.29,
66
+ "learning_rate": 8e-05,
67
+ "loss": 1.9695,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 0.29,
72
+ "eval_loss": 1.9710078239440918,
73
+ "eval_runtime": 488.0694,
74
+ "eval_samples_per_second": 13.633,
75
+ "eval_steps_per_second": 3.409,
76
+ "step": 5000
77
+ },
78
+ {
79
+ "epoch": 0.32,
80
+ "learning_rate": 7.800000000000001e-05,
81
+ "loss": 1.9591,
82
+ "step": 5500
83
+ },
84
+ {
85
+ "epoch": 0.35,
86
+ "learning_rate": 7.6e-05,
87
+ "loss": 1.9426,
88
+ "step": 6000
89
+ },
90
+ {
91
+ "epoch": 0.38,
92
+ "learning_rate": 7.4e-05,
93
+ "loss": 1.9172,
94
+ "step": 6500
95
+ },
96
+ {
97
+ "epoch": 0.41,
98
+ "learning_rate": 7.2e-05,
99
+ "loss": 1.9401,
100
+ "step": 7000
101
+ },
102
+ {
103
+ "epoch": 0.44,
104
+ "learning_rate": 7e-05,
105
+ "loss": 1.8924,
106
+ "step": 7500
107
+ },
108
+ {
109
+ "epoch": 0.47,
110
+ "learning_rate": 6.800000000000001e-05,
111
+ "loss": 1.8851,
112
+ "step": 8000
113
+ },
114
+ {
115
+ "epoch": 0.5,
116
+ "learning_rate": 6.6e-05,
117
+ "loss": 1.8811,
118
+ "step": 8500
119
+ },
120
+ {
121
+ "epoch": 0.53,
122
+ "learning_rate": 6.400000000000001e-05,
123
+ "loss": 1.8569,
124
+ "step": 9000
125
+ },
126
+ {
127
+ "epoch": 0.56,
128
+ "learning_rate": 6.2e-05,
129
+ "loss": 1.8477,
130
+ "step": 9500
131
+ },
132
+ {
133
+ "epoch": 0.58,
134
+ "learning_rate": 6e-05,
135
+ "loss": 1.854,
136
+ "step": 10000
137
+ },
138
+ {
139
+ "epoch": 0.58,
140
+ "eval_loss": 1.8423771858215332,
141
+ "eval_runtime": 487.9063,
142
+ "eval_samples_per_second": 13.638,
143
+ "eval_steps_per_second": 3.41,
144
+ "step": 10000
145
+ },
146
+ {
147
+ "epoch": 0.61,
148
+ "learning_rate": 5.8e-05,
149
+ "loss": 1.8315,
150
+ "step": 10500
151
+ },
152
+ {
153
+ "epoch": 0.64,
154
+ "learning_rate": 5.6000000000000006e-05,
155
+ "loss": 1.825,
156
+ "step": 11000
157
+ },
158
+ {
159
+ "epoch": 0.67,
160
+ "learning_rate": 5.4000000000000005e-05,
161
+ "loss": 1.8114,
162
+ "step": 11500
163
+ },
164
+ {
165
+ "epoch": 0.7,
166
+ "learning_rate": 5.2000000000000004e-05,
167
+ "loss": 1.8065,
168
+ "step": 12000
169
+ },
170
+ {
171
+ "epoch": 0.73,
172
+ "learning_rate": 5e-05,
173
+ "loss": 1.8002,
174
+ "step": 12500
175
+ },
176
+ {
177
+ "epoch": 0.76,
178
+ "learning_rate": 4.8e-05,
179
+ "loss": 1.7874,
180
+ "step": 13000
181
+ },
182
+ {
183
+ "epoch": 0.79,
184
+ "learning_rate": 4.600000000000001e-05,
185
+ "loss": 1.778,
186
+ "step": 13500
187
+ },
188
+ {
189
+ "epoch": 0.82,
190
+ "learning_rate": 4.4000000000000006e-05,
191
+ "loss": 1.7802,
192
+ "step": 14000
193
+ },
194
+ {
195
+ "epoch": 0.85,
196
+ "learning_rate": 4.2e-05,
197
+ "loss": 1.7596,
198
+ "step": 14500
199
+ },
200
+ {
201
+ "epoch": 0.88,
202
+ "learning_rate": 4e-05,
203
+ "loss": 1.7594,
204
+ "step": 15000
205
+ },
206
+ {
207
+ "epoch": 0.88,
208
+ "eval_loss": 1.7678033113479614,
209
+ "eval_runtime": 487.9075,
210
+ "eval_samples_per_second": 13.638,
211
+ "eval_steps_per_second": 3.41,
212
+ "step": 15000
213
+ },
214
+ {
215
+ "epoch": 0.91,
216
+ "learning_rate": 3.8e-05,
217
+ "loss": 1.7445,
218
+ "step": 15500
219
+ },
220
+ {
221
+ "epoch": 0.94,
222
+ "learning_rate": 3.6e-05,
223
+ "loss": 1.7464,
224
+ "step": 16000
225
+ },
226
+ {
227
+ "epoch": 0.96,
228
+ "learning_rate": 3.4000000000000007e-05,
229
+ "loss": 1.7426,
230
+ "step": 16500
231
+ },
232
+ {
233
+ "epoch": 0.99,
234
+ "learning_rate": 3.2000000000000005e-05,
235
+ "loss": 1.7352,
236
+ "step": 17000
237
+ },
238
+ {
239
+ "epoch": 1.02,
240
+ "learning_rate": 3e-05,
241
+ "loss": 1.6449,
242
+ "step": 17500
243
+ },
244
+ {
245
+ "epoch": 1.05,
246
+ "learning_rate": 2.8000000000000003e-05,
247
+ "loss": 1.6259,
248
+ "step": 18000
249
+ },
250
+ {
251
+ "epoch": 1.08,
252
+ "learning_rate": 2.6000000000000002e-05,
253
+ "loss": 1.6189,
254
+ "step": 18500
255
+ },
256
+ {
257
+ "epoch": 1.11,
258
+ "learning_rate": 2.4e-05,
259
+ "loss": 1.6132,
260
+ "step": 19000
261
+ },
262
+ {
263
+ "epoch": 1.14,
264
+ "learning_rate": 2.2000000000000003e-05,
265
+ "loss": 1.6124,
266
+ "step": 19500
267
+ },
268
+ {
269
+ "epoch": 1.17,
270
+ "learning_rate": 2e-05,
271
+ "loss": 1.6194,
272
+ "step": 20000
273
+ },
274
+ {
275
+ "epoch": 1.17,
276
+ "eval_loss": 1.722885251045227,
277
+ "eval_runtime": 487.7818,
278
+ "eval_samples_per_second": 13.641,
279
+ "eval_steps_per_second": 3.411,
280
+ "step": 20000
281
+ },
282
+ {
283
+ "epoch": 1.2,
284
+ "learning_rate": 1.8e-05,
285
+ "loss": 1.5974,
286
+ "step": 20500
287
+ },
288
+ {
289
+ "epoch": 1.23,
290
+ "learning_rate": 1.6000000000000003e-05,
291
+ "loss": 1.6006,
292
+ "step": 21000
293
+ },
294
+ {
295
+ "epoch": 1.26,
296
+ "learning_rate": 1.4000000000000001e-05,
297
+ "loss": 1.591,
298
+ "step": 21500
299
+ },
300
+ {
301
+ "epoch": 1.29,
302
+ "learning_rate": 1.2e-05,
303
+ "loss": 1.5985,
304
+ "step": 22000
305
+ },
306
+ {
307
+ "epoch": 1.32,
308
+ "learning_rate": 1e-05,
309
+ "loss": 1.5898,
310
+ "step": 22500
311
+ },
312
+ {
313
+ "epoch": 1.35,
314
+ "learning_rate": 8.000000000000001e-06,
315
+ "loss": 1.5857,
316
+ "step": 23000
317
+ },
318
+ {
319
+ "epoch": 1.37,
320
+ "learning_rate": 6e-06,
321
+ "loss": 1.5802,
322
+ "step": 23500
323
+ },
324
+ {
325
+ "epoch": 1.4,
326
+ "learning_rate": 4.000000000000001e-06,
327
+ "loss": 1.5797,
328
+ "step": 24000
329
+ },
330
+ {
331
+ "epoch": 1.43,
332
+ "learning_rate": 2.0000000000000003e-06,
333
+ "loss": 1.5817,
334
+ "step": 24500
335
+ },
336
+ {
337
+ "epoch": 1.46,
338
+ "learning_rate": 0.0,
339
+ "loss": 1.5837,
340
+ "step": 25000
341
+ },
342
+ {
343
+ "epoch": 1.46,
344
+ "eval_loss": 1.6942951679229736,
345
+ "eval_runtime": 488.0731,
346
+ "eval_samples_per_second": 13.633,
347
+ "eval_steps_per_second": 3.409,
348
+ "step": 25000
349
+ },
350
+ {
351
+ "epoch": 1.46,
352
+ "step": 25000,
353
+ "total_flos": 3.7147907956565606e+17,
354
+ "train_loss": 1.8233512182617186,
355
+ "train_runtime": 45032.5538,
356
+ "train_samples_per_second": 4.441,
357
+ "train_steps_per_second": 0.555
358
+ }
359
+ ],
360
+ "max_steps": 25000,
361
+ "num_train_epochs": 2,
362
+ "total_flos": 3.7147907956565606e+17,
363
+ "trial_name": null,
364
+ "trial_params": null
365
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b871752f3c441f14587f9f603bdfd761fdee1a838b97feeccda6721e152361a5
3
+ size 3375