Lhtie commited on
Commit
152f917
·
1 Parent(s): a17b3db

Upload folder using huggingface_hub

Browse files
commongen_pythia-14m/checkpoint-8600/config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/root/autodl-tmp/models/pythia-14m",
3
+ "architectures": [
4
+ "GPTNeoXForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": 0.1,
9
+ "eos_token_id": 0,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout": 0.0,
12
+ "hidden_size": 128,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 512,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 2048,
17
+ "model_type": "gpt_neox",
18
+ "num_attention_heads": 4,
19
+ "num_hidden_layers": 6,
20
+ "pad_token_id": 0,
21
+ "rope_scaling": null,
22
+ "rotary_emb_base": 10000,
23
+ "rotary_pct": 0.25,
24
+ "tie_word_embeddings": false,
25
+ "torch_dtype": "float32",
26
+ "transformers_version": "4.35.0",
27
+ "use_cache": true,
28
+ "use_parallel_residual": true,
29
+ "vocab_size": 50304
30
+ }
commongen_pythia-14m/checkpoint-8600/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.35.0"
6
+ }
commongen_pythia-14m/checkpoint-8600/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a3d507656b6bfdab2f4b334091bdb94ab680fa0ed4ac5fe6afb78a2841bd890
3
+ size 56279344
commongen_pythia-14m/checkpoint-8600/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:faf6af016182d48bdeafed698c79ef7fe6b2b7c970063f8416885336aa49f45b
3
+ size 112605242
commongen_pythia-14m/checkpoint-8600/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76c0b77b75b104d52df7208449e713d15585ed674305a174ce34842d80400b09
3
+ size 14244
commongen_pythia-14m/checkpoint-8600/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:190755b55d537470411618a77ef4166f486ecd76e22fa67093441787add8c1a8
3
+ size 1064
commongen_pythia-14m/checkpoint-8600/special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
commongen_pythia-14m/checkpoint-8600/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
commongen_pythia-14m/checkpoint-8600/tokenizer_config.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<|padding|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "50254": {
21
+ "content": " ",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": false
27
+ },
28
+ "50255": {
29
+ "content": " ",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": false
35
+ },
36
+ "50256": {
37
+ "content": " ",
38
+ "lstrip": false,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": false
43
+ },
44
+ "50257": {
45
+ "content": " ",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": false
51
+ },
52
+ "50258": {
53
+ "content": " ",
54
+ "lstrip": false,
55
+ "normalized": true,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": false
59
+ },
60
+ "50259": {
61
+ "content": " ",
62
+ "lstrip": false,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": false
67
+ },
68
+ "50260": {
69
+ "content": " ",
70
+ "lstrip": false,
71
+ "normalized": true,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": false
75
+ },
76
+ "50261": {
77
+ "content": " ",
78
+ "lstrip": false,
79
+ "normalized": true,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": false
83
+ },
84
+ "50262": {
85
+ "content": " ",
86
+ "lstrip": false,
87
+ "normalized": true,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": false
91
+ },
92
+ "50263": {
93
+ "content": " ",
94
+ "lstrip": false,
95
+ "normalized": true,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": false
99
+ },
100
+ "50264": {
101
+ "content": " ",
102
+ "lstrip": false,
103
+ "normalized": true,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": false
107
+ },
108
+ "50265": {
109
+ "content": " ",
110
+ "lstrip": false,
111
+ "normalized": true,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": false
115
+ },
116
+ "50266": {
117
+ "content": " ",
118
+ "lstrip": false,
119
+ "normalized": true,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": false
123
+ },
124
+ "50267": {
125
+ "content": " ",
126
+ "lstrip": false,
127
+ "normalized": true,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": false
131
+ },
132
+ "50268": {
133
+ "content": " ",
134
+ "lstrip": false,
135
+ "normalized": true,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": false
139
+ },
140
+ "50269": {
141
+ "content": " ",
142
+ "lstrip": false,
143
+ "normalized": true,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": false
147
+ },
148
+ "50270": {
149
+ "content": " ",
150
+ "lstrip": false,
151
+ "normalized": true,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": false
155
+ },
156
+ "50271": {
157
+ "content": " ",
158
+ "lstrip": false,
159
+ "normalized": true,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": false
163
+ },
164
+ "50272": {
165
+ "content": " ",
166
+ "lstrip": false,
167
+ "normalized": true,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": false
171
+ },
172
+ "50273": {
173
+ "content": " ",
174
+ "lstrip": false,
175
+ "normalized": true,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": false
179
+ },
180
+ "50274": {
181
+ "content": " ",
182
+ "lstrip": false,
183
+ "normalized": true,
184
+ "rstrip": false,
185
+ "single_word": false,
186
+ "special": false
187
+ },
188
+ "50275": {
189
+ "content": " ",
190
+ "lstrip": false,
191
+ "normalized": true,
192
+ "rstrip": false,
193
+ "single_word": false,
194
+ "special": false
195
+ },
196
+ "50276": {
197
+ "content": " ",
198
+ "lstrip": false,
199
+ "normalized": true,
200
+ "rstrip": false,
201
+ "single_word": false,
202
+ "special": false
203
+ }
204
+ },
205
+ "bos_token": "<|endoftext|>",
206
+ "clean_up_tokenization_spaces": true,
207
+ "eos_token": "<|endoftext|>",
208
+ "model_max_length": 1000000000000000019884624838656,
209
+ "pad_token": "<|endoftext|>",
210
+ "tokenizer_class": "GPTNeoXTokenizer",
211
+ "unk_token": "<|endoftext|>"
212
+ }
commongen_pythia-14m/checkpoint-8600/trainer_state.json ADDED
@@ -0,0 +1,465 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 54.41379928588867,
3
+ "best_model_checkpoint": "/root/autodl-tmp/commongen_pythia-14m/checkpoint-8600",
4
+ "epoch": 4.083570750237417,
5
+ "eval_steps": 200,
6
+ "global_step": 8600,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.09,
13
+ "eval_loss": 58.79890441894531,
14
+ "eval_runtime": 9.9925,
15
+ "eval_samples_per_second": 402.101,
16
+ "eval_steps_per_second": 25.219,
17
+ "step": 200
18
+ },
19
+ {
20
+ "epoch": 0.19,
21
+ "eval_loss": 58.18754577636719,
22
+ "eval_runtime": 10.0163,
23
+ "eval_samples_per_second": 401.145,
24
+ "eval_steps_per_second": 25.159,
25
+ "step": 400
26
+ },
27
+ {
28
+ "epoch": 0.24,
29
+ "learning_rate": 2.857549857549858e-06,
30
+ "loss": 57.7915,
31
+ "step": 500
32
+ },
33
+ {
34
+ "epoch": 0.28,
35
+ "eval_loss": 57.72026824951172,
36
+ "eval_runtime": 10.0204,
37
+ "eval_samples_per_second": 400.983,
38
+ "eval_steps_per_second": 25.149,
39
+ "step": 600
40
+ },
41
+ {
42
+ "epoch": 0.38,
43
+ "eval_loss": 57.014251708984375,
44
+ "eval_runtime": 10.0062,
45
+ "eval_samples_per_second": 401.551,
46
+ "eval_steps_per_second": 25.184,
47
+ "step": 800
48
+ },
49
+ {
50
+ "epoch": 0.47,
51
+ "learning_rate": 2.7150997150997152e-06,
52
+ "loss": 53.4826,
53
+ "step": 1000
54
+ },
55
+ {
56
+ "epoch": 0.47,
57
+ "eval_loss": 56.92744445800781,
58
+ "eval_runtime": 10.0163,
59
+ "eval_samples_per_second": 401.148,
60
+ "eval_steps_per_second": 25.159,
61
+ "step": 1000
62
+ },
63
+ {
64
+ "epoch": 0.57,
65
+ "eval_loss": 56.540775299072266,
66
+ "eval_runtime": 9.9969,
67
+ "eval_samples_per_second": 401.923,
68
+ "eval_steps_per_second": 25.208,
69
+ "step": 1200
70
+ },
71
+ {
72
+ "epoch": 0.66,
73
+ "eval_loss": 56.20606994628906,
74
+ "eval_runtime": 10.0482,
75
+ "eval_samples_per_second": 399.873,
76
+ "eval_steps_per_second": 25.079,
77
+ "step": 1400
78
+ },
79
+ {
80
+ "epoch": 0.71,
81
+ "learning_rate": 2.5726495726495726e-06,
82
+ "loss": 51.5395,
83
+ "step": 1500
84
+ },
85
+ {
86
+ "epoch": 0.76,
87
+ "eval_loss": 56.079891204833984,
88
+ "eval_runtime": 10.0186,
89
+ "eval_samples_per_second": 401.053,
90
+ "eval_steps_per_second": 25.153,
91
+ "step": 1600
92
+ },
93
+ {
94
+ "epoch": 0.85,
95
+ "eval_loss": 56.06427764892578,
96
+ "eval_runtime": 10.0084,
97
+ "eval_samples_per_second": 401.463,
98
+ "eval_steps_per_second": 25.179,
99
+ "step": 1800
100
+ },
101
+ {
102
+ "epoch": 0.95,
103
+ "learning_rate": 2.43019943019943e-06,
104
+ "loss": 50.4811,
105
+ "step": 2000
106
+ },
107
+ {
108
+ "epoch": 0.95,
109
+ "eval_loss": 55.9398193359375,
110
+ "eval_runtime": 10.0191,
111
+ "eval_samples_per_second": 401.035,
112
+ "eval_steps_per_second": 25.152,
113
+ "step": 2000
114
+ },
115
+ {
116
+ "epoch": 1.04,
117
+ "eval_loss": 55.618412017822266,
118
+ "eval_runtime": 10.0279,
119
+ "eval_samples_per_second": 400.682,
120
+ "eval_steps_per_second": 25.13,
121
+ "step": 2200
122
+ },
123
+ {
124
+ "epoch": 1.14,
125
+ "eval_loss": 55.55973815917969,
126
+ "eval_runtime": 10.0262,
127
+ "eval_samples_per_second": 400.749,
128
+ "eval_steps_per_second": 25.134,
129
+ "step": 2400
130
+ },
131
+ {
132
+ "epoch": 1.19,
133
+ "learning_rate": 2.2877492877492878e-06,
134
+ "loss": 49.6942,
135
+ "step": 2500
136
+ },
137
+ {
138
+ "epoch": 1.23,
139
+ "eval_loss": 55.494384765625,
140
+ "eval_runtime": 10.0514,
141
+ "eval_samples_per_second": 399.746,
142
+ "eval_steps_per_second": 25.071,
143
+ "step": 2600
144
+ },
145
+ {
146
+ "epoch": 1.33,
147
+ "eval_loss": 55.11967849731445,
148
+ "eval_runtime": 10.0372,
149
+ "eval_samples_per_second": 400.309,
150
+ "eval_steps_per_second": 25.107,
151
+ "step": 2800
152
+ },
153
+ {
154
+ "epoch": 1.42,
155
+ "learning_rate": 2.1452991452991456e-06,
156
+ "loss": 49.2263,
157
+ "step": 3000
158
+ },
159
+ {
160
+ "epoch": 1.42,
161
+ "eval_loss": 55.596561431884766,
162
+ "eval_runtime": 10.0512,
163
+ "eval_samples_per_second": 399.755,
164
+ "eval_steps_per_second": 25.072,
165
+ "step": 3000
166
+ },
167
+ {
168
+ "epoch": 1.52,
169
+ "eval_loss": 55.22591018676758,
170
+ "eval_runtime": 10.0077,
171
+ "eval_samples_per_second": 401.49,
172
+ "eval_steps_per_second": 25.181,
173
+ "step": 3200
174
+ },
175
+ {
176
+ "epoch": 1.61,
177
+ "eval_loss": 55.27616882324219,
178
+ "eval_runtime": 10.0993,
179
+ "eval_samples_per_second": 397.851,
180
+ "eval_steps_per_second": 24.952,
181
+ "step": 3400
182
+ },
183
+ {
184
+ "epoch": 1.66,
185
+ "learning_rate": 2.002849002849003e-06,
186
+ "loss": 48.5609,
187
+ "step": 3500
188
+ },
189
+ {
190
+ "epoch": 1.71,
191
+ "eval_loss": 55.0584716796875,
192
+ "eval_runtime": 10.033,
193
+ "eval_samples_per_second": 400.478,
194
+ "eval_steps_per_second": 25.117,
195
+ "step": 3600
196
+ },
197
+ {
198
+ "epoch": 1.8,
199
+ "eval_loss": 55.16390609741211,
200
+ "eval_runtime": 10.0176,
201
+ "eval_samples_per_second": 401.095,
202
+ "eval_steps_per_second": 25.156,
203
+ "step": 3800
204
+ },
205
+ {
206
+ "epoch": 1.9,
207
+ "learning_rate": 1.8603988603988605e-06,
208
+ "loss": 48.5437,
209
+ "step": 4000
210
+ },
211
+ {
212
+ "epoch": 1.9,
213
+ "eval_loss": 55.10284423828125,
214
+ "eval_runtime": 10.0065,
215
+ "eval_samples_per_second": 401.539,
216
+ "eval_steps_per_second": 25.184,
217
+ "step": 4000
218
+ },
219
+ {
220
+ "epoch": 1.99,
221
+ "eval_loss": 54.93409729003906,
222
+ "eval_runtime": 10.0491,
223
+ "eval_samples_per_second": 399.837,
224
+ "eval_steps_per_second": 25.077,
225
+ "step": 4200
226
+ },
227
+ {
228
+ "epoch": 2.09,
229
+ "eval_loss": 54.955745697021484,
230
+ "eval_runtime": 10.0389,
231
+ "eval_samples_per_second": 400.243,
232
+ "eval_steps_per_second": 25.102,
233
+ "step": 4400
234
+ },
235
+ {
236
+ "epoch": 2.14,
237
+ "learning_rate": 1.7179487179487179e-06,
238
+ "loss": 47.7019,
239
+ "step": 4500
240
+ },
241
+ {
242
+ "epoch": 2.18,
243
+ "eval_loss": 54.88346862792969,
244
+ "eval_runtime": 10.0155,
245
+ "eval_samples_per_second": 401.177,
246
+ "eval_steps_per_second": 25.161,
247
+ "step": 4600
248
+ },
249
+ {
250
+ "epoch": 2.28,
251
+ "eval_loss": 54.79231643676758,
252
+ "eval_runtime": 10.0246,
253
+ "eval_samples_per_second": 400.814,
254
+ "eval_steps_per_second": 25.138,
255
+ "step": 4800
256
+ },
257
+ {
258
+ "epoch": 2.37,
259
+ "learning_rate": 1.5754985754985755e-06,
260
+ "loss": 47.4877,
261
+ "step": 5000
262
+ },
263
+ {
264
+ "epoch": 2.37,
265
+ "eval_loss": 54.79960632324219,
266
+ "eval_runtime": 10.0491,
267
+ "eval_samples_per_second": 399.838,
268
+ "eval_steps_per_second": 25.077,
269
+ "step": 5000
270
+ },
271
+ {
272
+ "epoch": 2.47,
273
+ "eval_loss": 54.864017486572266,
274
+ "eval_runtime": 10.0635,
275
+ "eval_samples_per_second": 399.266,
276
+ "eval_steps_per_second": 25.041,
277
+ "step": 5200
278
+ },
279
+ {
280
+ "epoch": 2.56,
281
+ "eval_loss": 54.67957305908203,
282
+ "eval_runtime": 10.0753,
283
+ "eval_samples_per_second": 398.795,
284
+ "eval_steps_per_second": 25.012,
285
+ "step": 5400
286
+ },
287
+ {
288
+ "epoch": 2.61,
289
+ "learning_rate": 1.433048433048433e-06,
290
+ "loss": 47.3051,
291
+ "step": 5500
292
+ },
293
+ {
294
+ "epoch": 2.66,
295
+ "eval_loss": 54.71949005126953,
296
+ "eval_runtime": 10.0262,
297
+ "eval_samples_per_second": 400.749,
298
+ "eval_steps_per_second": 25.134,
299
+ "step": 5600
300
+ },
301
+ {
302
+ "epoch": 2.75,
303
+ "eval_loss": 54.726844787597656,
304
+ "eval_runtime": 10.0156,
305
+ "eval_samples_per_second": 401.173,
306
+ "eval_steps_per_second": 25.161,
307
+ "step": 5800
308
+ },
309
+ {
310
+ "epoch": 2.85,
311
+ "learning_rate": 1.2905982905982906e-06,
312
+ "loss": 47.4089,
313
+ "step": 6000
314
+ },
315
+ {
316
+ "epoch": 2.85,
317
+ "eval_loss": 54.711177825927734,
318
+ "eval_runtime": 10.0125,
319
+ "eval_samples_per_second": 401.3,
320
+ "eval_steps_per_second": 25.169,
321
+ "step": 6000
322
+ },
323
+ {
324
+ "epoch": 2.94,
325
+ "eval_loss": 54.760623931884766,
326
+ "eval_runtime": 10.0437,
327
+ "eval_samples_per_second": 400.05,
328
+ "eval_steps_per_second": 25.09,
329
+ "step": 6200
330
+ },
331
+ {
332
+ "epoch": 3.04,
333
+ "eval_loss": 54.60385513305664,
334
+ "eval_runtime": 10.0451,
335
+ "eval_samples_per_second": 399.996,
336
+ "eval_steps_per_second": 25.087,
337
+ "step": 6400
338
+ },
339
+ {
340
+ "epoch": 3.09,
341
+ "learning_rate": 1.1481481481481482e-06,
342
+ "loss": 46.9487,
343
+ "step": 6500
344
+ },
345
+ {
346
+ "epoch": 3.13,
347
+ "eval_loss": 54.6673469543457,
348
+ "eval_runtime": 10.1876,
349
+ "eval_samples_per_second": 394.4,
350
+ "eval_steps_per_second": 24.736,
351
+ "step": 6600
352
+ },
353
+ {
354
+ "epoch": 3.23,
355
+ "eval_loss": 54.58108139038086,
356
+ "eval_runtime": 10.0257,
357
+ "eval_samples_per_second": 400.772,
358
+ "eval_steps_per_second": 25.136,
359
+ "step": 6800
360
+ },
361
+ {
362
+ "epoch": 3.32,
363
+ "learning_rate": 1.0056980056980058e-06,
364
+ "loss": 46.701,
365
+ "step": 7000
366
+ },
367
+ {
368
+ "epoch": 3.32,
369
+ "eval_loss": 54.747596740722656,
370
+ "eval_runtime": 10.0284,
371
+ "eval_samples_per_second": 400.662,
372
+ "eval_steps_per_second": 25.129,
373
+ "step": 7000
374
+ },
375
+ {
376
+ "epoch": 3.42,
377
+ "eval_loss": 54.573760986328125,
378
+ "eval_runtime": 10.0309,
379
+ "eval_samples_per_second": 400.563,
380
+ "eval_steps_per_second": 25.122,
381
+ "step": 7200
382
+ },
383
+ {
384
+ "epoch": 3.51,
385
+ "eval_loss": 54.65139389038086,
386
+ "eval_runtime": 10.0453,
387
+ "eval_samples_per_second": 399.988,
388
+ "eval_steps_per_second": 25.086,
389
+ "step": 7400
390
+ },
391
+ {
392
+ "epoch": 3.56,
393
+ "learning_rate": 8.632478632478633e-07,
394
+ "loss": 46.3709,
395
+ "step": 7500
396
+ },
397
+ {
398
+ "epoch": 3.61,
399
+ "eval_loss": 54.60676193237305,
400
+ "eval_runtime": 10.0575,
401
+ "eval_samples_per_second": 399.504,
402
+ "eval_steps_per_second": 25.056,
403
+ "step": 7600
404
+ },
405
+ {
406
+ "epoch": 3.7,
407
+ "eval_loss": 54.56814193725586,
408
+ "eval_runtime": 10.0268,
409
+ "eval_samples_per_second": 400.728,
410
+ "eval_steps_per_second": 25.133,
411
+ "step": 7800
412
+ },
413
+ {
414
+ "epoch": 3.8,
415
+ "learning_rate": 7.207977207977208e-07,
416
+ "loss": 46.4731,
417
+ "step": 8000
418
+ },
419
+ {
420
+ "epoch": 3.8,
421
+ "eval_loss": 54.44728469848633,
422
+ "eval_runtime": 10.0513,
423
+ "eval_samples_per_second": 399.751,
424
+ "eval_steps_per_second": 25.071,
425
+ "step": 8000
426
+ },
427
+ {
428
+ "epoch": 3.89,
429
+ "eval_loss": 54.50142288208008,
430
+ "eval_runtime": 10.0849,
431
+ "eval_samples_per_second": 398.416,
432
+ "eval_steps_per_second": 24.988,
433
+ "step": 8200
434
+ },
435
+ {
436
+ "epoch": 3.99,
437
+ "eval_loss": 54.60088348388672,
438
+ "eval_runtime": 10.0468,
439
+ "eval_samples_per_second": 399.93,
440
+ "eval_steps_per_second": 25.083,
441
+ "step": 8400
442
+ },
443
+ {
444
+ "epoch": 4.04,
445
+ "learning_rate": 5.783475783475783e-07,
446
+ "loss": 46.3391,
447
+ "step": 8500
448
+ },
449
+ {
450
+ "epoch": 4.08,
451
+ "eval_loss": 54.41379928588867,
452
+ "eval_runtime": 10.0449,
453
+ "eval_samples_per_second": 400.003,
454
+ "eval_steps_per_second": 25.087,
455
+ "step": 8600
456
+ }
457
+ ],
458
+ "logging_steps": 500,
459
+ "max_steps": 10530,
460
+ "num_train_epochs": 5,
461
+ "save_steps": 200,
462
+ "total_flos": 6449216146636800.0,
463
+ "trial_name": null,
464
+ "trial_params": null
465
+ }
commongen_pythia-14m/checkpoint-8600/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c16564ae78ac521e044a6e37f59b07d81e3756fca89730bd7b3917073e8bb586
3
+ size 4600
commongen_pythia-14m/config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/root/autodl-tmp/models/pythia-14m",
3
+ "architectures": [
4
+ "GPTNeoXForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": 0.1,
9
+ "eos_token_id": 0,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout": 0.0,
12
+ "hidden_size": 128,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 512,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 2048,
17
+ "model_type": "gpt_neox",
18
+ "num_attention_heads": 4,
19
+ "num_hidden_layers": 6,
20
+ "pad_token_id": 0,
21
+ "rope_scaling": null,
22
+ "rotary_emb_base": 10000,
23
+ "rotary_pct": 0.25,
24
+ "tie_word_embeddings": false,
25
+ "torch_dtype": "float32",
26
+ "transformers_version": "4.35.0",
27
+ "use_cache": true,
28
+ "use_parallel_residual": true,
29
+ "vocab_size": 50304
30
+ }
commongen_pythia-14m/eval_results_lm.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ perplexity = 4.281665832306431e+23
commongen_pythia-14m/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.35.0"
6
+ }
commongen_pythia-14m/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a3d507656b6bfdab2f4b334091bdb94ab680fa0ed4ac5fe6afb78a2841bd890
3
+ size 56279344
commongen_pythia-14m/special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
commongen_pythia-14m/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
commongen_pythia-14m/tokenizer_config.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<|padding|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "50254": {
21
+ "content": " ",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": false
27
+ },
28
+ "50255": {
29
+ "content": " ",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": false
35
+ },
36
+ "50256": {
37
+ "content": " ",
38
+ "lstrip": false,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": false
43
+ },
44
+ "50257": {
45
+ "content": " ",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": false
51
+ },
52
+ "50258": {
53
+ "content": " ",
54
+ "lstrip": false,
55
+ "normalized": true,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": false
59
+ },
60
+ "50259": {
61
+ "content": " ",
62
+ "lstrip": false,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": false
67
+ },
68
+ "50260": {
69
+ "content": " ",
70
+ "lstrip": false,
71
+ "normalized": true,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": false
75
+ },
76
+ "50261": {
77
+ "content": " ",
78
+ "lstrip": false,
79
+ "normalized": true,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": false
83
+ },
84
+ "50262": {
85
+ "content": " ",
86
+ "lstrip": false,
87
+ "normalized": true,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": false
91
+ },
92
+ "50263": {
93
+ "content": " ",
94
+ "lstrip": false,
95
+ "normalized": true,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": false
99
+ },
100
+ "50264": {
101
+ "content": " ",
102
+ "lstrip": false,
103
+ "normalized": true,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": false
107
+ },
108
+ "50265": {
109
+ "content": " ",
110
+ "lstrip": false,
111
+ "normalized": true,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": false
115
+ },
116
+ "50266": {
117
+ "content": " ",
118
+ "lstrip": false,
119
+ "normalized": true,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": false
123
+ },
124
+ "50267": {
125
+ "content": " ",
126
+ "lstrip": false,
127
+ "normalized": true,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": false
131
+ },
132
+ "50268": {
133
+ "content": " ",
134
+ "lstrip": false,
135
+ "normalized": true,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": false
139
+ },
140
+ "50269": {
141
+ "content": " ",
142
+ "lstrip": false,
143
+ "normalized": true,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": false
147
+ },
148
+ "50270": {
149
+ "content": " ",
150
+ "lstrip": false,
151
+ "normalized": true,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": false
155
+ },
156
+ "50271": {
157
+ "content": " ",
158
+ "lstrip": false,
159
+ "normalized": true,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": false
163
+ },
164
+ "50272": {
165
+ "content": " ",
166
+ "lstrip": false,
167
+ "normalized": true,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": false
171
+ },
172
+ "50273": {
173
+ "content": " ",
174
+ "lstrip": false,
175
+ "normalized": true,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": false
179
+ },
180
+ "50274": {
181
+ "content": " ",
182
+ "lstrip": false,
183
+ "normalized": true,
184
+ "rstrip": false,
185
+ "single_word": false,
186
+ "special": false
187
+ },
188
+ "50275": {
189
+ "content": " ",
190
+ "lstrip": false,
191
+ "normalized": true,
192
+ "rstrip": false,
193
+ "single_word": false,
194
+ "special": false
195
+ },
196
+ "50276": {
197
+ "content": " ",
198
+ "lstrip": false,
199
+ "normalized": true,
200
+ "rstrip": false,
201
+ "single_word": false,
202
+ "special": false
203
+ }
204
+ },
205
+ "bos_token": "<|endoftext|>",
206
+ "clean_up_tokenization_spaces": true,
207
+ "eos_token": "<|endoftext|>",
208
+ "model_max_length": 1000000000000000019884624838656,
209
+ "pad_token": "<|endoftext|>",
210
+ "tokenizer_class": "GPTNeoXTokenizer",
211
+ "unk_token": "<|endoftext|>"
212
+ }
commongen_pythia-14m/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c16564ae78ac521e044a6e37f59b07d81e3756fca89730bd7b3917073e8bb586
3
+ size 4600