fpadovani commited on
Commit
ea495e3
·
verified ·
1 Parent(s): 87933b9

Training in progress, step 7500, checkpoint

Browse files
checkpoint-7500/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": 30000
33
+ }
checkpoint-7500/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-7500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74150a13c4781721d9c70858ec71673cbc6282e9819740e7e3685261c7576f93
3
+ size 435544704
checkpoint-7500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81fc8388b723378e23b44c3abcf9f8f7c0a0e68734ddcaa9a7862df5509d1c79
3
+ size 871183627
checkpoint-7500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00f94b51b46c77de2ae03635fa3f0bc465e34461ff844560c09c4cc171e9f3a4
3
+ size 14645
checkpoint-7500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:642827d515e7c3d88d20c04224822701abbbc9250deb1d366045f974d08b8258
3
+ size 1383
checkpoint-7500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca9b8a5501507d40ffae7081184eb5717a9cb61e1a824d7d0f7f2087efd5ab69
3
+ size 1465
checkpoint-7500/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-7500/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-7500/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-7500/trainer_state.json ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 5200,
3
+ "best_metric": 5.3865814208984375,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 18.75,
6
+ "eval_steps": 500,
7
+ "global_step": 7500,
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.0025,
14
+ "grad_norm": 15.434585571289062,
15
+ "learning_rate": 0.0,
16
+ "loss": 10.3861,
17
+ "step": 1
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "grad_norm": 1.4324626922607422,
22
+ "learning_rate": 7.98e-05,
23
+ "loss": 7.0655,
24
+ "step": 400
25
+ },
26
+ {
27
+ "epoch": 1.0,
28
+ "eval_loss": 6.195863723754883,
29
+ "eval_runtime": 15.8913,
30
+ "eval_samples_per_second": 1616.543,
31
+ "eval_steps_per_second": 6.356,
32
+ "step": 400
33
+ },
34
+ {
35
+ "epoch": 2.0,
36
+ "grad_norm": 1.2896528244018555,
37
+ "learning_rate": 9.600267379679145e-05,
38
+ "loss": 5.961,
39
+ "step": 800
40
+ },
41
+ {
42
+ "epoch": 2.0,
43
+ "eval_loss": 5.855167865753174,
44
+ "eval_runtime": 15.8141,
45
+ "eval_samples_per_second": 1624.433,
46
+ "eval_steps_per_second": 6.387,
47
+ "step": 800
48
+ },
49
+ {
50
+ "epoch": 3.0,
51
+ "grad_norm": 1.3000872135162354,
52
+ "learning_rate": 9.065508021390375e-05,
53
+ "loss": 5.7374,
54
+ "step": 1200
55
+ },
56
+ {
57
+ "epoch": 3.0,
58
+ "eval_loss": 5.720485210418701,
59
+ "eval_runtime": 15.9219,
60
+ "eval_samples_per_second": 1613.439,
61
+ "eval_steps_per_second": 6.343,
62
+ "step": 1200
63
+ },
64
+ {
65
+ "epoch": 4.0,
66
+ "grad_norm": 1.3779990673065186,
67
+ "learning_rate": 8.530748663101604e-05,
68
+ "loss": 5.6049,
69
+ "step": 1600
70
+ },
71
+ {
72
+ "epoch": 4.0,
73
+ "eval_loss": 5.622853755950928,
74
+ "eval_runtime": 15.9133,
75
+ "eval_samples_per_second": 1614.309,
76
+ "eval_steps_per_second": 6.347,
77
+ "step": 1600
78
+ },
79
+ {
80
+ "epoch": 5.0,
81
+ "grad_norm": 1.3654310703277588,
82
+ "learning_rate": 7.995989304812834e-05,
83
+ "loss": 5.498,
84
+ "step": 2000
85
+ },
86
+ {
87
+ "epoch": 5.0,
88
+ "eval_loss": 5.553117752075195,
89
+ "eval_runtime": 15.8859,
90
+ "eval_samples_per_second": 1617.098,
91
+ "eval_steps_per_second": 6.358,
92
+ "step": 2000
93
+ },
94
+ {
95
+ "epoch": 6.0,
96
+ "grad_norm": 1.5499093532562256,
97
+ "learning_rate": 7.461229946524065e-05,
98
+ "loss": 5.4056,
99
+ "step": 2400
100
+ },
101
+ {
102
+ "epoch": 6.0,
103
+ "eval_loss": 5.501069068908691,
104
+ "eval_runtime": 16.1578,
105
+ "eval_samples_per_second": 1589.881,
106
+ "eval_steps_per_second": 6.251,
107
+ "step": 2400
108
+ },
109
+ {
110
+ "epoch": 7.0,
111
+ "grad_norm": 1.5300403833389282,
112
+ "learning_rate": 6.926470588235294e-05,
113
+ "loss": 5.3232,
114
+ "step": 2800
115
+ },
116
+ {
117
+ "epoch": 7.0,
118
+ "eval_loss": 5.458910942077637,
119
+ "eval_runtime": 15.935,
120
+ "eval_samples_per_second": 1612.116,
121
+ "eval_steps_per_second": 6.338,
122
+ "step": 2800
123
+ },
124
+ {
125
+ "epoch": 8.0,
126
+ "grad_norm": 1.7089910507202148,
127
+ "learning_rate": 6.391711229946524e-05,
128
+ "loss": 5.2493,
129
+ "step": 3200
130
+ },
131
+ {
132
+ "epoch": 8.0,
133
+ "eval_loss": 5.4299397468566895,
134
+ "eval_runtime": 15.9035,
135
+ "eval_samples_per_second": 1615.306,
136
+ "eval_steps_per_second": 6.351,
137
+ "step": 3200
138
+ },
139
+ {
140
+ "epoch": 9.0,
141
+ "grad_norm": 1.7956774234771729,
142
+ "learning_rate": 5.8569518716577545e-05,
143
+ "loss": 5.1807,
144
+ "step": 3600
145
+ },
146
+ {
147
+ "epoch": 9.0,
148
+ "eval_loss": 5.408169746398926,
149
+ "eval_runtime": 15.9504,
150
+ "eval_samples_per_second": 1610.556,
151
+ "eval_steps_per_second": 6.332,
152
+ "step": 3600
153
+ },
154
+ {
155
+ "epoch": 10.0,
156
+ "grad_norm": 1.8985629081726074,
157
+ "learning_rate": 5.322192513368984e-05,
158
+ "loss": 5.1166,
159
+ "step": 4000
160
+ },
161
+ {
162
+ "epoch": 10.0,
163
+ "eval_loss": 5.396389484405518,
164
+ "eval_runtime": 15.8668,
165
+ "eval_samples_per_second": 1619.042,
166
+ "eval_steps_per_second": 6.365,
167
+ "step": 4000
168
+ },
169
+ {
170
+ "epoch": 11.0,
171
+ "grad_norm": 2.1109752655029297,
172
+ "learning_rate": 4.787433155080214e-05,
173
+ "loss": 5.0565,
174
+ "step": 4400
175
+ },
176
+ {
177
+ "epoch": 11.0,
178
+ "eval_loss": 5.391461372375488,
179
+ "eval_runtime": 15.9444,
180
+ "eval_samples_per_second": 1611.158,
181
+ "eval_steps_per_second": 6.334,
182
+ "step": 4400
183
+ },
184
+ {
185
+ "epoch": 12.0,
186
+ "grad_norm": 2.173975706100464,
187
+ "learning_rate": 4.2526737967914444e-05,
188
+ "loss": 4.9988,
189
+ "step": 4800
190
+ },
191
+ {
192
+ "epoch": 12.0,
193
+ "eval_loss": 5.386951446533203,
194
+ "eval_runtime": 15.9572,
195
+ "eval_samples_per_second": 1609.873,
196
+ "eval_steps_per_second": 6.329,
197
+ "step": 4800
198
+ },
199
+ {
200
+ "epoch": 13.0,
201
+ "grad_norm": 2.42020320892334,
202
+ "learning_rate": 3.717914438502674e-05,
203
+ "loss": 4.9451,
204
+ "step": 5200
205
+ },
206
+ {
207
+ "epoch": 13.0,
208
+ "eval_loss": 5.3865814208984375,
209
+ "eval_runtime": 15.9144,
210
+ "eval_samples_per_second": 1614.195,
211
+ "eval_steps_per_second": 6.346,
212
+ "step": 5200
213
+ },
214
+ {
215
+ "epoch": 14.0,
216
+ "grad_norm": 2.445915460586548,
217
+ "learning_rate": 3.1831550802139034e-05,
218
+ "loss": 4.8937,
219
+ "step": 5600
220
+ },
221
+ {
222
+ "epoch": 14.0,
223
+ "eval_loss": 5.3876051902771,
224
+ "eval_runtime": 15.9181,
225
+ "eval_samples_per_second": 1613.821,
226
+ "eval_steps_per_second": 6.345,
227
+ "step": 5600
228
+ },
229
+ {
230
+ "epoch": 15.0,
231
+ "grad_norm": 2.670419931411743,
232
+ "learning_rate": 2.6483957219251336e-05,
233
+ "loss": 4.8481,
234
+ "step": 6000
235
+ },
236
+ {
237
+ "epoch": 15.0,
238
+ "eval_loss": 5.392692565917969,
239
+ "eval_runtime": 15.8838,
240
+ "eval_samples_per_second": 1617.305,
241
+ "eval_steps_per_second": 6.359,
242
+ "step": 6000
243
+ },
244
+ {
245
+ "epoch": 16.0,
246
+ "grad_norm": 2.802830696105957,
247
+ "learning_rate": 2.1136363636363638e-05,
248
+ "loss": 4.8086,
249
+ "step": 6400
250
+ },
251
+ {
252
+ "epoch": 16.0,
253
+ "eval_loss": 5.402634143829346,
254
+ "eval_runtime": 15.9917,
255
+ "eval_samples_per_second": 1606.394,
256
+ "eval_steps_per_second": 6.316,
257
+ "step": 6400
258
+ },
259
+ {
260
+ "epoch": 17.0,
261
+ "grad_norm": 2.862973213195801,
262
+ "learning_rate": 1.5788770053475936e-05,
263
+ "loss": 4.7739,
264
+ "step": 6800
265
+ },
266
+ {
267
+ "epoch": 17.0,
268
+ "eval_loss": 5.406018257141113,
269
+ "eval_runtime": 16.0183,
270
+ "eval_samples_per_second": 1603.731,
271
+ "eval_steps_per_second": 6.305,
272
+ "step": 6800
273
+ },
274
+ {
275
+ "epoch": 18.0,
276
+ "grad_norm": 3.028500556945801,
277
+ "learning_rate": 1.0441176470588236e-05,
278
+ "loss": 4.7446,
279
+ "step": 7200
280
+ },
281
+ {
282
+ "epoch": 18.0,
283
+ "eval_loss": 5.41143274307251,
284
+ "eval_runtime": 15.9098,
285
+ "eval_samples_per_second": 1614.669,
286
+ "eval_steps_per_second": 6.348,
287
+ "step": 7200
288
+ }
289
+ ],
290
+ "logging_steps": 500,
291
+ "max_steps": 8000,
292
+ "num_input_tokens_seen": 0,
293
+ "num_train_epochs": 20,
294
+ "save_steps": 500,
295
+ "stateful_callbacks": {
296
+ "TrainerControl": {
297
+ "args": {
298
+ "should_epoch_stop": false,
299
+ "should_evaluate": false,
300
+ "should_log": false,
301
+ "should_save": true,
302
+ "should_training_stop": false
303
+ },
304
+ "attributes": {}
305
+ }
306
+ },
307
+ "total_flos": 1.25169726947328e+17,
308
+ "train_batch_size": 256,
309
+ "trial_name": null,
310
+ "trial_params": null
311
+ }
checkpoint-7500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8dac463600d9699a5f5710b48ecc0e815d9a4d7122a6e0126282ae7df64e105a
3
+ size 5905