fpadovani commited on
Commit
ec9bb3e
·
verified ·
1 Parent(s): 684f5d8

Training in progress, epoch 20, checkpoint

Browse files
checkpoint-2180/config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 1,
8
+ "dtype": "float32",
9
+ "embd_pdrop": 0.1,
10
+ "eos_token_id": 2,
11
+ "initializer_range": 0.02,
12
+ "layer_norm_epsilon": 1e-05,
13
+ "model_type": "gpt2",
14
+ "n_ctx": 1024,
15
+ "n_embd": 768,
16
+ "n_head": 12,
17
+ "n_inner": null,
18
+ "n_layer": 12,
19
+ "n_positions": 1024,
20
+ "pad_token_id": 2,
21
+ "reorder_and_upcast_attn": false,
22
+ "resid_pdrop": 0.1,
23
+ "scale_attn_by_inverse_layer_idx": false,
24
+ "scale_attn_weights": true,
25
+ "summary_activation": null,
26
+ "summary_first_dropout": 0.1,
27
+ "summary_proj_to_labels": true,
28
+ "summary_type": "cls_index",
29
+ "summary_use_proj": true,
30
+ "transformers_version": "4.56.1",
31
+ "use_cache": true,
32
+ "vocab_size": 16000
33
+ }
checkpoint-2180/generation_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": [
5
+ 2
6
+ ],
7
+ "pad_token_id": 2,
8
+ "transformers_version": "4.56.1"
9
+ }
checkpoint-2180/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5337b323bfee7d38e768be6914712dfb881f8f6c7ce6d3d6e500a2b17fc49ca2
3
+ size 392536704
checkpoint-2180/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d62e2c6b71f78b2a8a66a94f45487eef4d78d90373a490985c615d77f6da3df
3
+ size 785167627
checkpoint-2180/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c08212eae59325eddcc905431fc414adc7c91653307d79bb0397b15a71e93829
3
+ size 14709
checkpoint-2180/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c1fb8508a33ad3c545138e43ff8777529743f676b9430f3cfde3914550cec0d
3
+ size 1383
checkpoint-2180/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:591ec2b1df2ef6511d604f6ad15d412b7b2d00d57793d236121743b187b3f29b
3
+ size 1465
checkpoint-2180/special_tokens_map.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "eos_token": "</s>",
4
+ "mask_token": "<mask>",
5
+ "pad_token": "</s>",
6
+ "sep_token": "<sep>",
7
+ "unk_token": "<unk>"
8
+ }
checkpoint-2180/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-2180/tokenizer_config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<pad>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<s>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "<mask>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "5": {
44
+ "content": "<sep>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ }
51
+ },
52
+ "bos_token": "<s>",
53
+ "clean_up_tokenization_spaces": false,
54
+ "eos_token": "</s>",
55
+ "extra_special_tokens": {},
56
+ "mask_token": "<mask>",
57
+ "model_max_length": 1000000000000000019884624838656,
58
+ "pad_token": "</s>",
59
+ "sep_token": "<sep>",
60
+ "tokenizer_class": "PreTrainedTokenizerFast",
61
+ "unk_token": "<unk>"
62
+ }
checkpoint-2180/trainer_state.json ADDED
@@ -0,0 +1,341 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 1635,
3
+ "best_metric": 3.888904333114624,
4
+ "best_model_checkpoint": "/home/p318482/syntactic-bootstrapping/model_trained/ads_o_fr_13/checkpoint-1635",
5
+ "epoch": 20.0,
6
+ "eval_steps": 500,
7
+ "global_step": 2180,
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.009174311926605505,
14
+ "grad_norm": 17.951679229736328,
15
+ "learning_rate": 0.0,
16
+ "loss": 9.7117,
17
+ "step": 1
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "grad_norm": 2.537048101425171,
22
+ "learning_rate": 2.16e-05,
23
+ "loss": 7.4816,
24
+ "step": 109
25
+ },
26
+ {
27
+ "epoch": 1.0,
28
+ "eval_loss": 6.125095367431641,
29
+ "eval_runtime": 3.3374,
30
+ "eval_samples_per_second": 810.812,
31
+ "eval_steps_per_second": 3.296,
32
+ "step": 109
33
+ },
34
+ {
35
+ "epoch": 2.0,
36
+ "grad_norm": 1.5311152935028076,
37
+ "learning_rate": 4.3400000000000005e-05,
38
+ "loss": 5.2479,
39
+ "step": 218
40
+ },
41
+ {
42
+ "epoch": 2.0,
43
+ "eval_loss": 4.817795753479004,
44
+ "eval_runtime": 3.4684,
45
+ "eval_samples_per_second": 780.181,
46
+ "eval_steps_per_second": 3.171,
47
+ "step": 218
48
+ },
49
+ {
50
+ "epoch": 3.0,
51
+ "grad_norm": 1.4448567628860474,
52
+ "learning_rate": 6.52e-05,
53
+ "loss": 4.5633,
54
+ "step": 327
55
+ },
56
+ {
57
+ "epoch": 3.0,
58
+ "eval_loss": 4.496043682098389,
59
+ "eval_runtime": 3.3595,
60
+ "eval_samples_per_second": 805.47,
61
+ "eval_steps_per_second": 3.274,
62
+ "step": 327
63
+ },
64
+ {
65
+ "epoch": 4.0,
66
+ "grad_norm": 1.4543994665145874,
67
+ "learning_rate": 8.7e-05,
68
+ "loss": 4.3166,
69
+ "step": 436
70
+ },
71
+ {
72
+ "epoch": 4.0,
73
+ "eval_loss": 4.303887844085693,
74
+ "eval_runtime": 3.3467,
75
+ "eval_samples_per_second": 808.547,
76
+ "eval_steps_per_second": 3.287,
77
+ "step": 436
78
+ },
79
+ {
80
+ "epoch": 5.0,
81
+ "grad_norm": 1.3640012741088867,
82
+ "learning_rate": 9.734939759036145e-05,
83
+ "loss": 4.1561,
84
+ "step": 545
85
+ },
86
+ {
87
+ "epoch": 5.0,
88
+ "eval_loss": 4.18064546585083,
89
+ "eval_runtime": 3.442,
90
+ "eval_samples_per_second": 786.172,
91
+ "eval_steps_per_second": 3.196,
92
+ "step": 545
93
+ },
94
+ {
95
+ "epoch": 6.0,
96
+ "grad_norm": 1.4212088584899902,
97
+ "learning_rate": 9.078313253012048e-05,
98
+ "loss": 4.0364,
99
+ "step": 654
100
+ },
101
+ {
102
+ "epoch": 6.0,
103
+ "eval_loss": 4.096184253692627,
104
+ "eval_runtime": 3.3717,
105
+ "eval_samples_per_second": 802.573,
106
+ "eval_steps_per_second": 3.262,
107
+ "step": 654
108
+ },
109
+ {
110
+ "epoch": 7.0,
111
+ "grad_norm": 1.3861104249954224,
112
+ "learning_rate": 8.421686746987952e-05,
113
+ "loss": 3.9474,
114
+ "step": 763
115
+ },
116
+ {
117
+ "epoch": 7.0,
118
+ "eval_loss": 4.03432559967041,
119
+ "eval_runtime": 3.3677,
120
+ "eval_samples_per_second": 803.523,
121
+ "eval_steps_per_second": 3.266,
122
+ "step": 763
123
+ },
124
+ {
125
+ "epoch": 8.0,
126
+ "grad_norm": 1.4673441648483276,
127
+ "learning_rate": 7.765060240963856e-05,
128
+ "loss": 3.8753,
129
+ "step": 872
130
+ },
131
+ {
132
+ "epoch": 8.0,
133
+ "eval_loss": 3.990006685256958,
134
+ "eval_runtime": 3.3655,
135
+ "eval_samples_per_second": 804.045,
136
+ "eval_steps_per_second": 3.268,
137
+ "step": 872
138
+ },
139
+ {
140
+ "epoch": 9.0,
141
+ "grad_norm": 1.5565500259399414,
142
+ "learning_rate": 7.108433734939759e-05,
143
+ "loss": 3.8118,
144
+ "step": 981
145
+ },
146
+ {
147
+ "epoch": 9.0,
148
+ "eval_loss": 3.9578440189361572,
149
+ "eval_runtime": 3.351,
150
+ "eval_samples_per_second": 807.532,
151
+ "eval_steps_per_second": 3.283,
152
+ "step": 981
153
+ },
154
+ {
155
+ "epoch": 10.0,
156
+ "grad_norm": 1.5941141843795776,
157
+ "learning_rate": 6.451807228915663e-05,
158
+ "loss": 3.7548,
159
+ "step": 1090
160
+ },
161
+ {
162
+ "epoch": 10.0,
163
+ "eval_loss": 3.9341025352478027,
164
+ "eval_runtime": 3.5715,
165
+ "eval_samples_per_second": 757.67,
166
+ "eval_steps_per_second": 3.08,
167
+ "step": 1090
168
+ },
169
+ {
170
+ "epoch": 11.0,
171
+ "grad_norm": 1.778639793395996,
172
+ "learning_rate": 5.7951807228915664e-05,
173
+ "loss": 3.7009,
174
+ "step": 1199
175
+ },
176
+ {
177
+ "epoch": 11.0,
178
+ "eval_loss": 3.916402578353882,
179
+ "eval_runtime": 3.3516,
180
+ "eval_samples_per_second": 807.368,
181
+ "eval_steps_per_second": 3.282,
182
+ "step": 1199
183
+ },
184
+ {
185
+ "epoch": 12.0,
186
+ "grad_norm": 1.8095911741256714,
187
+ "learning_rate": 5.1385542168674705e-05,
188
+ "loss": 3.6487,
189
+ "step": 1308
190
+ },
191
+ {
192
+ "epoch": 12.0,
193
+ "eval_loss": 3.9003710746765137,
194
+ "eval_runtime": 3.6596,
195
+ "eval_samples_per_second": 739.417,
196
+ "eval_steps_per_second": 3.006,
197
+ "step": 1308
198
+ },
199
+ {
200
+ "epoch": 13.0,
201
+ "grad_norm": 2.016235113143921,
202
+ "learning_rate": 4.481927710843374e-05,
203
+ "loss": 3.6004,
204
+ "step": 1417
205
+ },
206
+ {
207
+ "epoch": 13.0,
208
+ "eval_loss": 3.892484188079834,
209
+ "eval_runtime": 3.3724,
210
+ "eval_samples_per_second": 802.396,
211
+ "eval_steps_per_second": 3.262,
212
+ "step": 1417
213
+ },
214
+ {
215
+ "epoch": 14.0,
216
+ "grad_norm": 2.106367826461792,
217
+ "learning_rate": 3.8253012048192774e-05,
218
+ "loss": 3.5523,
219
+ "step": 1526
220
+ },
221
+ {
222
+ "epoch": 14.0,
223
+ "eval_loss": 3.894266366958618,
224
+ "eval_runtime": 3.5059,
225
+ "eval_samples_per_second": 771.833,
226
+ "eval_steps_per_second": 3.138,
227
+ "step": 1526
228
+ },
229
+ {
230
+ "epoch": 15.0,
231
+ "grad_norm": 2.2525172233581543,
232
+ "learning_rate": 3.168674698795181e-05,
233
+ "loss": 3.5049,
234
+ "step": 1635
235
+ },
236
+ {
237
+ "epoch": 15.0,
238
+ "eval_loss": 3.888904333114624,
239
+ "eval_runtime": 3.51,
240
+ "eval_samples_per_second": 770.943,
241
+ "eval_steps_per_second": 3.134,
242
+ "step": 1635
243
+ },
244
+ {
245
+ "epoch": 16.0,
246
+ "grad_norm": 2.453017473220825,
247
+ "learning_rate": 2.5120481927710842e-05,
248
+ "loss": 3.4614,
249
+ "step": 1744
250
+ },
251
+ {
252
+ "epoch": 16.0,
253
+ "eval_loss": 3.8933465480804443,
254
+ "eval_runtime": 3.3572,
255
+ "eval_samples_per_second": 806.026,
256
+ "eval_steps_per_second": 3.277,
257
+ "step": 1744
258
+ },
259
+ {
260
+ "epoch": 17.0,
261
+ "grad_norm": 2.57418155670166,
262
+ "learning_rate": 1.855421686746988e-05,
263
+ "loss": 3.4222,
264
+ "step": 1853
265
+ },
266
+ {
267
+ "epoch": 17.0,
268
+ "eval_loss": 3.897146701812744,
269
+ "eval_runtime": 3.6657,
270
+ "eval_samples_per_second": 738.188,
271
+ "eval_steps_per_second": 3.001,
272
+ "step": 1853
273
+ },
274
+ {
275
+ "epoch": 18.0,
276
+ "grad_norm": 2.6725003719329834,
277
+ "learning_rate": 1.1987951807228916e-05,
278
+ "loss": 3.3888,
279
+ "step": 1962
280
+ },
281
+ {
282
+ "epoch": 18.0,
283
+ "eval_loss": 3.903165102005005,
284
+ "eval_runtime": 3.4414,
285
+ "eval_samples_per_second": 786.298,
286
+ "eval_steps_per_second": 3.196,
287
+ "step": 1962
288
+ },
289
+ {
290
+ "epoch": 19.0,
291
+ "grad_norm": 2.7827844619750977,
292
+ "learning_rate": 5.421686746987952e-06,
293
+ "loss": 3.3613,
294
+ "step": 2071
295
+ },
296
+ {
297
+ "epoch": 19.0,
298
+ "eval_loss": 3.9074323177337646,
299
+ "eval_runtime": 3.458,
300
+ "eval_samples_per_second": 782.524,
301
+ "eval_steps_per_second": 3.181,
302
+ "step": 2071
303
+ },
304
+ {
305
+ "epoch": 20.0,
306
+ "grad_norm": 2.883789539337158,
307
+ "learning_rate": 0.0,
308
+ "loss": 3.3411,
309
+ "step": 2180
310
+ },
311
+ {
312
+ "epoch": 20.0,
313
+ "eval_loss": 3.9097225666046143,
314
+ "eval_runtime": 3.3666,
315
+ "eval_samples_per_second": 803.771,
316
+ "eval_steps_per_second": 3.267,
317
+ "step": 2180
318
+ }
319
+ ],
320
+ "logging_steps": 500,
321
+ "max_steps": 2180,
322
+ "num_input_tokens_seen": 0,
323
+ "num_train_epochs": 20,
324
+ "save_steps": 500,
325
+ "stateful_callbacks": {
326
+ "TrainerControl": {
327
+ "args": {
328
+ "should_epoch_stop": false,
329
+ "should_evaluate": false,
330
+ "should_log": false,
331
+ "should_save": true,
332
+ "should_training_stop": true
333
+ },
334
+ "attributes": {}
335
+ }
336
+ },
337
+ "total_flos": 3.618633351168e+16,
338
+ "train_batch_size": 256,
339
+ "trial_name": null,
340
+ "trial_params": null
341
+ }
checkpoint-2180/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07d93deb681371270a2b11ffb08b575a730e512a2a8d3e5c01548c44e1aa4ff7
3
+ size 5905