w-ahmad commited on
Commit
2c7d9be
·
verified ·
1 Parent(s): 0c9cb42

Auto upload 2026-08-06T21:20:39.743386

Browse files
out/glu-linear-21L_run/README.md ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - generated_from_trainer
5
+ model-index:
6
+ - name: ACTio-glu-linear-21L
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # ACTio-glu-linear-21L
14
+
15
+ This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Loss: 0.2170
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - learning_rate: 0.0003
37
+ - train_batch_size: 64
38
+ - eval_batch_size: 64
39
+ - seed: 42
40
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
41
+ - lr_scheduler_type: constant
42
+ - training_steps: 2000
43
+
44
+ ### Training results
45
+
46
+ | Training Loss | Epoch | Step | Validation Loss |
47
+ |:-------------:|:------:|:----:|:---------------:|
48
+ | 3.1315 | 0.0135 | 200 | 2.9516 |
49
+ | 0.9027 | 0.0270 | 400 | 0.8581 |
50
+ | 0.4480 | 0.0404 | 600 | 0.4363 |
51
+ | 0.3353 | 0.0539 | 800 | 0.3362 |
52
+ | 0.2943 | 0.0674 | 1000 | 0.2928 |
53
+ | 0.2781 | 0.0809 | 1200 | 0.2711 |
54
+ | 0.2458 | 0.0944 | 1400 | 0.2476 |
55
+ | 0.2494 | 0.1079 | 1600 | 0.2558 |
56
+ | 0.2262 | 0.1213 | 1800 | 0.2250 |
57
+ | 0.2155 | 0.1348 | 2000 | 0.2170 |
58
+
59
+
60
+ ### Framework versions
61
+
62
+ - Transformers 5.15.0.dev0
63
+ - Pytorch 2.6.0+cu124
64
+ - Datasets 5.0.1
65
+ - Tokenizers 0.22.2
out/glu-linear-21L_run/checkpoint-1500/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "linear",
3
+ "architectures": [
4
+ "TinyLlamaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 1,
9
+ "dtype": "bfloat16",
10
+ "eos_token_id": 2,
11
+ "head_dim": 32,
12
+ "hidden_act": "silu",
13
+ "hidden_size": 128,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 256,
16
+ "max_position_embeddings": 512,
17
+ "mlp_bias": false,
18
+ "mlp_type": "glu",
19
+ "model_type": "tiny_llama",
20
+ "num_attention_heads": 4,
21
+ "num_hidden_layers": 21,
22
+ "num_key_value_heads": 4,
23
+ "pad_token_id": 0,
24
+ "pretraining_tp": 1,
25
+ "rms_norm_eps": 1e-06,
26
+ "rope_parameters": {
27
+ "rope_theta": 10000.0,
28
+ "rope_type": "default"
29
+ },
30
+ "tie_word_embeddings": true,
31
+ "tokenizer_name": "w-ahmad/tiny-stories-tokenizer",
32
+ "transformers_version": "5.15.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096
35
+ }
out/glu-linear-21L_run/checkpoint-1500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0ebceba14d7615d44c6395c274d6d5af2b95a99f1a1126c13de31de812d2150
3
+ size 7961528
out/glu-linear-21L_run/checkpoint-1500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34f6aad768aadc4847ff45de9e37793c7902817bfcf7e29251d228bd4718bfe3
3
+ size 16041274
out/glu-linear-21L_run/checkpoint-1500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2768285b45b2a0c05f6f50bbb8c0287fca6f62a8cde6d1b1f02151ac72ee8dc
3
+ size 14244
out/glu-linear-21L_run/checkpoint-1500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b26a759cc207e28dea06eae7ad531c2dd78a09978f6972edbfc1566adccd7142
3
+ size 1064
out/glu-linear-21L_run/checkpoint-1500/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
out/glu-linear-21L_run/checkpoint-1500/tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "eos_token": "<|endoftext|>",
6
+ "errors": "replace",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "model_max_length": 1024,
10
+ "pad_token": "<|endoftext|>",
11
+ "tokenizer_class": "GPT2Tokenizer",
12
+ "unk_token": "<|endoftext|>"
13
+ }
out/glu-linear-21L_run/checkpoint-1500/trainer_state.json ADDED
@@ -0,0 +1,615 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.10111223458038422,
6
+ "eval_steps": 200,
7
+ "global_step": 1500,
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.0013481631277384564,
14
+ "grad_norm": 1.2578125,
15
+ "learning_rate": 0.0003,
16
+ "loss": 7.812018585205078,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.1796875,
22
+ "learning_rate": 0.0003,
23
+ "loss": 7.013896942138672,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.1171875,
29
+ "learning_rate": 0.0003,
30
+ "loss": 6.431048583984375,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 2.09375,
36
+ "learning_rate": 0.0003,
37
+ "loss": 5.950856399536133,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.734375,
43
+ "learning_rate": 0.0003,
44
+ "loss": 5.422636413574219,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 2.796875,
50
+ "learning_rate": 0.0003,
51
+ "loss": 4.8625225067138675,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 2.40625,
57
+ "learning_rate": 0.0003,
58
+ "loss": 4.365345001220703,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 2.46875,
64
+ "learning_rate": 0.0003,
65
+ "loss": 3.927184295654297,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 2.171875,
71
+ "learning_rate": 0.0003,
72
+ "loss": 3.5190196990966798,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 2.296875,
78
+ "learning_rate": 0.0003,
79
+ "loss": 3.131466865539551,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.013481631277384564,
84
+ "eval_loss": 2.9516451358795166,
85
+ "eval_runtime": 9.5431,
86
+ "eval_samples_per_second": 998.314,
87
+ "eval_steps_per_second": 15.613,
88
+ "step": 200
89
+ },
90
+ {
91
+ "epoch": 0.01482979440512302,
92
+ "grad_norm": 1.953125,
93
+ "learning_rate": 0.0003,
94
+ "loss": 2.790528678894043,
95
+ "step": 220
96
+ },
97
+ {
98
+ "epoch": 0.016177957532861477,
99
+ "grad_norm": 1.6015625,
100
+ "learning_rate": 0.0003,
101
+ "loss": 2.474166679382324,
102
+ "step": 240
103
+ },
104
+ {
105
+ "epoch": 0.01752612066059993,
106
+ "grad_norm": 2.25,
107
+ "learning_rate": 0.0003,
108
+ "loss": 2.1872188568115236,
109
+ "step": 260
110
+ },
111
+ {
112
+ "epoch": 0.01887428378833839,
113
+ "grad_norm": 2.5625,
114
+ "learning_rate": 0.0003,
115
+ "loss": 1.9124364852905273,
116
+ "step": 280
117
+ },
118
+ {
119
+ "epoch": 0.020222446916076844,
120
+ "grad_norm": 2.0625,
121
+ "learning_rate": 0.0003,
122
+ "loss": 1.6639739990234375,
123
+ "step": 300
124
+ },
125
+ {
126
+ "epoch": 0.021570610043815303,
127
+ "grad_norm": 1.921875,
128
+ "learning_rate": 0.0003,
129
+ "loss": 1.4615384101867677,
130
+ "step": 320
131
+ },
132
+ {
133
+ "epoch": 0.022918773171553757,
134
+ "grad_norm": 2.09375,
135
+ "learning_rate": 0.0003,
136
+ "loss": 1.2803497314453125,
137
+ "step": 340
138
+ },
139
+ {
140
+ "epoch": 0.024266936299292215,
141
+ "grad_norm": 1.8203125,
142
+ "learning_rate": 0.0003,
143
+ "loss": 1.142194938659668,
144
+ "step": 360
145
+ },
146
+ {
147
+ "epoch": 0.02561509942703067,
148
+ "grad_norm": 1.7421875,
149
+ "learning_rate": 0.0003,
150
+ "loss": 1.0200925827026368,
151
+ "step": 380
152
+ },
153
+ {
154
+ "epoch": 0.026963262554769128,
155
+ "grad_norm": 1.1796875,
156
+ "learning_rate": 0.0003,
157
+ "loss": 0.9027064323425293,
158
+ "step": 400
159
+ },
160
+ {
161
+ "epoch": 0.026963262554769128,
162
+ "eval_loss": 0.8580902218818665,
163
+ "eval_runtime": 9.5025,
164
+ "eval_samples_per_second": 1002.573,
165
+ "eval_steps_per_second": 15.68,
166
+ "step": 400
167
+ },
168
+ {
169
+ "epoch": 0.028311425682507583,
170
+ "grad_norm": 1.0390625,
171
+ "learning_rate": 0.0003,
172
+ "loss": 0.8012882232666015,
173
+ "step": 420
174
+ },
175
+ {
176
+ "epoch": 0.02965958881024604,
177
+ "grad_norm": 0.474609375,
178
+ "learning_rate": 0.0003,
179
+ "loss": 0.7280299663543701,
180
+ "step": 440
181
+ },
182
+ {
183
+ "epoch": 0.031007751937984496,
184
+ "grad_norm": 0.3828125,
185
+ "learning_rate": 0.0003,
186
+ "loss": 0.644657564163208,
187
+ "step": 460
188
+ },
189
+ {
190
+ "epoch": 0.032355915065722954,
191
+ "grad_norm": 0.359375,
192
+ "learning_rate": 0.0003,
193
+ "loss": 0.589603853225708,
194
+ "step": 480
195
+ },
196
+ {
197
+ "epoch": 0.03370407819346141,
198
+ "grad_norm": 0.365234375,
199
+ "learning_rate": 0.0003,
200
+ "loss": 0.5414067268371582,
201
+ "step": 500
202
+ },
203
+ {
204
+ "epoch": 0.03505224132119986,
205
+ "grad_norm": 0.33984375,
206
+ "learning_rate": 0.0003,
207
+ "loss": 0.5065312862396241,
208
+ "step": 520
209
+ },
210
+ {
211
+ "epoch": 0.03640040444893832,
212
+ "grad_norm": 0.35546875,
213
+ "learning_rate": 0.0003,
214
+ "loss": 0.47360758781433104,
215
+ "step": 540
216
+ },
217
+ {
218
+ "epoch": 0.03774856757667678,
219
+ "grad_norm": 0.48046875,
220
+ "learning_rate": 0.0003,
221
+ "loss": 0.4548943996429443,
222
+ "step": 560
223
+ },
224
+ {
225
+ "epoch": 0.03909673070441524,
226
+ "grad_norm": 1.0546875,
227
+ "learning_rate": 0.0003,
228
+ "loss": 0.45261592864990235,
229
+ "step": 580
230
+ },
231
+ {
232
+ "epoch": 0.04044489383215369,
233
+ "grad_norm": 0.921875,
234
+ "learning_rate": 0.0003,
235
+ "loss": 0.4480260372161865,
236
+ "step": 600
237
+ },
238
+ {
239
+ "epoch": 0.04044489383215369,
240
+ "eval_loss": 0.43627649545669556,
241
+ "eval_runtime": 9.5647,
242
+ "eval_samples_per_second": 996.06,
243
+ "eval_steps_per_second": 15.578,
244
+ "step": 600
245
+ },
246
+ {
247
+ "epoch": 0.04179305695989215,
248
+ "grad_norm": 0.361328125,
249
+ "learning_rate": 0.0003,
250
+ "loss": 0.42321023941040037,
251
+ "step": 620
252
+ },
253
+ {
254
+ "epoch": 0.043141220087630605,
255
+ "grad_norm": 0.30859375,
256
+ "learning_rate": 0.0003,
257
+ "loss": 0.41012063026428225,
258
+ "step": 640
259
+ },
260
+ {
261
+ "epoch": 0.044489383215369056,
262
+ "grad_norm": 0.287109375,
263
+ "learning_rate": 0.0003,
264
+ "loss": 0.3932011842727661,
265
+ "step": 660
266
+ },
267
+ {
268
+ "epoch": 0.045837546343107514,
269
+ "grad_norm": 0.287109375,
270
+ "learning_rate": 0.0003,
271
+ "loss": 0.3835625648498535,
272
+ "step": 680
273
+ },
274
+ {
275
+ "epoch": 0.04718570947084597,
276
+ "grad_norm": 0.294921875,
277
+ "learning_rate": 0.0003,
278
+ "loss": 0.3730705499649048,
279
+ "step": 700
280
+ },
281
+ {
282
+ "epoch": 0.04853387259858443,
283
+ "grad_norm": 0.291015625,
284
+ "learning_rate": 0.0003,
285
+ "loss": 0.36589076519012453,
286
+ "step": 720
287
+ },
288
+ {
289
+ "epoch": 0.04988203572632288,
290
+ "grad_norm": 0.2890625,
291
+ "learning_rate": 0.0003,
292
+ "loss": 0.3550434589385986,
293
+ "step": 740
294
+ },
295
+ {
296
+ "epoch": 0.05123019885406134,
297
+ "grad_norm": 0.279296875,
298
+ "learning_rate": 0.0003,
299
+ "loss": 0.35113232135772704,
300
+ "step": 760
301
+ },
302
+ {
303
+ "epoch": 0.0525783619817998,
304
+ "grad_norm": 0.44140625,
305
+ "learning_rate": 0.0003,
306
+ "loss": 0.34433393478393554,
307
+ "step": 780
308
+ },
309
+ {
310
+ "epoch": 0.053926525109538256,
311
+ "grad_norm": 0.283203125,
312
+ "learning_rate": 0.0003,
313
+ "loss": 0.33530120849609374,
314
+ "step": 800
315
+ },
316
+ {
317
+ "epoch": 0.053926525109538256,
318
+ "eval_loss": 0.3361909091472626,
319
+ "eval_runtime": 9.5722,
320
+ "eval_samples_per_second": 995.282,
321
+ "eval_steps_per_second": 15.566,
322
+ "step": 800
323
+ },
324
+ {
325
+ "epoch": 0.05527468823727671,
326
+ "grad_norm": 0.328125,
327
+ "learning_rate": 0.0003,
328
+ "loss": 0.33123824596405027,
329
+ "step": 820
330
+ },
331
+ {
332
+ "epoch": 0.056622851365015166,
333
+ "grad_norm": 0.33984375,
334
+ "learning_rate": 0.0003,
335
+ "loss": 0.32522556781768797,
336
+ "step": 840
337
+ },
338
+ {
339
+ "epoch": 0.057971014492753624,
340
+ "grad_norm": 1.5546875,
341
+ "learning_rate": 0.0003,
342
+ "loss": 0.3394989252090454,
343
+ "step": 860
344
+ },
345
+ {
346
+ "epoch": 0.05931917762049208,
347
+ "grad_norm": 0.81640625,
348
+ "learning_rate": 0.0003,
349
+ "loss": 0.3292835235595703,
350
+ "step": 880
351
+ },
352
+ {
353
+ "epoch": 0.06066734074823053,
354
+ "grad_norm": 0.318359375,
355
+ "learning_rate": 0.0003,
356
+ "loss": 0.31739518642425535,
357
+ "step": 900
358
+ },
359
+ {
360
+ "epoch": 0.06201550387596899,
361
+ "grad_norm": 0.259765625,
362
+ "learning_rate": 0.0003,
363
+ "loss": 0.3103918552398682,
364
+ "step": 920
365
+ },
366
+ {
367
+ "epoch": 0.06336366700370745,
368
+ "grad_norm": 0.2578125,
369
+ "learning_rate": 0.0003,
370
+ "loss": 0.30496537685394287,
371
+ "step": 940
372
+ },
373
+ {
374
+ "epoch": 0.06471183013144591,
375
+ "grad_norm": 0.25390625,
376
+ "learning_rate": 0.0003,
377
+ "loss": 0.2993361711502075,
378
+ "step": 960
379
+ },
380
+ {
381
+ "epoch": 0.06605999325918437,
382
+ "grad_norm": 0.25390625,
383
+ "learning_rate": 0.0003,
384
+ "loss": 0.29626312255859377,
385
+ "step": 980
386
+ },
387
+ {
388
+ "epoch": 0.06740815638692282,
389
+ "grad_norm": 0.25390625,
390
+ "learning_rate": 0.0003,
391
+ "loss": 0.29426524639129636,
392
+ "step": 1000
393
+ },
394
+ {
395
+ "epoch": 0.06740815638692282,
396
+ "eval_loss": 0.292793869972229,
397
+ "eval_runtime": 9.5221,
398
+ "eval_samples_per_second": 1000.512,
399
+ "eval_steps_per_second": 15.648,
400
+ "step": 1000
401
+ },
402
+ {
403
+ "epoch": 0.06875631951466127,
404
+ "grad_norm": 0.25,
405
+ "learning_rate": 0.0003,
406
+ "loss": 0.28772878646850586,
407
+ "step": 1020
408
+ },
409
+ {
410
+ "epoch": 0.07010448264239973,
411
+ "grad_norm": 0.2412109375,
412
+ "learning_rate": 0.0003,
413
+ "loss": 0.28324379920959475,
414
+ "step": 1040
415
+ },
416
+ {
417
+ "epoch": 0.07145264577013818,
418
+ "grad_norm": 0.2578125,
419
+ "learning_rate": 0.0003,
420
+ "loss": 0.2820605993270874,
421
+ "step": 1060
422
+ },
423
+ {
424
+ "epoch": 0.07280080889787664,
425
+ "grad_norm": 0.267578125,
426
+ "learning_rate": 0.0003,
427
+ "loss": 0.2773261070251465,
428
+ "step": 1080
429
+ },
430
+ {
431
+ "epoch": 0.0741489720256151,
432
+ "grad_norm": 0.2890625,
433
+ "learning_rate": 0.0003,
434
+ "loss": 0.2759279489517212,
435
+ "step": 1100
436
+ },
437
+ {
438
+ "epoch": 0.07549713515335356,
439
+ "grad_norm": 0.259765625,
440
+ "learning_rate": 0.0003,
441
+ "loss": 0.27228398323059083,
442
+ "step": 1120
443
+ },
444
+ {
445
+ "epoch": 0.07684529828109202,
446
+ "grad_norm": 0.26171875,
447
+ "learning_rate": 0.0003,
448
+ "loss": 0.2675980806350708,
449
+ "step": 1140
450
+ },
451
+ {
452
+ "epoch": 0.07819346140883048,
453
+ "grad_norm": 0.291015625,
454
+ "learning_rate": 0.0003,
455
+ "loss": 0.2672437191009521,
456
+ "step": 1160
457
+ },
458
+ {
459
+ "epoch": 0.07954162453656892,
460
+ "grad_norm": 1.109375,
461
+ "learning_rate": 0.0003,
462
+ "loss": 0.2952279090881348,
463
+ "step": 1180
464
+ },
465
+ {
466
+ "epoch": 0.08088978766430738,
467
+ "grad_norm": 0.40625,
468
+ "learning_rate": 0.0003,
469
+ "loss": 0.27812211513519286,
470
+ "step": 1200
471
+ },
472
+ {
473
+ "epoch": 0.08088978766430738,
474
+ "eval_loss": 0.2710629403591156,
475
+ "eval_runtime": 9.6114,
476
+ "eval_samples_per_second": 991.218,
477
+ "eval_steps_per_second": 15.502,
478
+ "step": 1200
479
+ },
480
+ {
481
+ "epoch": 0.08223795079204584,
482
+ "grad_norm": 0.265625,
483
+ "learning_rate": 0.0003,
484
+ "loss": 0.26534721851348875,
485
+ "step": 1220
486
+ },
487
+ {
488
+ "epoch": 0.0835861139197843,
489
+ "grad_norm": 0.244140625,
490
+ "learning_rate": 0.0003,
491
+ "loss": 0.2620635747909546,
492
+ "step": 1240
493
+ },
494
+ {
495
+ "epoch": 0.08493427704752275,
496
+ "grad_norm": 0.2431640625,
497
+ "learning_rate": 0.0003,
498
+ "loss": 0.26040074825286863,
499
+ "step": 1260
500
+ },
501
+ {
502
+ "epoch": 0.08628244017526121,
503
+ "grad_norm": 0.23828125,
504
+ "learning_rate": 0.0003,
505
+ "loss": 0.2555613279342651,
506
+ "step": 1280
507
+ },
508
+ {
509
+ "epoch": 0.08763060330299967,
510
+ "grad_norm": 0.24609375,
511
+ "learning_rate": 0.0003,
512
+ "loss": 0.25553059577941895,
513
+ "step": 1300
514
+ },
515
+ {
516
+ "epoch": 0.08897876643073811,
517
+ "grad_norm": 0.2373046875,
518
+ "learning_rate": 0.0003,
519
+ "loss": 0.2525977611541748,
520
+ "step": 1320
521
+ },
522
+ {
523
+ "epoch": 0.09032692955847657,
524
+ "grad_norm": 0.2412109375,
525
+ "learning_rate": 0.0003,
526
+ "loss": 0.2513373374938965,
527
+ "step": 1340
528
+ },
529
+ {
530
+ "epoch": 0.09167509268621503,
531
+ "grad_norm": 0.232421875,
532
+ "learning_rate": 0.0003,
533
+ "loss": 0.2486825704574585,
534
+ "step": 1360
535
+ },
536
+ {
537
+ "epoch": 0.09302325581395349,
538
+ "grad_norm": 0.2373046875,
539
+ "learning_rate": 0.0003,
540
+ "loss": 0.24810290336608887,
541
+ "step": 1380
542
+ },
543
+ {
544
+ "epoch": 0.09437141894169195,
545
+ "grad_norm": 0.2392578125,
546
+ "learning_rate": 0.0003,
547
+ "loss": 0.24577083587646484,
548
+ "step": 1400
549
+ },
550
+ {
551
+ "epoch": 0.09437141894169195,
552
+ "eval_loss": 0.2475873976945877,
553
+ "eval_runtime": 9.6597,
554
+ "eval_samples_per_second": 986.26,
555
+ "eval_steps_per_second": 15.425,
556
+ "step": 1400
557
+ },
558
+ {
559
+ "epoch": 0.0957195820694304,
560
+ "grad_norm": 0.244140625,
561
+ "learning_rate": 0.0003,
562
+ "loss": 0.24549760818481445,
563
+ "step": 1420
564
+ },
565
+ {
566
+ "epoch": 0.09706774519716886,
567
+ "grad_norm": 0.23046875,
568
+ "learning_rate": 0.0003,
569
+ "loss": 0.24300794601440429,
570
+ "step": 1440
571
+ },
572
+ {
573
+ "epoch": 0.09841590832490732,
574
+ "grad_norm": 0.240234375,
575
+ "learning_rate": 0.0003,
576
+ "loss": 0.2409191370010376,
577
+ "step": 1460
578
+ },
579
+ {
580
+ "epoch": 0.09976407145264576,
581
+ "grad_norm": 0.234375,
582
+ "learning_rate": 0.0003,
583
+ "loss": 0.23888604640960692,
584
+ "step": 1480
585
+ },
586
+ {
587
+ "epoch": 0.10111223458038422,
588
+ "grad_norm": 0.228515625,
589
+ "learning_rate": 0.0003,
590
+ "loss": 0.23784949779510497,
591
+ "step": 1500
592
+ }
593
+ ],
594
+ "logging_steps": 20,
595
+ "max_steps": 2000,
596
+ "num_input_tokens_seen": 0,
597
+ "num_train_epochs": 1,
598
+ "save_steps": 500,
599
+ "stateful_callbacks": {
600
+ "TrainerControl": {
601
+ "args": {
602
+ "should_epoch_stop": false,
603
+ "should_evaluate": false,
604
+ "should_log": false,
605
+ "should_save": true,
606
+ "should_training_stop": false
607
+ },
608
+ "attributes": {}
609
+ }
610
+ },
611
+ "total_flos": 1016309219328000.0,
612
+ "train_batch_size": 64,
613
+ "trial_name": null,
614
+ "trial_params": null
615
+ }
out/glu-linear-21L_run/checkpoint-1500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:636c39b00b4f494a5fba64615c0e4abbc2e20c130b3919cffd1511c862e74f2a
3
+ size 4920
out/glu-linear-21L_run/checkpoint-2000/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "linear",
3
+ "architectures": [
4
+ "TinyLlamaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 1,
9
+ "dtype": "bfloat16",
10
+ "eos_token_id": 2,
11
+ "head_dim": 32,
12
+ "hidden_act": "silu",
13
+ "hidden_size": 128,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 256,
16
+ "max_position_embeddings": 512,
17
+ "mlp_bias": false,
18
+ "mlp_type": "glu",
19
+ "model_type": "tiny_llama",
20
+ "num_attention_heads": 4,
21
+ "num_hidden_layers": 21,
22
+ "num_key_value_heads": 4,
23
+ "pad_token_id": 0,
24
+ "pretraining_tp": 1,
25
+ "rms_norm_eps": 1e-06,
26
+ "rope_parameters": {
27
+ "rope_theta": 10000.0,
28
+ "rope_type": "default"
29
+ },
30
+ "tie_word_embeddings": true,
31
+ "tokenizer_name": "w-ahmad/tiny-stories-tokenizer",
32
+ "transformers_version": "5.15.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096
35
+ }
out/glu-linear-21L_run/checkpoint-2000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c08ff1fe71e9926e0aec1ac742b9c1692911ae4dc9d83f4652a3149759d8c44
3
+ size 7961528
out/glu-linear-21L_run/checkpoint-2000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9a7b125ba1b97cd856fb2592bca0ef5da3418613ae0880d80b7326372500dad
3
+ size 16041274
out/glu-linear-21L_run/checkpoint-2000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1a97db8e41139aa1239ba7fb79ddeb0af5998c6305a440c1fe182e6ad02f2f5
3
+ size 14244
out/glu-linear-21L_run/checkpoint-2000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e91519ad666af6acc4beafd44aa617aefb90f632388a0f75cbf88f957400c190
3
+ size 1064
out/glu-linear-21L_run/checkpoint-2000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
out/glu-linear-21L_run/checkpoint-2000/tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "eos_token": "<|endoftext|>",
6
+ "errors": "replace",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "model_max_length": 1024,
10
+ "pad_token": "<|endoftext|>",
11
+ "tokenizer_class": "GPT2Tokenizer",
12
+ "unk_token": "<|endoftext|>"
13
+ }
out/glu-linear-21L_run/checkpoint-2000/trainer_state.json ADDED
@@ -0,0 +1,814 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.13481631277384565,
6
+ "eval_steps": 200,
7
+ "global_step": 2000,
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.0013481631277384564,
14
+ "grad_norm": 1.2578125,
15
+ "learning_rate": 0.0003,
16
+ "loss": 7.812018585205078,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.1796875,
22
+ "learning_rate": 0.0003,
23
+ "loss": 7.013896942138672,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.1171875,
29
+ "learning_rate": 0.0003,
30
+ "loss": 6.431048583984375,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 2.09375,
36
+ "learning_rate": 0.0003,
37
+ "loss": 5.950856399536133,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.734375,
43
+ "learning_rate": 0.0003,
44
+ "loss": 5.422636413574219,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 2.796875,
50
+ "learning_rate": 0.0003,
51
+ "loss": 4.8625225067138675,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 2.40625,
57
+ "learning_rate": 0.0003,
58
+ "loss": 4.365345001220703,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 2.46875,
64
+ "learning_rate": 0.0003,
65
+ "loss": 3.927184295654297,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 2.171875,
71
+ "learning_rate": 0.0003,
72
+ "loss": 3.5190196990966798,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 2.296875,
78
+ "learning_rate": 0.0003,
79
+ "loss": 3.131466865539551,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.013481631277384564,
84
+ "eval_loss": 2.9516451358795166,
85
+ "eval_runtime": 9.5431,
86
+ "eval_samples_per_second": 998.314,
87
+ "eval_steps_per_second": 15.613,
88
+ "step": 200
89
+ },
90
+ {
91
+ "epoch": 0.01482979440512302,
92
+ "grad_norm": 1.953125,
93
+ "learning_rate": 0.0003,
94
+ "loss": 2.790528678894043,
95
+ "step": 220
96
+ },
97
+ {
98
+ "epoch": 0.016177957532861477,
99
+ "grad_norm": 1.6015625,
100
+ "learning_rate": 0.0003,
101
+ "loss": 2.474166679382324,
102
+ "step": 240
103
+ },
104
+ {
105
+ "epoch": 0.01752612066059993,
106
+ "grad_norm": 2.25,
107
+ "learning_rate": 0.0003,
108
+ "loss": 2.1872188568115236,
109
+ "step": 260
110
+ },
111
+ {
112
+ "epoch": 0.01887428378833839,
113
+ "grad_norm": 2.5625,
114
+ "learning_rate": 0.0003,
115
+ "loss": 1.9124364852905273,
116
+ "step": 280
117
+ },
118
+ {
119
+ "epoch": 0.020222446916076844,
120
+ "grad_norm": 2.0625,
121
+ "learning_rate": 0.0003,
122
+ "loss": 1.6639739990234375,
123
+ "step": 300
124
+ },
125
+ {
126
+ "epoch": 0.021570610043815303,
127
+ "grad_norm": 1.921875,
128
+ "learning_rate": 0.0003,
129
+ "loss": 1.4615384101867677,
130
+ "step": 320
131
+ },
132
+ {
133
+ "epoch": 0.022918773171553757,
134
+ "grad_norm": 2.09375,
135
+ "learning_rate": 0.0003,
136
+ "loss": 1.2803497314453125,
137
+ "step": 340
138
+ },
139
+ {
140
+ "epoch": 0.024266936299292215,
141
+ "grad_norm": 1.8203125,
142
+ "learning_rate": 0.0003,
143
+ "loss": 1.142194938659668,
144
+ "step": 360
145
+ },
146
+ {
147
+ "epoch": 0.02561509942703067,
148
+ "grad_norm": 1.7421875,
149
+ "learning_rate": 0.0003,
150
+ "loss": 1.0200925827026368,
151
+ "step": 380
152
+ },
153
+ {
154
+ "epoch": 0.026963262554769128,
155
+ "grad_norm": 1.1796875,
156
+ "learning_rate": 0.0003,
157
+ "loss": 0.9027064323425293,
158
+ "step": 400
159
+ },
160
+ {
161
+ "epoch": 0.026963262554769128,
162
+ "eval_loss": 0.8580902218818665,
163
+ "eval_runtime": 9.5025,
164
+ "eval_samples_per_second": 1002.573,
165
+ "eval_steps_per_second": 15.68,
166
+ "step": 400
167
+ },
168
+ {
169
+ "epoch": 0.028311425682507583,
170
+ "grad_norm": 1.0390625,
171
+ "learning_rate": 0.0003,
172
+ "loss": 0.8012882232666015,
173
+ "step": 420
174
+ },
175
+ {
176
+ "epoch": 0.02965958881024604,
177
+ "grad_norm": 0.474609375,
178
+ "learning_rate": 0.0003,
179
+ "loss": 0.7280299663543701,
180
+ "step": 440
181
+ },
182
+ {
183
+ "epoch": 0.031007751937984496,
184
+ "grad_norm": 0.3828125,
185
+ "learning_rate": 0.0003,
186
+ "loss": 0.644657564163208,
187
+ "step": 460
188
+ },
189
+ {
190
+ "epoch": 0.032355915065722954,
191
+ "grad_norm": 0.359375,
192
+ "learning_rate": 0.0003,
193
+ "loss": 0.589603853225708,
194
+ "step": 480
195
+ },
196
+ {
197
+ "epoch": 0.03370407819346141,
198
+ "grad_norm": 0.365234375,
199
+ "learning_rate": 0.0003,
200
+ "loss": 0.5414067268371582,
201
+ "step": 500
202
+ },
203
+ {
204
+ "epoch": 0.03505224132119986,
205
+ "grad_norm": 0.33984375,
206
+ "learning_rate": 0.0003,
207
+ "loss": 0.5065312862396241,
208
+ "step": 520
209
+ },
210
+ {
211
+ "epoch": 0.03640040444893832,
212
+ "grad_norm": 0.35546875,
213
+ "learning_rate": 0.0003,
214
+ "loss": 0.47360758781433104,
215
+ "step": 540
216
+ },
217
+ {
218
+ "epoch": 0.03774856757667678,
219
+ "grad_norm": 0.48046875,
220
+ "learning_rate": 0.0003,
221
+ "loss": 0.4548943996429443,
222
+ "step": 560
223
+ },
224
+ {
225
+ "epoch": 0.03909673070441524,
226
+ "grad_norm": 1.0546875,
227
+ "learning_rate": 0.0003,
228
+ "loss": 0.45261592864990235,
229
+ "step": 580
230
+ },
231
+ {
232
+ "epoch": 0.04044489383215369,
233
+ "grad_norm": 0.921875,
234
+ "learning_rate": 0.0003,
235
+ "loss": 0.4480260372161865,
236
+ "step": 600
237
+ },
238
+ {
239
+ "epoch": 0.04044489383215369,
240
+ "eval_loss": 0.43627649545669556,
241
+ "eval_runtime": 9.5647,
242
+ "eval_samples_per_second": 996.06,
243
+ "eval_steps_per_second": 15.578,
244
+ "step": 600
245
+ },
246
+ {
247
+ "epoch": 0.04179305695989215,
248
+ "grad_norm": 0.361328125,
249
+ "learning_rate": 0.0003,
250
+ "loss": 0.42321023941040037,
251
+ "step": 620
252
+ },
253
+ {
254
+ "epoch": 0.043141220087630605,
255
+ "grad_norm": 0.30859375,
256
+ "learning_rate": 0.0003,
257
+ "loss": 0.41012063026428225,
258
+ "step": 640
259
+ },
260
+ {
261
+ "epoch": 0.044489383215369056,
262
+ "grad_norm": 0.287109375,
263
+ "learning_rate": 0.0003,
264
+ "loss": 0.3932011842727661,
265
+ "step": 660
266
+ },
267
+ {
268
+ "epoch": 0.045837546343107514,
269
+ "grad_norm": 0.287109375,
270
+ "learning_rate": 0.0003,
271
+ "loss": 0.3835625648498535,
272
+ "step": 680
273
+ },
274
+ {
275
+ "epoch": 0.04718570947084597,
276
+ "grad_norm": 0.294921875,
277
+ "learning_rate": 0.0003,
278
+ "loss": 0.3730705499649048,
279
+ "step": 700
280
+ },
281
+ {
282
+ "epoch": 0.04853387259858443,
283
+ "grad_norm": 0.291015625,
284
+ "learning_rate": 0.0003,
285
+ "loss": 0.36589076519012453,
286
+ "step": 720
287
+ },
288
+ {
289
+ "epoch": 0.04988203572632288,
290
+ "grad_norm": 0.2890625,
291
+ "learning_rate": 0.0003,
292
+ "loss": 0.3550434589385986,
293
+ "step": 740
294
+ },
295
+ {
296
+ "epoch": 0.05123019885406134,
297
+ "grad_norm": 0.279296875,
298
+ "learning_rate": 0.0003,
299
+ "loss": 0.35113232135772704,
300
+ "step": 760
301
+ },
302
+ {
303
+ "epoch": 0.0525783619817998,
304
+ "grad_norm": 0.44140625,
305
+ "learning_rate": 0.0003,
306
+ "loss": 0.34433393478393554,
307
+ "step": 780
308
+ },
309
+ {
310
+ "epoch": 0.053926525109538256,
311
+ "grad_norm": 0.283203125,
312
+ "learning_rate": 0.0003,
313
+ "loss": 0.33530120849609374,
314
+ "step": 800
315
+ },
316
+ {
317
+ "epoch": 0.053926525109538256,
318
+ "eval_loss": 0.3361909091472626,
319
+ "eval_runtime": 9.5722,
320
+ "eval_samples_per_second": 995.282,
321
+ "eval_steps_per_second": 15.566,
322
+ "step": 800
323
+ },
324
+ {
325
+ "epoch": 0.05527468823727671,
326
+ "grad_norm": 0.328125,
327
+ "learning_rate": 0.0003,
328
+ "loss": 0.33123824596405027,
329
+ "step": 820
330
+ },
331
+ {
332
+ "epoch": 0.056622851365015166,
333
+ "grad_norm": 0.33984375,
334
+ "learning_rate": 0.0003,
335
+ "loss": 0.32522556781768797,
336
+ "step": 840
337
+ },
338
+ {
339
+ "epoch": 0.057971014492753624,
340
+ "grad_norm": 1.5546875,
341
+ "learning_rate": 0.0003,
342
+ "loss": 0.3394989252090454,
343
+ "step": 860
344
+ },
345
+ {
346
+ "epoch": 0.05931917762049208,
347
+ "grad_norm": 0.81640625,
348
+ "learning_rate": 0.0003,
349
+ "loss": 0.3292835235595703,
350
+ "step": 880
351
+ },
352
+ {
353
+ "epoch": 0.06066734074823053,
354
+ "grad_norm": 0.318359375,
355
+ "learning_rate": 0.0003,
356
+ "loss": 0.31739518642425535,
357
+ "step": 900
358
+ },
359
+ {
360
+ "epoch": 0.06201550387596899,
361
+ "grad_norm": 0.259765625,
362
+ "learning_rate": 0.0003,
363
+ "loss": 0.3103918552398682,
364
+ "step": 920
365
+ },
366
+ {
367
+ "epoch": 0.06336366700370745,
368
+ "grad_norm": 0.2578125,
369
+ "learning_rate": 0.0003,
370
+ "loss": 0.30496537685394287,
371
+ "step": 940
372
+ },
373
+ {
374
+ "epoch": 0.06471183013144591,
375
+ "grad_norm": 0.25390625,
376
+ "learning_rate": 0.0003,
377
+ "loss": 0.2993361711502075,
378
+ "step": 960
379
+ },
380
+ {
381
+ "epoch": 0.06605999325918437,
382
+ "grad_norm": 0.25390625,
383
+ "learning_rate": 0.0003,
384
+ "loss": 0.29626312255859377,
385
+ "step": 980
386
+ },
387
+ {
388
+ "epoch": 0.06740815638692282,
389
+ "grad_norm": 0.25390625,
390
+ "learning_rate": 0.0003,
391
+ "loss": 0.29426524639129636,
392
+ "step": 1000
393
+ },
394
+ {
395
+ "epoch": 0.06740815638692282,
396
+ "eval_loss": 0.292793869972229,
397
+ "eval_runtime": 9.5221,
398
+ "eval_samples_per_second": 1000.512,
399
+ "eval_steps_per_second": 15.648,
400
+ "step": 1000
401
+ },
402
+ {
403
+ "epoch": 0.06875631951466127,
404
+ "grad_norm": 0.25,
405
+ "learning_rate": 0.0003,
406
+ "loss": 0.28772878646850586,
407
+ "step": 1020
408
+ },
409
+ {
410
+ "epoch": 0.07010448264239973,
411
+ "grad_norm": 0.2412109375,
412
+ "learning_rate": 0.0003,
413
+ "loss": 0.28324379920959475,
414
+ "step": 1040
415
+ },
416
+ {
417
+ "epoch": 0.07145264577013818,
418
+ "grad_norm": 0.2578125,
419
+ "learning_rate": 0.0003,
420
+ "loss": 0.2820605993270874,
421
+ "step": 1060
422
+ },
423
+ {
424
+ "epoch": 0.07280080889787664,
425
+ "grad_norm": 0.267578125,
426
+ "learning_rate": 0.0003,
427
+ "loss": 0.2773261070251465,
428
+ "step": 1080
429
+ },
430
+ {
431
+ "epoch": 0.0741489720256151,
432
+ "grad_norm": 0.2890625,
433
+ "learning_rate": 0.0003,
434
+ "loss": 0.2759279489517212,
435
+ "step": 1100
436
+ },
437
+ {
438
+ "epoch": 0.07549713515335356,
439
+ "grad_norm": 0.259765625,
440
+ "learning_rate": 0.0003,
441
+ "loss": 0.27228398323059083,
442
+ "step": 1120
443
+ },
444
+ {
445
+ "epoch": 0.07684529828109202,
446
+ "grad_norm": 0.26171875,
447
+ "learning_rate": 0.0003,
448
+ "loss": 0.2675980806350708,
449
+ "step": 1140
450
+ },
451
+ {
452
+ "epoch": 0.07819346140883048,
453
+ "grad_norm": 0.291015625,
454
+ "learning_rate": 0.0003,
455
+ "loss": 0.2672437191009521,
456
+ "step": 1160
457
+ },
458
+ {
459
+ "epoch": 0.07954162453656892,
460
+ "grad_norm": 1.109375,
461
+ "learning_rate": 0.0003,
462
+ "loss": 0.2952279090881348,
463
+ "step": 1180
464
+ },
465
+ {
466
+ "epoch": 0.08088978766430738,
467
+ "grad_norm": 0.40625,
468
+ "learning_rate": 0.0003,
469
+ "loss": 0.27812211513519286,
470
+ "step": 1200
471
+ },
472
+ {
473
+ "epoch": 0.08088978766430738,
474
+ "eval_loss": 0.2710629403591156,
475
+ "eval_runtime": 9.6114,
476
+ "eval_samples_per_second": 991.218,
477
+ "eval_steps_per_second": 15.502,
478
+ "step": 1200
479
+ },
480
+ {
481
+ "epoch": 0.08223795079204584,
482
+ "grad_norm": 0.265625,
483
+ "learning_rate": 0.0003,
484
+ "loss": 0.26534721851348875,
485
+ "step": 1220
486
+ },
487
+ {
488
+ "epoch": 0.0835861139197843,
489
+ "grad_norm": 0.244140625,
490
+ "learning_rate": 0.0003,
491
+ "loss": 0.2620635747909546,
492
+ "step": 1240
493
+ },
494
+ {
495
+ "epoch": 0.08493427704752275,
496
+ "grad_norm": 0.2431640625,
497
+ "learning_rate": 0.0003,
498
+ "loss": 0.26040074825286863,
499
+ "step": 1260
500
+ },
501
+ {
502
+ "epoch": 0.08628244017526121,
503
+ "grad_norm": 0.23828125,
504
+ "learning_rate": 0.0003,
505
+ "loss": 0.2555613279342651,
506
+ "step": 1280
507
+ },
508
+ {
509
+ "epoch": 0.08763060330299967,
510
+ "grad_norm": 0.24609375,
511
+ "learning_rate": 0.0003,
512
+ "loss": 0.25553059577941895,
513
+ "step": 1300
514
+ },
515
+ {
516
+ "epoch": 0.08897876643073811,
517
+ "grad_norm": 0.2373046875,
518
+ "learning_rate": 0.0003,
519
+ "loss": 0.2525977611541748,
520
+ "step": 1320
521
+ },
522
+ {
523
+ "epoch": 0.09032692955847657,
524
+ "grad_norm": 0.2412109375,
525
+ "learning_rate": 0.0003,
526
+ "loss": 0.2513373374938965,
527
+ "step": 1340
528
+ },
529
+ {
530
+ "epoch": 0.09167509268621503,
531
+ "grad_norm": 0.232421875,
532
+ "learning_rate": 0.0003,
533
+ "loss": 0.2486825704574585,
534
+ "step": 1360
535
+ },
536
+ {
537
+ "epoch": 0.09302325581395349,
538
+ "grad_norm": 0.2373046875,
539
+ "learning_rate": 0.0003,
540
+ "loss": 0.24810290336608887,
541
+ "step": 1380
542
+ },
543
+ {
544
+ "epoch": 0.09437141894169195,
545
+ "grad_norm": 0.2392578125,
546
+ "learning_rate": 0.0003,
547
+ "loss": 0.24577083587646484,
548
+ "step": 1400
549
+ },
550
+ {
551
+ "epoch": 0.09437141894169195,
552
+ "eval_loss": 0.2475873976945877,
553
+ "eval_runtime": 9.6597,
554
+ "eval_samples_per_second": 986.26,
555
+ "eval_steps_per_second": 15.425,
556
+ "step": 1400
557
+ },
558
+ {
559
+ "epoch": 0.0957195820694304,
560
+ "grad_norm": 0.244140625,
561
+ "learning_rate": 0.0003,
562
+ "loss": 0.24549760818481445,
563
+ "step": 1420
564
+ },
565
+ {
566
+ "epoch": 0.09706774519716886,
567
+ "grad_norm": 0.23046875,
568
+ "learning_rate": 0.0003,
569
+ "loss": 0.24300794601440429,
570
+ "step": 1440
571
+ },
572
+ {
573
+ "epoch": 0.09841590832490732,
574
+ "grad_norm": 0.240234375,
575
+ "learning_rate": 0.0003,
576
+ "loss": 0.2409191370010376,
577
+ "step": 1460
578
+ },
579
+ {
580
+ "epoch": 0.09976407145264576,
581
+ "grad_norm": 0.234375,
582
+ "learning_rate": 0.0003,
583
+ "loss": 0.23888604640960692,
584
+ "step": 1480
585
+ },
586
+ {
587
+ "epoch": 0.10111223458038422,
588
+ "grad_norm": 0.228515625,
589
+ "learning_rate": 0.0003,
590
+ "loss": 0.23784949779510497,
591
+ "step": 1500
592
+ },
593
+ {
594
+ "epoch": 0.10246039770812268,
595
+ "grad_norm": 0.2392578125,
596
+ "learning_rate": 0.0003,
597
+ "loss": 0.23578906059265137,
598
+ "step": 1520
599
+ },
600
+ {
601
+ "epoch": 0.10380856083586114,
602
+ "grad_norm": 0.23046875,
603
+ "learning_rate": 0.0003,
604
+ "loss": 0.23505594730377197,
605
+ "step": 1540
606
+ },
607
+ {
608
+ "epoch": 0.1051567239635996,
609
+ "grad_norm": 0.234375,
610
+ "learning_rate": 0.0003,
611
+ "loss": 0.23460586071014405,
612
+ "step": 1560
613
+ },
614
+ {
615
+ "epoch": 0.10650488709133805,
616
+ "grad_norm": 0.2392578125,
617
+ "learning_rate": 0.0003,
618
+ "loss": 0.23490052223205565,
619
+ "step": 1580
620
+ },
621
+ {
622
+ "epoch": 0.10785305021907651,
623
+ "grad_norm": 0.94140625,
624
+ "learning_rate": 0.0003,
625
+ "loss": 0.2493673086166382,
626
+ "step": 1600
627
+ },
628
+ {
629
+ "epoch": 0.10785305021907651,
630
+ "eval_loss": 0.2557738721370697,
631
+ "eval_runtime": 9.7901,
632
+ "eval_samples_per_second": 973.125,
633
+ "eval_steps_per_second": 15.219,
634
+ "step": 1600
635
+ },
636
+ {
637
+ "epoch": 0.10920121334681497,
638
+ "grad_norm": 0.384765625,
639
+ "learning_rate": 0.0003,
640
+ "loss": 0.24278874397277833,
641
+ "step": 1620
642
+ },
643
+ {
644
+ "epoch": 0.11054937647455342,
645
+ "grad_norm": 0.255859375,
646
+ "learning_rate": 0.0003,
647
+ "loss": 0.23434929847717284,
648
+ "step": 1640
649
+ },
650
+ {
651
+ "epoch": 0.11189753960229187,
652
+ "grad_norm": 0.2275390625,
653
+ "learning_rate": 0.0003,
654
+ "loss": 0.23051168918609619,
655
+ "step": 1660
656
+ },
657
+ {
658
+ "epoch": 0.11324570273003033,
659
+ "grad_norm": 0.2275390625,
660
+ "learning_rate": 0.0003,
661
+ "loss": 0.22942900657653809,
662
+ "step": 1680
663
+ },
664
+ {
665
+ "epoch": 0.11459386585776879,
666
+ "grad_norm": 0.220703125,
667
+ "learning_rate": 0.0003,
668
+ "loss": 0.22944440841674804,
669
+ "step": 1700
670
+ },
671
+ {
672
+ "epoch": 0.11594202898550725,
673
+ "grad_norm": 0.224609375,
674
+ "learning_rate": 0.0003,
675
+ "loss": 0.22700812816619872,
676
+ "step": 1720
677
+ },
678
+ {
679
+ "epoch": 0.1172901921132457,
680
+ "grad_norm": 0.22265625,
681
+ "learning_rate": 0.0003,
682
+ "loss": 0.22709190845489502,
683
+ "step": 1740
684
+ },
685
+ {
686
+ "epoch": 0.11863835524098416,
687
+ "grad_norm": 0.2255859375,
688
+ "learning_rate": 0.0003,
689
+ "loss": 0.22538518905639648,
690
+ "step": 1760
691
+ },
692
+ {
693
+ "epoch": 0.11998651836872262,
694
+ "grad_norm": 0.220703125,
695
+ "learning_rate": 0.0003,
696
+ "loss": 0.2261672019958496,
697
+ "step": 1780
698
+ },
699
+ {
700
+ "epoch": 0.12133468149646107,
701
+ "grad_norm": 0.220703125,
702
+ "learning_rate": 0.0003,
703
+ "loss": 0.22621147632598876,
704
+ "step": 1800
705
+ },
706
+ {
707
+ "epoch": 0.12133468149646107,
708
+ "eval_loss": 0.22504448890686035,
709
+ "eval_runtime": 9.5477,
710
+ "eval_samples_per_second": 997.834,
711
+ "eval_steps_per_second": 15.606,
712
+ "step": 1800
713
+ },
714
+ {
715
+ "epoch": 0.12268284462419952,
716
+ "grad_norm": 0.2236328125,
717
+ "learning_rate": 0.0003,
718
+ "loss": 0.2254558563232422,
719
+ "step": 1820
720
+ },
721
+ {
722
+ "epoch": 0.12403100775193798,
723
+ "grad_norm": 0.22265625,
724
+ "learning_rate": 0.0003,
725
+ "loss": 0.22039217948913575,
726
+ "step": 1840
727
+ },
728
+ {
729
+ "epoch": 0.12537917087967643,
730
+ "grad_norm": 0.2255859375,
731
+ "learning_rate": 0.0003,
732
+ "loss": 0.22042975425720215,
733
+ "step": 1860
734
+ },
735
+ {
736
+ "epoch": 0.1267273340074149,
737
+ "grad_norm": 0.2255859375,
738
+ "learning_rate": 0.0003,
739
+ "loss": 0.2201547384262085,
740
+ "step": 1880
741
+ },
742
+ {
743
+ "epoch": 0.12807549713515334,
744
+ "grad_norm": 0.23046875,
745
+ "learning_rate": 0.0003,
746
+ "loss": 0.21954381465911865,
747
+ "step": 1900
748
+ },
749
+ {
750
+ "epoch": 0.12942366026289182,
751
+ "grad_norm": 0.22265625,
752
+ "learning_rate": 0.0003,
753
+ "loss": 0.21725733280181886,
754
+ "step": 1920
755
+ },
756
+ {
757
+ "epoch": 0.13077182339063026,
758
+ "grad_norm": 0.22265625,
759
+ "learning_rate": 0.0003,
760
+ "loss": 0.21800496578216552,
761
+ "step": 1940
762
+ },
763
+ {
764
+ "epoch": 0.13211998651836873,
765
+ "grad_norm": 0.2216796875,
766
+ "learning_rate": 0.0003,
767
+ "loss": 0.21533818244934083,
768
+ "step": 1960
769
+ },
770
+ {
771
+ "epoch": 0.13346814964610718,
772
+ "grad_norm": 0.2255859375,
773
+ "learning_rate": 0.0003,
774
+ "loss": 0.21634423732757568,
775
+ "step": 1980
776
+ },
777
+ {
778
+ "epoch": 0.13481631277384565,
779
+ "grad_norm": 0.224609375,
780
+ "learning_rate": 0.0003,
781
+ "loss": 0.2155174970626831,
782
+ "step": 2000
783
+ },
784
+ {
785
+ "epoch": 0.13481631277384565,
786
+ "eval_loss": 0.21702784299850464,
787
+ "eval_runtime": 9.5219,
788
+ "eval_samples_per_second": 1000.54,
789
+ "eval_steps_per_second": 15.648,
790
+ "step": 2000
791
+ }
792
+ ],
793
+ "logging_steps": 20,
794
+ "max_steps": 2000,
795
+ "num_input_tokens_seen": 0,
796
+ "num_train_epochs": 1,
797
+ "save_steps": 500,
798
+ "stateful_callbacks": {
799
+ "TrainerControl": {
800
+ "args": {
801
+ "should_epoch_stop": false,
802
+ "should_evaluate": false,
803
+ "should_log": false,
804
+ "should_save": true,
805
+ "should_training_stop": true
806
+ },
807
+ "attributes": {}
808
+ }
809
+ },
810
+ "total_flos": 1355078959104000.0,
811
+ "train_batch_size": 64,
812
+ "trial_name": null,
813
+ "trial_params": null
814
+ }
out/glu-linear-21L_run/checkpoint-2000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:636c39b00b4f494a5fba64615c0e4abbc2e20c130b3919cffd1511c862e74f2a
3
+ size 4920
out/glu-linear-21L_run/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc9f3c67403db21cc4014335a1d41a93e3ffcb4a0247f90f300ae22ba851ad38
3
  size 7961528
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c08ff1fe71e9926e0aec1ac742b9c1692911ae4dc9d83f4652a3149759d8c44
3
  size 7961528
out/glu-linear-21L_run/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
out/glu-linear-21L_run/tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "eos_token": "<|endoftext|>",
6
+ "errors": "replace",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "model_max_length": 1024,
10
+ "pad_token": "<|endoftext|>",
11
+ "tokenizer_class": "GPT2Tokenizer",
12
+ "unk_token": "<|endoftext|>"
13
+ }
out/glu-linear-21L_run/training_log.jsonl CHANGED
The diff for this file is too large to render. See raw diff