fpadovani commited on
Commit
e509dc4
·
verified ·
1 Parent(s): dcd85cd

Training in progress, step 7000, checkpoint

Browse files
checkpoint-7000/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-7000/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-7000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c2fcd8af7f1efd72b971298edd4d96d8b759a5cd4f2e82c4e49ded1a912e4f4
3
+ size 435544704
checkpoint-7000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d64dfa3907176a6aa1cef5fd07e4ab2cba32d4b64f90fdf12b29b692947473b2
3
+ size 871183627
checkpoint-7000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e714ffd08532ee9954cc56815add8e3918280dbf292c5ce1bd42585a4a1e1776
3
+ size 14645
checkpoint-7000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be40a03f70bb2a35bb567114fb5ceb05ce1222c8df24b1e1100746c617c5be10
3
+ size 1383
checkpoint-7000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:deb1edafa26b4caaefb6cf165cf402eced9c3b27b58fd3375723ad698e99026a
3
+ size 1465
checkpoint-7000/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-7000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-7000/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-7000/trainer_state.json ADDED
@@ -0,0 +1,296 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 5865,
3
+ "best_metric": 4.335602283477783,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 17.902813299232736,
6
+ "eval_steps": 500,
7
+ "global_step": 7000,
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.0025575447570332483,
14
+ "grad_norm": 15.849249839782715,
15
+ "learning_rate": 0.0,
16
+ "loss": 10.3746,
17
+ "step": 1
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "grad_norm": 1.031598448753357,
22
+ "learning_rate": 7.800000000000001e-05,
23
+ "loss": 6.8365,
24
+ "step": 391
25
+ },
26
+ {
27
+ "epoch": 1.0,
28
+ "eval_loss": 5.74590539932251,
29
+ "eval_runtime": 19.393,
30
+ "eval_samples_per_second": 1296.345,
31
+ "eval_steps_per_second": 5.105,
32
+ "step": 391
33
+ },
34
+ {
35
+ "epoch": 2.0,
36
+ "grad_norm": 0.9619669914245605,
37
+ "learning_rate": 9.615068493150685e-05,
38
+ "loss": 5.3979,
39
+ "step": 782
40
+ },
41
+ {
42
+ "epoch": 2.0,
43
+ "eval_loss": 5.181668758392334,
44
+ "eval_runtime": 19.3113,
45
+ "eval_samples_per_second": 1301.828,
46
+ "eval_steps_per_second": 5.127,
47
+ "step": 782
48
+ },
49
+ {
50
+ "epoch": 3.0,
51
+ "grad_norm": 0.9032778739929199,
52
+ "learning_rate": 9.07945205479452e-05,
53
+ "loss": 5.0061,
54
+ "step": 1173
55
+ },
56
+ {
57
+ "epoch": 3.0,
58
+ "eval_loss": 4.930461883544922,
59
+ "eval_runtime": 19.4629,
60
+ "eval_samples_per_second": 1291.685,
61
+ "eval_steps_per_second": 5.087,
62
+ "step": 1173
63
+ },
64
+ {
65
+ "epoch": 4.0,
66
+ "grad_norm": 0.9254869222640991,
67
+ "learning_rate": 8.543835616438357e-05,
68
+ "loss": 4.7815,
69
+ "step": 1564
70
+ },
71
+ {
72
+ "epoch": 4.0,
73
+ "eval_loss": 4.774387836456299,
74
+ "eval_runtime": 19.391,
75
+ "eval_samples_per_second": 1296.48,
76
+ "eval_steps_per_second": 5.105,
77
+ "step": 1564
78
+ },
79
+ {
80
+ "epoch": 5.0,
81
+ "grad_norm": 0.9707768559455872,
82
+ "learning_rate": 8.008219178082192e-05,
83
+ "loss": 4.6165,
84
+ "step": 1955
85
+ },
86
+ {
87
+ "epoch": 5.0,
88
+ "eval_loss": 4.658854007720947,
89
+ "eval_runtime": 19.4424,
90
+ "eval_samples_per_second": 1293.051,
91
+ "eval_steps_per_second": 5.092,
92
+ "step": 1955
93
+ },
94
+ {
95
+ "epoch": 6.0,
96
+ "grad_norm": 0.9788696765899658,
97
+ "learning_rate": 7.472602739726028e-05,
98
+ "loss": 4.4815,
99
+ "step": 2346
100
+ },
101
+ {
102
+ "epoch": 6.0,
103
+ "eval_loss": 4.57161283493042,
104
+ "eval_runtime": 19.4724,
105
+ "eval_samples_per_second": 1291.06,
106
+ "eval_steps_per_second": 5.084,
107
+ "step": 2346
108
+ },
109
+ {
110
+ "epoch": 7.0,
111
+ "grad_norm": 1.0454139709472656,
112
+ "learning_rate": 6.936986301369864e-05,
113
+ "loss": 4.3662,
114
+ "step": 2737
115
+ },
116
+ {
117
+ "epoch": 7.0,
118
+ "eval_loss": 4.505728721618652,
119
+ "eval_runtime": 19.3722,
120
+ "eval_samples_per_second": 1297.737,
121
+ "eval_steps_per_second": 5.11,
122
+ "step": 2737
123
+ },
124
+ {
125
+ "epoch": 8.0,
126
+ "grad_norm": 1.0849652290344238,
127
+ "learning_rate": 6.401369863013699e-05,
128
+ "loss": 4.2647,
129
+ "step": 3128
130
+ },
131
+ {
132
+ "epoch": 8.0,
133
+ "eval_loss": 4.456013202667236,
134
+ "eval_runtime": 19.4196,
135
+ "eval_samples_per_second": 1294.567,
136
+ "eval_steps_per_second": 5.098,
137
+ "step": 3128
138
+ },
139
+ {
140
+ "epoch": 9.0,
141
+ "grad_norm": 1.142387866973877,
142
+ "learning_rate": 5.865753424657534e-05,
143
+ "loss": 4.1728,
144
+ "step": 3519
145
+ },
146
+ {
147
+ "epoch": 9.0,
148
+ "eval_loss": 4.4147868156433105,
149
+ "eval_runtime": 19.3974,
150
+ "eval_samples_per_second": 1296.053,
151
+ "eval_steps_per_second": 5.104,
152
+ "step": 3519
153
+ },
154
+ {
155
+ "epoch": 10.0,
156
+ "grad_norm": 1.1628907918930054,
157
+ "learning_rate": 5.330136986301371e-05,
158
+ "loss": 4.0896,
159
+ "step": 3910
160
+ },
161
+ {
162
+ "epoch": 10.0,
163
+ "eval_loss": 4.385285377502441,
164
+ "eval_runtime": 19.4087,
165
+ "eval_samples_per_second": 1295.295,
166
+ "eval_steps_per_second": 5.101,
167
+ "step": 3910
168
+ },
169
+ {
170
+ "epoch": 11.0,
171
+ "grad_norm": 1.2972906827926636,
172
+ "learning_rate": 4.794520547945205e-05,
173
+ "loss": 4.0153,
174
+ "step": 4301
175
+ },
176
+ {
177
+ "epoch": 11.0,
178
+ "eval_loss": 4.366960525512695,
179
+ "eval_runtime": 19.4166,
180
+ "eval_samples_per_second": 1294.767,
181
+ "eval_steps_per_second": 5.099,
182
+ "step": 4301
183
+ },
184
+ {
185
+ "epoch": 12.0,
186
+ "grad_norm": 1.3291264772415161,
187
+ "learning_rate": 4.258904109589041e-05,
188
+ "loss": 3.9487,
189
+ "step": 4692
190
+ },
191
+ {
192
+ "epoch": 12.0,
193
+ "eval_loss": 4.35342264175415,
194
+ "eval_runtime": 19.6756,
195
+ "eval_samples_per_second": 1277.723,
196
+ "eval_steps_per_second": 5.032,
197
+ "step": 4692
198
+ },
199
+ {
200
+ "epoch": 13.0,
201
+ "grad_norm": 1.3737049102783203,
202
+ "learning_rate": 3.723287671232877e-05,
203
+ "loss": 3.8886,
204
+ "step": 5083
205
+ },
206
+ {
207
+ "epoch": 13.0,
208
+ "eval_loss": 4.344508647918701,
209
+ "eval_runtime": 19.3839,
210
+ "eval_samples_per_second": 1296.954,
211
+ "eval_steps_per_second": 5.107,
212
+ "step": 5083
213
+ },
214
+ {
215
+ "epoch": 14.0,
216
+ "grad_norm": 1.4470322132110596,
217
+ "learning_rate": 3.187671232876712e-05,
218
+ "loss": 3.8345,
219
+ "step": 5474
220
+ },
221
+ {
222
+ "epoch": 14.0,
223
+ "eval_loss": 4.340525150299072,
224
+ "eval_runtime": 19.4296,
225
+ "eval_samples_per_second": 1293.9,
226
+ "eval_steps_per_second": 5.095,
227
+ "step": 5474
228
+ },
229
+ {
230
+ "epoch": 15.0,
231
+ "grad_norm": 1.5098106861114502,
232
+ "learning_rate": 2.6520547945205482e-05,
233
+ "loss": 3.786,
234
+ "step": 5865
235
+ },
236
+ {
237
+ "epoch": 15.0,
238
+ "eval_loss": 4.335602283477783,
239
+ "eval_runtime": 19.3809,
240
+ "eval_samples_per_second": 1297.15,
241
+ "eval_steps_per_second": 5.108,
242
+ "step": 5865
243
+ },
244
+ {
245
+ "epoch": 16.0,
246
+ "grad_norm": 1.5827183723449707,
247
+ "learning_rate": 2.1164383561643834e-05,
248
+ "loss": 3.7426,
249
+ "step": 6256
250
+ },
251
+ {
252
+ "epoch": 16.0,
253
+ "eval_loss": 4.336056232452393,
254
+ "eval_runtime": 19.5099,
255
+ "eval_samples_per_second": 1288.579,
256
+ "eval_steps_per_second": 5.074,
257
+ "step": 6256
258
+ },
259
+ {
260
+ "epoch": 17.0,
261
+ "grad_norm": 1.6074565649032593,
262
+ "learning_rate": 1.5808219178082193e-05,
263
+ "loss": 3.7044,
264
+ "step": 6647
265
+ },
266
+ {
267
+ "epoch": 17.0,
268
+ "eval_loss": 4.337679386138916,
269
+ "eval_runtime": 19.4478,
270
+ "eval_samples_per_second": 1292.693,
271
+ "eval_steps_per_second": 5.091,
272
+ "step": 6647
273
+ }
274
+ ],
275
+ "logging_steps": 500,
276
+ "max_steps": 7820,
277
+ "num_input_tokens_seen": 0,
278
+ "num_train_epochs": 20,
279
+ "save_steps": 500,
280
+ "stateful_callbacks": {
281
+ "TrainerControl": {
282
+ "args": {
283
+ "should_epoch_stop": false,
284
+ "should_evaluate": false,
285
+ "should_log": false,
286
+ "should_save": true,
287
+ "should_training_stop": false
288
+ },
289
+ "attributes": {}
290
+ }
291
+ },
292
+ "total_flos": 1.16987758903296e+17,
293
+ "train_batch_size": 256,
294
+ "trial_name": null,
295
+ "trial_params": null
296
+ }
checkpoint-7000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:970bd6d16b2730dba3763fdd1a14419de1f1cb106b82c2fb79c1d1c6012263ba
3
+ size 5905