w-ahmad commited on
Commit
88d4f27
·
verified ·
1 Parent(s): 53200c1

Auto upload zain 2026-08-12T21:49:22.340786

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/config.json +36 -0
  2. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/model.safetensors +3 -0
  3. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/optimizer.pt +3 -0
  4. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/rng_state.pth +3 -0
  5. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/scheduler.pt +3 -0
  6. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/tokenizer.json +0 -0
  7. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/tokenizer_config.json +13 -0
  8. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/trainer_state.json +544 -0
  9. zain/Activation/out/glu-silu-100L_run/checkpoint-1000/training_args.bin +3 -0
  10. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/config.json +36 -0
  11. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/model.safetensors +3 -0
  12. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/optimizer.pt +3 -0
  13. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/rng_state.pth +3 -0
  14. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/scheduler.pt +3 -0
  15. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/tokenizer.json +0 -0
  16. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/tokenizer_config.json +13 -0
  17. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/trainer_state.json +595 -0
  18. zain/Activation/out/glu-silu-100L_run/checkpoint-1100/training_args.bin +3 -0
  19. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/config.json +36 -0
  20. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/model.safetensors +3 -0
  21. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/optimizer.pt +3 -0
  22. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/rng_state.pth +3 -0
  23. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/scheduler.pt +3 -0
  24. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/tokenizer.json +0 -0
  25. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/tokenizer_config.json +13 -0
  26. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/trainer_state.json +646 -0
  27. zain/Activation/out/glu-silu-100L_run/checkpoint-1200/training_args.bin +3 -0
  28. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/config.json +36 -0
  29. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/model.safetensors +3 -0
  30. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/optimizer.pt +3 -0
  31. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/rng_state.pth +3 -0
  32. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/scheduler.pt +3 -0
  33. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/tokenizer.json +0 -0
  34. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/tokenizer_config.json +13 -0
  35. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/trainer_state.json +697 -0
  36. zain/Activation/out/glu-silu-100L_run/checkpoint-1300/training_args.bin +3 -0
  37. zain/Activation/out/glu-silu-100L_run/checkpoint-800/config.json +36 -0
  38. zain/Activation/out/glu-silu-100L_run/checkpoint-800/model.safetensors +3 -0
  39. zain/Activation/out/glu-silu-100L_run/checkpoint-800/optimizer.pt +3 -0
  40. zain/Activation/out/glu-silu-100L_run/checkpoint-800/rng_state.pth +3 -0
  41. zain/Activation/out/glu-silu-100L_run/checkpoint-800/scheduler.pt +3 -0
  42. zain/Activation/out/glu-silu-100L_run/checkpoint-800/tokenizer.json +0 -0
  43. zain/Activation/out/glu-silu-100L_run/checkpoint-800/tokenizer_config.json +13 -0
  44. zain/Activation/out/glu-silu-100L_run/checkpoint-800/trainer_state.json +442 -0
  45. zain/Activation/out/glu-silu-100L_run/checkpoint-800/training_args.bin +3 -0
  46. zain/Activation/out/glu-silu-100L_run/checkpoint-900/config.json +36 -0
  47. zain/Activation/out/glu-silu-100L_run/checkpoint-900/model.safetensors +3 -0
  48. zain/Activation/out/glu-silu-100L_run/checkpoint-900/optimizer.pt +3 -0
  49. zain/Activation/out/glu-silu-100L_run/checkpoint-900/rng_state.pth +3 -0
  50. zain/Activation/out/glu-silu-100L_run/checkpoint-900/scheduler.pt +3 -0
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "silu",
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": 100,
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.16.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096,
35
+ "waleed_beta": 10.0
36
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47bdc5ce7a92f40c6b3eaedaab763a9017036e50fba634dad06694dcdac3971a
3
+ size 33967272
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e231f77bfa3e1681a0d3f079dfe574ae29f7befaf2ea590692e543f42138cab
3
+ size 68504996
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:645b71843375baf4cc10bc75a2f0f04e91b8e3f8e8929f518a87022898b3bc20
3
+ size 14244
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf8d0c79c23f451fbfe25677a50bf40ae55d38cdf93a853c1a033c179996385d
3
+ size 1064
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/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
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/trainer_state.json ADDED
@@ -0,0 +1,544 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.06740815638692282,
6
+ "eval_steps": 50,
7
+ "global_step": 1000,
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": 2.078125,
15
+ "learning_rate": 2.66e-05,
16
+ "loss": 8.29962387084961,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.3515625,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047650909423828,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.003370407819346141,
28
+ "eval_loss": 7.835535049438477,
29
+ "eval_runtime": 15.2165,
30
+ "eval_samples_per_second": 626.095,
31
+ "eval_steps_per_second": 0.657,
32
+ "step": 50
33
+ },
34
+ {
35
+ "epoch": 0.004044489383215369,
36
+ "grad_norm": 1.2734375,
37
+ "learning_rate": 8.259999999999999e-05,
38
+ "loss": 7.8378746032714846,
39
+ "step": 60
40
+ },
41
+ {
42
+ "epoch": 0.005392652510953826,
43
+ "grad_norm": 1.265625,
44
+ "learning_rate": 0.0001106,
45
+ "loss": 7.5728271484375,
46
+ "step": 80
47
+ },
48
+ {
49
+ "epoch": 0.006740815638692282,
50
+ "grad_norm": 1.1953125,
51
+ "learning_rate": 0.0001386,
52
+ "loss": 7.2332916259765625,
53
+ "step": 100
54
+ },
55
+ {
56
+ "epoch": 0.006740815638692282,
57
+ "eval_loss": 7.039053440093994,
58
+ "eval_runtime": 15.2074,
59
+ "eval_samples_per_second": 626.471,
60
+ "eval_steps_per_second": 0.658,
61
+ "step": 100
62
+ },
63
+ {
64
+ "epoch": 0.008088978766430738,
65
+ "grad_norm": 1.1328125,
66
+ "learning_rate": 0.00016659999999999998,
67
+ "loss": 6.8608039855957035,
68
+ "step": 120
69
+ },
70
+ {
71
+ "epoch": 0.009437141894169195,
72
+ "grad_norm": 0.94921875,
73
+ "learning_rate": 0.00019460000000000001,
74
+ "loss": 6.517842864990234,
75
+ "step": 140
76
+ },
77
+ {
78
+ "epoch": 0.010111223458038422,
79
+ "eval_loss": 6.229617595672607,
80
+ "eval_runtime": 15.1978,
81
+ "eval_samples_per_second": 626.867,
82
+ "eval_steps_per_second": 0.658,
83
+ "step": 150
84
+ },
85
+ {
86
+ "epoch": 0.010785305021907651,
87
+ "grad_norm": 0.80078125,
88
+ "learning_rate": 0.0002226,
89
+ "loss": 6.2399169921875,
90
+ "step": 160
91
+ },
92
+ {
93
+ "epoch": 0.012133468149646108,
94
+ "grad_norm": 0.6640625,
95
+ "learning_rate": 0.00025059999999999997,
96
+ "loss": 6.036548614501953,
97
+ "step": 180
98
+ },
99
+ {
100
+ "epoch": 0.013481631277384564,
101
+ "grad_norm": 0.61328125,
102
+ "learning_rate": 0.0002786,
103
+ "loss": 5.857078552246094,
104
+ "step": 200
105
+ },
106
+ {
107
+ "epoch": 0.013481631277384564,
108
+ "eval_loss": 5.765108585357666,
109
+ "eval_runtime": 15.5271,
110
+ "eval_samples_per_second": 613.574,
111
+ "eval_steps_per_second": 0.644,
112
+ "step": 200
113
+ },
114
+ {
115
+ "epoch": 0.01482979440512302,
116
+ "grad_norm": 0.61328125,
117
+ "learning_rate": 0.00030659999999999997,
118
+ "loss": 5.680564117431641,
119
+ "step": 220
120
+ },
121
+ {
122
+ "epoch": 0.016177957532861477,
123
+ "grad_norm": 2.0625,
124
+ "learning_rate": 0.0003346,
125
+ "loss": 5.472259902954102,
126
+ "step": 240
127
+ },
128
+ {
129
+ "epoch": 0.016852039096730706,
130
+ "eval_loss": 5.240335464477539,
131
+ "eval_runtime": 15.2596,
132
+ "eval_samples_per_second": 624.33,
133
+ "eval_steps_per_second": 0.655,
134
+ "step": 250
135
+ },
136
+ {
137
+ "epoch": 0.01752612066059993,
138
+ "grad_norm": 1.7421875,
139
+ "learning_rate": 0.00036260000000000003,
140
+ "loss": 5.24955940246582,
141
+ "step": 260
142
+ },
143
+ {
144
+ "epoch": 0.01887428378833839,
145
+ "grad_norm": 0.62109375,
146
+ "learning_rate": 0.0003906,
147
+ "loss": 4.995759963989258,
148
+ "step": 280
149
+ },
150
+ {
151
+ "epoch": 0.020222446916076844,
152
+ "grad_norm": 1.390625,
153
+ "learning_rate": 0.0004186,
154
+ "loss": 4.769353866577148,
155
+ "step": 300
156
+ },
157
+ {
158
+ "epoch": 0.020222446916076844,
159
+ "eval_loss": 4.669629096984863,
160
+ "eval_runtime": 15.2595,
161
+ "eval_samples_per_second": 624.332,
162
+ "eval_steps_per_second": 0.655,
163
+ "step": 300
164
+ },
165
+ {
166
+ "epoch": 0.021570610043815303,
167
+ "grad_norm": 0.73828125,
168
+ "learning_rate": 0.0004466,
169
+ "loss": 4.569867324829102,
170
+ "step": 320
171
+ },
172
+ {
173
+ "epoch": 0.022918773171553757,
174
+ "grad_norm": 0.765625,
175
+ "learning_rate": 0.00047460000000000004,
176
+ "loss": 4.389446258544922,
177
+ "step": 340
178
+ },
179
+ {
180
+ "epoch": 0.023592854735422986,
181
+ "eval_loss": 4.231207847595215,
182
+ "eval_runtime": 15.2143,
183
+ "eval_samples_per_second": 626.188,
184
+ "eval_steps_per_second": 0.657,
185
+ "step": 350
186
+ },
187
+ {
188
+ "epoch": 0.024266936299292215,
189
+ "grad_norm": 1.2421875,
190
+ "learning_rate": 0.0005026,
191
+ "loss": 4.243848419189453,
192
+ "step": 360
193
+ },
194
+ {
195
+ "epoch": 0.02561509942703067,
196
+ "grad_norm": 0.7109375,
197
+ "learning_rate": 0.0005306,
198
+ "loss": 4.1090648651123045,
199
+ "step": 380
200
+ },
201
+ {
202
+ "epoch": 0.026963262554769128,
203
+ "grad_norm": 0.484375,
204
+ "learning_rate": 0.0005586,
205
+ "loss": 3.966172790527344,
206
+ "step": 400
207
+ },
208
+ {
209
+ "epoch": 0.026963262554769128,
210
+ "eval_loss": 3.923773765563965,
211
+ "eval_runtime": 15.5308,
212
+ "eval_samples_per_second": 613.426,
213
+ "eval_steps_per_second": 0.644,
214
+ "step": 400
215
+ },
216
+ {
217
+ "epoch": 0.028311425682507583,
218
+ "grad_norm": 0.6640625,
219
+ "learning_rate": 0.0005866,
220
+ "loss": 3.8559597015380858,
221
+ "step": 420
222
+ },
223
+ {
224
+ "epoch": 0.02965958881024604,
225
+ "grad_norm": 0.515625,
226
+ "learning_rate": 0.0006146,
227
+ "loss": 3.7943866729736326,
228
+ "step": 440
229
+ },
230
+ {
231
+ "epoch": 0.030333670374115267,
232
+ "eval_loss": 3.6977553367614746,
233
+ "eval_runtime": 15.2001,
234
+ "eval_samples_per_second": 626.77,
235
+ "eval_steps_per_second": 0.658,
236
+ "step": 450
237
+ },
238
+ {
239
+ "epoch": 0.031007751937984496,
240
+ "grad_norm": 0.51953125,
241
+ "learning_rate": 0.0006426,
242
+ "loss": 3.686443328857422,
243
+ "step": 460
244
+ },
245
+ {
246
+ "epoch": 0.032355915065722954,
247
+ "grad_norm": 0.62109375,
248
+ "learning_rate": 0.0006705999999999999,
249
+ "loss": 3.6123157501220704,
250
+ "step": 480
251
+ },
252
+ {
253
+ "epoch": 0.03370407819346141,
254
+ "grad_norm": 0.70703125,
255
+ "learning_rate": 0.0006986,
256
+ "loss": 3.542654800415039,
257
+ "step": 500
258
+ },
259
+ {
260
+ "epoch": 0.03370407819346141,
261
+ "eval_loss": 3.513820171356201,
262
+ "eval_runtime": 15.2193,
263
+ "eval_samples_per_second": 625.983,
264
+ "eval_steps_per_second": 0.657,
265
+ "step": 500
266
+ },
267
+ {
268
+ "epoch": 0.03505224132119986,
269
+ "grad_norm": 0.48046875,
270
+ "learning_rate": 0.0007,
271
+ "loss": 3.472199249267578,
272
+ "step": 520
273
+ },
274
+ {
275
+ "epoch": 0.03640040444893832,
276
+ "grad_norm": 0.53515625,
277
+ "learning_rate": 0.0007,
278
+ "loss": 3.396291732788086,
279
+ "step": 540
280
+ },
281
+ {
282
+ "epoch": 0.03707448601280755,
283
+ "eval_loss": 3.3581483364105225,
284
+ "eval_runtime": 15.3044,
285
+ "eval_samples_per_second": 622.502,
286
+ "eval_steps_per_second": 0.653,
287
+ "step": 550
288
+ },
289
+ {
290
+ "epoch": 0.03774856757667678,
291
+ "grad_norm": 0.47265625,
292
+ "learning_rate": 0.0007,
293
+ "loss": 3.347300720214844,
294
+ "step": 560
295
+ },
296
+ {
297
+ "epoch": 0.03909673070441524,
298
+ "grad_norm": 0.458984375,
299
+ "learning_rate": 0.0007,
300
+ "loss": 3.2885902404785154,
301
+ "step": 580
302
+ },
303
+ {
304
+ "epoch": 0.04044489383215369,
305
+ "grad_norm": 0.4375,
306
+ "learning_rate": 0.0007,
307
+ "loss": 3.25723876953125,
308
+ "step": 600
309
+ },
310
+ {
311
+ "epoch": 0.04044489383215369,
312
+ "eval_loss": 3.2229573726654053,
313
+ "eval_runtime": 15.2967,
314
+ "eval_samples_per_second": 622.812,
315
+ "eval_steps_per_second": 0.654,
316
+ "step": 600
317
+ },
318
+ {
319
+ "epoch": 0.04179305695989215,
320
+ "grad_norm": 0.458984375,
321
+ "learning_rate": 0.0007,
322
+ "loss": 3.1902334213256838,
323
+ "step": 620
324
+ },
325
+ {
326
+ "epoch": 0.043141220087630605,
327
+ "grad_norm": 0.40625,
328
+ "learning_rate": 0.0007,
329
+ "loss": 3.167861557006836,
330
+ "step": 640
331
+ },
332
+ {
333
+ "epoch": 0.043815301651499834,
334
+ "eval_loss": 3.120636463165283,
335
+ "eval_runtime": 15.3751,
336
+ "eval_samples_per_second": 619.639,
337
+ "eval_steps_per_second": 0.65,
338
+ "step": 650
339
+ },
340
+ {
341
+ "epoch": 0.044489383215369056,
342
+ "grad_norm": 0.494140625,
343
+ "learning_rate": 0.0007,
344
+ "loss": 3.1089324951171875,
345
+ "step": 660
346
+ },
347
+ {
348
+ "epoch": 0.045837546343107514,
349
+ "grad_norm": 0.435546875,
350
+ "learning_rate": 0.0007,
351
+ "loss": 3.0828268051147463,
352
+ "step": 680
353
+ },
354
+ {
355
+ "epoch": 0.04718570947084597,
356
+ "grad_norm": 0.427734375,
357
+ "learning_rate": 0.0007,
358
+ "loss": 3.0396488189697264,
359
+ "step": 700
360
+ },
361
+ {
362
+ "epoch": 0.04718570947084597,
363
+ "eval_loss": 3.0276408195495605,
364
+ "eval_runtime": 15.2735,
365
+ "eval_samples_per_second": 623.761,
366
+ "eval_steps_per_second": 0.655,
367
+ "step": 700
368
+ },
369
+ {
370
+ "epoch": 0.04853387259858443,
371
+ "grad_norm": 0.4375,
372
+ "learning_rate": 0.0007,
373
+ "loss": 3.0034805297851563,
374
+ "step": 720
375
+ },
376
+ {
377
+ "epoch": 0.04988203572632288,
378
+ "grad_norm": 0.52734375,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.96556396484375,
381
+ "step": 740
382
+ },
383
+ {
384
+ "epoch": 0.05055611729019211,
385
+ "eval_loss": 2.950009346008301,
386
+ "eval_runtime": 15.243,
387
+ "eval_samples_per_second": 625.009,
388
+ "eval_steps_per_second": 0.656,
389
+ "step": 750
390
+ },
391
+ {
392
+ "epoch": 0.05123019885406134,
393
+ "grad_norm": 0.451171875,
394
+ "learning_rate": 0.0007,
395
+ "loss": 2.950779151916504,
396
+ "step": 760
397
+ },
398
+ {
399
+ "epoch": 0.0525783619817998,
400
+ "grad_norm": 0.400390625,
401
+ "learning_rate": 0.0007,
402
+ "loss": 2.956203079223633,
403
+ "step": 780
404
+ },
405
+ {
406
+ "epoch": 0.053926525109538256,
407
+ "grad_norm": 0.40625,
408
+ "learning_rate": 0.0007,
409
+ "loss": 2.9017578125,
410
+ "step": 800
411
+ },
412
+ {
413
+ "epoch": 0.053926525109538256,
414
+ "eval_loss": 2.8844006061553955,
415
+ "eval_runtime": 15.2903,
416
+ "eval_samples_per_second": 623.073,
417
+ "eval_steps_per_second": 0.654,
418
+ "step": 800
419
+ },
420
+ {
421
+ "epoch": 0.05527468823727671,
422
+ "grad_norm": 0.41015625,
423
+ "learning_rate": 0.0007,
424
+ "loss": 2.887515640258789,
425
+ "step": 820
426
+ },
427
+ {
428
+ "epoch": 0.056622851365015166,
429
+ "grad_norm": 0.41796875,
430
+ "learning_rate": 0.0007,
431
+ "loss": 2.8556928634643555,
432
+ "step": 840
433
+ },
434
+ {
435
+ "epoch": 0.057296932928884395,
436
+ "eval_loss": 2.821610450744629,
437
+ "eval_runtime": 15.2253,
438
+ "eval_samples_per_second": 625.733,
439
+ "eval_steps_per_second": 0.657,
440
+ "step": 850
441
+ },
442
+ {
443
+ "epoch": 0.057971014492753624,
444
+ "grad_norm": 0.423828125,
445
+ "learning_rate": 0.0007,
446
+ "loss": 2.8130855560302734,
447
+ "step": 860
448
+ },
449
+ {
450
+ "epoch": 0.05931917762049208,
451
+ "grad_norm": 0.38671875,
452
+ "learning_rate": 0.0007,
453
+ "loss": 2.790329170227051,
454
+ "step": 880
455
+ },
456
+ {
457
+ "epoch": 0.06066734074823053,
458
+ "grad_norm": 0.427734375,
459
+ "learning_rate": 0.0007,
460
+ "loss": 2.7760162353515625,
461
+ "step": 900
462
+ },
463
+ {
464
+ "epoch": 0.06066734074823053,
465
+ "eval_loss": 2.7661373615264893,
466
+ "eval_runtime": 15.2736,
467
+ "eval_samples_per_second": 623.755,
468
+ "eval_steps_per_second": 0.655,
469
+ "step": 900
470
+ },
471
+ {
472
+ "epoch": 0.06201550387596899,
473
+ "grad_norm": 0.435546875,
474
+ "learning_rate": 0.0007,
475
+ "loss": 2.7462047576904296,
476
+ "step": 920
477
+ },
478
+ {
479
+ "epoch": 0.06336366700370745,
480
+ "grad_norm": 0.4140625,
481
+ "learning_rate": 0.0007,
482
+ "loss": 2.732274627685547,
483
+ "step": 940
484
+ },
485
+ {
486
+ "epoch": 0.06403774856757667,
487
+ "eval_loss": 2.713942289352417,
488
+ "eval_runtime": 15.6062,
489
+ "eval_samples_per_second": 610.463,
490
+ "eval_steps_per_second": 0.641,
491
+ "step": 950
492
+ },
493
+ {
494
+ "epoch": 0.06471183013144591,
495
+ "grad_norm": 0.419921875,
496
+ "learning_rate": 0.0007,
497
+ "loss": 2.705318069458008,
498
+ "step": 960
499
+ },
500
+ {
501
+ "epoch": 0.06605999325918437,
502
+ "grad_norm": 0.4296875,
503
+ "learning_rate": 0.0007,
504
+ "loss": 2.6892818450927733,
505
+ "step": 980
506
+ },
507
+ {
508
+ "epoch": 0.06740815638692282,
509
+ "grad_norm": 0.4453125,
510
+ "learning_rate": 0.0007,
511
+ "loss": 2.687115287780762,
512
+ "step": 1000
513
+ },
514
+ {
515
+ "epoch": 0.06740815638692282,
516
+ "eval_loss": 2.668647289276123,
517
+ "eval_runtime": 15.2597,
518
+ "eval_samples_per_second": 624.326,
519
+ "eval_steps_per_second": 0.655,
520
+ "step": 1000
521
+ }
522
+ ],
523
+ "logging_steps": 20,
524
+ "max_steps": 2500,
525
+ "num_input_tokens_seen": 0,
526
+ "num_train_epochs": 1,
527
+ "save_steps": 100,
528
+ "stateful_callbacks": {
529
+ "TrainerControl": {
530
+ "args": {
531
+ "should_epoch_stop": false,
532
+ "should_evaluate": false,
533
+ "should_log": false,
534
+ "should_save": true,
535
+ "should_training_stop": false
536
+ },
537
+ "attributes": {}
538
+ }
539
+ },
540
+ "total_flos": 3226283802624000.0,
541
+ "train_batch_size": 64,
542
+ "trial_name": null,
543
+ "trial_params": null
544
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd90512e56f813612dc668676d6f8cab72da110a42f23c61cafebf91064625a1
3
+ size 4920
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "silu",
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": 100,
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.16.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096,
35
+ "waleed_beta": 10.0
36
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fab0b2d2b9b3bf9cac5bb1ca6cfb0e1fdaf69fcf053033e1acc211ab028eb99
3
+ size 33967272
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:784b273b212fc8827838070b0b2e439d635ee89deb36a22ff889396baf857221
3
+ size 68504996
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d7dc5cd1ed4bd6f56fc9ef12090bb8dcd7e4f162336bc05d8c592731dbe5b42
3
+ size 14244
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88121baf7de31688d97eca5c32cb058d585673d52f936d32dcd04e8b1903dcac
3
+ size 1064
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/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
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/trainer_state.json ADDED
@@ -0,0 +1,595 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.0741489720256151,
6
+ "eval_steps": 50,
7
+ "global_step": 1100,
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": 2.078125,
15
+ "learning_rate": 2.66e-05,
16
+ "loss": 8.29962387084961,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.3515625,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047650909423828,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.003370407819346141,
28
+ "eval_loss": 7.835535049438477,
29
+ "eval_runtime": 15.2165,
30
+ "eval_samples_per_second": 626.095,
31
+ "eval_steps_per_second": 0.657,
32
+ "step": 50
33
+ },
34
+ {
35
+ "epoch": 0.004044489383215369,
36
+ "grad_norm": 1.2734375,
37
+ "learning_rate": 8.259999999999999e-05,
38
+ "loss": 7.8378746032714846,
39
+ "step": 60
40
+ },
41
+ {
42
+ "epoch": 0.005392652510953826,
43
+ "grad_norm": 1.265625,
44
+ "learning_rate": 0.0001106,
45
+ "loss": 7.5728271484375,
46
+ "step": 80
47
+ },
48
+ {
49
+ "epoch": 0.006740815638692282,
50
+ "grad_norm": 1.1953125,
51
+ "learning_rate": 0.0001386,
52
+ "loss": 7.2332916259765625,
53
+ "step": 100
54
+ },
55
+ {
56
+ "epoch": 0.006740815638692282,
57
+ "eval_loss": 7.039053440093994,
58
+ "eval_runtime": 15.2074,
59
+ "eval_samples_per_second": 626.471,
60
+ "eval_steps_per_second": 0.658,
61
+ "step": 100
62
+ },
63
+ {
64
+ "epoch": 0.008088978766430738,
65
+ "grad_norm": 1.1328125,
66
+ "learning_rate": 0.00016659999999999998,
67
+ "loss": 6.8608039855957035,
68
+ "step": 120
69
+ },
70
+ {
71
+ "epoch": 0.009437141894169195,
72
+ "grad_norm": 0.94921875,
73
+ "learning_rate": 0.00019460000000000001,
74
+ "loss": 6.517842864990234,
75
+ "step": 140
76
+ },
77
+ {
78
+ "epoch": 0.010111223458038422,
79
+ "eval_loss": 6.229617595672607,
80
+ "eval_runtime": 15.1978,
81
+ "eval_samples_per_second": 626.867,
82
+ "eval_steps_per_second": 0.658,
83
+ "step": 150
84
+ },
85
+ {
86
+ "epoch": 0.010785305021907651,
87
+ "grad_norm": 0.80078125,
88
+ "learning_rate": 0.0002226,
89
+ "loss": 6.2399169921875,
90
+ "step": 160
91
+ },
92
+ {
93
+ "epoch": 0.012133468149646108,
94
+ "grad_norm": 0.6640625,
95
+ "learning_rate": 0.00025059999999999997,
96
+ "loss": 6.036548614501953,
97
+ "step": 180
98
+ },
99
+ {
100
+ "epoch": 0.013481631277384564,
101
+ "grad_norm": 0.61328125,
102
+ "learning_rate": 0.0002786,
103
+ "loss": 5.857078552246094,
104
+ "step": 200
105
+ },
106
+ {
107
+ "epoch": 0.013481631277384564,
108
+ "eval_loss": 5.765108585357666,
109
+ "eval_runtime": 15.5271,
110
+ "eval_samples_per_second": 613.574,
111
+ "eval_steps_per_second": 0.644,
112
+ "step": 200
113
+ },
114
+ {
115
+ "epoch": 0.01482979440512302,
116
+ "grad_norm": 0.61328125,
117
+ "learning_rate": 0.00030659999999999997,
118
+ "loss": 5.680564117431641,
119
+ "step": 220
120
+ },
121
+ {
122
+ "epoch": 0.016177957532861477,
123
+ "grad_norm": 2.0625,
124
+ "learning_rate": 0.0003346,
125
+ "loss": 5.472259902954102,
126
+ "step": 240
127
+ },
128
+ {
129
+ "epoch": 0.016852039096730706,
130
+ "eval_loss": 5.240335464477539,
131
+ "eval_runtime": 15.2596,
132
+ "eval_samples_per_second": 624.33,
133
+ "eval_steps_per_second": 0.655,
134
+ "step": 250
135
+ },
136
+ {
137
+ "epoch": 0.01752612066059993,
138
+ "grad_norm": 1.7421875,
139
+ "learning_rate": 0.00036260000000000003,
140
+ "loss": 5.24955940246582,
141
+ "step": 260
142
+ },
143
+ {
144
+ "epoch": 0.01887428378833839,
145
+ "grad_norm": 0.62109375,
146
+ "learning_rate": 0.0003906,
147
+ "loss": 4.995759963989258,
148
+ "step": 280
149
+ },
150
+ {
151
+ "epoch": 0.020222446916076844,
152
+ "grad_norm": 1.390625,
153
+ "learning_rate": 0.0004186,
154
+ "loss": 4.769353866577148,
155
+ "step": 300
156
+ },
157
+ {
158
+ "epoch": 0.020222446916076844,
159
+ "eval_loss": 4.669629096984863,
160
+ "eval_runtime": 15.2595,
161
+ "eval_samples_per_second": 624.332,
162
+ "eval_steps_per_second": 0.655,
163
+ "step": 300
164
+ },
165
+ {
166
+ "epoch": 0.021570610043815303,
167
+ "grad_norm": 0.73828125,
168
+ "learning_rate": 0.0004466,
169
+ "loss": 4.569867324829102,
170
+ "step": 320
171
+ },
172
+ {
173
+ "epoch": 0.022918773171553757,
174
+ "grad_norm": 0.765625,
175
+ "learning_rate": 0.00047460000000000004,
176
+ "loss": 4.389446258544922,
177
+ "step": 340
178
+ },
179
+ {
180
+ "epoch": 0.023592854735422986,
181
+ "eval_loss": 4.231207847595215,
182
+ "eval_runtime": 15.2143,
183
+ "eval_samples_per_second": 626.188,
184
+ "eval_steps_per_second": 0.657,
185
+ "step": 350
186
+ },
187
+ {
188
+ "epoch": 0.024266936299292215,
189
+ "grad_norm": 1.2421875,
190
+ "learning_rate": 0.0005026,
191
+ "loss": 4.243848419189453,
192
+ "step": 360
193
+ },
194
+ {
195
+ "epoch": 0.02561509942703067,
196
+ "grad_norm": 0.7109375,
197
+ "learning_rate": 0.0005306,
198
+ "loss": 4.1090648651123045,
199
+ "step": 380
200
+ },
201
+ {
202
+ "epoch": 0.026963262554769128,
203
+ "grad_norm": 0.484375,
204
+ "learning_rate": 0.0005586,
205
+ "loss": 3.966172790527344,
206
+ "step": 400
207
+ },
208
+ {
209
+ "epoch": 0.026963262554769128,
210
+ "eval_loss": 3.923773765563965,
211
+ "eval_runtime": 15.5308,
212
+ "eval_samples_per_second": 613.426,
213
+ "eval_steps_per_second": 0.644,
214
+ "step": 400
215
+ },
216
+ {
217
+ "epoch": 0.028311425682507583,
218
+ "grad_norm": 0.6640625,
219
+ "learning_rate": 0.0005866,
220
+ "loss": 3.8559597015380858,
221
+ "step": 420
222
+ },
223
+ {
224
+ "epoch": 0.02965958881024604,
225
+ "grad_norm": 0.515625,
226
+ "learning_rate": 0.0006146,
227
+ "loss": 3.7943866729736326,
228
+ "step": 440
229
+ },
230
+ {
231
+ "epoch": 0.030333670374115267,
232
+ "eval_loss": 3.6977553367614746,
233
+ "eval_runtime": 15.2001,
234
+ "eval_samples_per_second": 626.77,
235
+ "eval_steps_per_second": 0.658,
236
+ "step": 450
237
+ },
238
+ {
239
+ "epoch": 0.031007751937984496,
240
+ "grad_norm": 0.51953125,
241
+ "learning_rate": 0.0006426,
242
+ "loss": 3.686443328857422,
243
+ "step": 460
244
+ },
245
+ {
246
+ "epoch": 0.032355915065722954,
247
+ "grad_norm": 0.62109375,
248
+ "learning_rate": 0.0006705999999999999,
249
+ "loss": 3.6123157501220704,
250
+ "step": 480
251
+ },
252
+ {
253
+ "epoch": 0.03370407819346141,
254
+ "grad_norm": 0.70703125,
255
+ "learning_rate": 0.0006986,
256
+ "loss": 3.542654800415039,
257
+ "step": 500
258
+ },
259
+ {
260
+ "epoch": 0.03370407819346141,
261
+ "eval_loss": 3.513820171356201,
262
+ "eval_runtime": 15.2193,
263
+ "eval_samples_per_second": 625.983,
264
+ "eval_steps_per_second": 0.657,
265
+ "step": 500
266
+ },
267
+ {
268
+ "epoch": 0.03505224132119986,
269
+ "grad_norm": 0.48046875,
270
+ "learning_rate": 0.0007,
271
+ "loss": 3.472199249267578,
272
+ "step": 520
273
+ },
274
+ {
275
+ "epoch": 0.03640040444893832,
276
+ "grad_norm": 0.53515625,
277
+ "learning_rate": 0.0007,
278
+ "loss": 3.396291732788086,
279
+ "step": 540
280
+ },
281
+ {
282
+ "epoch": 0.03707448601280755,
283
+ "eval_loss": 3.3581483364105225,
284
+ "eval_runtime": 15.3044,
285
+ "eval_samples_per_second": 622.502,
286
+ "eval_steps_per_second": 0.653,
287
+ "step": 550
288
+ },
289
+ {
290
+ "epoch": 0.03774856757667678,
291
+ "grad_norm": 0.47265625,
292
+ "learning_rate": 0.0007,
293
+ "loss": 3.347300720214844,
294
+ "step": 560
295
+ },
296
+ {
297
+ "epoch": 0.03909673070441524,
298
+ "grad_norm": 0.458984375,
299
+ "learning_rate": 0.0007,
300
+ "loss": 3.2885902404785154,
301
+ "step": 580
302
+ },
303
+ {
304
+ "epoch": 0.04044489383215369,
305
+ "grad_norm": 0.4375,
306
+ "learning_rate": 0.0007,
307
+ "loss": 3.25723876953125,
308
+ "step": 600
309
+ },
310
+ {
311
+ "epoch": 0.04044489383215369,
312
+ "eval_loss": 3.2229573726654053,
313
+ "eval_runtime": 15.2967,
314
+ "eval_samples_per_second": 622.812,
315
+ "eval_steps_per_second": 0.654,
316
+ "step": 600
317
+ },
318
+ {
319
+ "epoch": 0.04179305695989215,
320
+ "grad_norm": 0.458984375,
321
+ "learning_rate": 0.0007,
322
+ "loss": 3.1902334213256838,
323
+ "step": 620
324
+ },
325
+ {
326
+ "epoch": 0.043141220087630605,
327
+ "grad_norm": 0.40625,
328
+ "learning_rate": 0.0007,
329
+ "loss": 3.167861557006836,
330
+ "step": 640
331
+ },
332
+ {
333
+ "epoch": 0.043815301651499834,
334
+ "eval_loss": 3.120636463165283,
335
+ "eval_runtime": 15.3751,
336
+ "eval_samples_per_second": 619.639,
337
+ "eval_steps_per_second": 0.65,
338
+ "step": 650
339
+ },
340
+ {
341
+ "epoch": 0.044489383215369056,
342
+ "grad_norm": 0.494140625,
343
+ "learning_rate": 0.0007,
344
+ "loss": 3.1089324951171875,
345
+ "step": 660
346
+ },
347
+ {
348
+ "epoch": 0.045837546343107514,
349
+ "grad_norm": 0.435546875,
350
+ "learning_rate": 0.0007,
351
+ "loss": 3.0828268051147463,
352
+ "step": 680
353
+ },
354
+ {
355
+ "epoch": 0.04718570947084597,
356
+ "grad_norm": 0.427734375,
357
+ "learning_rate": 0.0007,
358
+ "loss": 3.0396488189697264,
359
+ "step": 700
360
+ },
361
+ {
362
+ "epoch": 0.04718570947084597,
363
+ "eval_loss": 3.0276408195495605,
364
+ "eval_runtime": 15.2735,
365
+ "eval_samples_per_second": 623.761,
366
+ "eval_steps_per_second": 0.655,
367
+ "step": 700
368
+ },
369
+ {
370
+ "epoch": 0.04853387259858443,
371
+ "grad_norm": 0.4375,
372
+ "learning_rate": 0.0007,
373
+ "loss": 3.0034805297851563,
374
+ "step": 720
375
+ },
376
+ {
377
+ "epoch": 0.04988203572632288,
378
+ "grad_norm": 0.52734375,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.96556396484375,
381
+ "step": 740
382
+ },
383
+ {
384
+ "epoch": 0.05055611729019211,
385
+ "eval_loss": 2.950009346008301,
386
+ "eval_runtime": 15.243,
387
+ "eval_samples_per_second": 625.009,
388
+ "eval_steps_per_second": 0.656,
389
+ "step": 750
390
+ },
391
+ {
392
+ "epoch": 0.05123019885406134,
393
+ "grad_norm": 0.451171875,
394
+ "learning_rate": 0.0007,
395
+ "loss": 2.950779151916504,
396
+ "step": 760
397
+ },
398
+ {
399
+ "epoch": 0.0525783619817998,
400
+ "grad_norm": 0.400390625,
401
+ "learning_rate": 0.0007,
402
+ "loss": 2.956203079223633,
403
+ "step": 780
404
+ },
405
+ {
406
+ "epoch": 0.053926525109538256,
407
+ "grad_norm": 0.40625,
408
+ "learning_rate": 0.0007,
409
+ "loss": 2.9017578125,
410
+ "step": 800
411
+ },
412
+ {
413
+ "epoch": 0.053926525109538256,
414
+ "eval_loss": 2.8844006061553955,
415
+ "eval_runtime": 15.2903,
416
+ "eval_samples_per_second": 623.073,
417
+ "eval_steps_per_second": 0.654,
418
+ "step": 800
419
+ },
420
+ {
421
+ "epoch": 0.05527468823727671,
422
+ "grad_norm": 0.41015625,
423
+ "learning_rate": 0.0007,
424
+ "loss": 2.887515640258789,
425
+ "step": 820
426
+ },
427
+ {
428
+ "epoch": 0.056622851365015166,
429
+ "grad_norm": 0.41796875,
430
+ "learning_rate": 0.0007,
431
+ "loss": 2.8556928634643555,
432
+ "step": 840
433
+ },
434
+ {
435
+ "epoch": 0.057296932928884395,
436
+ "eval_loss": 2.821610450744629,
437
+ "eval_runtime": 15.2253,
438
+ "eval_samples_per_second": 625.733,
439
+ "eval_steps_per_second": 0.657,
440
+ "step": 850
441
+ },
442
+ {
443
+ "epoch": 0.057971014492753624,
444
+ "grad_norm": 0.423828125,
445
+ "learning_rate": 0.0007,
446
+ "loss": 2.8130855560302734,
447
+ "step": 860
448
+ },
449
+ {
450
+ "epoch": 0.05931917762049208,
451
+ "grad_norm": 0.38671875,
452
+ "learning_rate": 0.0007,
453
+ "loss": 2.790329170227051,
454
+ "step": 880
455
+ },
456
+ {
457
+ "epoch": 0.06066734074823053,
458
+ "grad_norm": 0.427734375,
459
+ "learning_rate": 0.0007,
460
+ "loss": 2.7760162353515625,
461
+ "step": 900
462
+ },
463
+ {
464
+ "epoch": 0.06066734074823053,
465
+ "eval_loss": 2.7661373615264893,
466
+ "eval_runtime": 15.2736,
467
+ "eval_samples_per_second": 623.755,
468
+ "eval_steps_per_second": 0.655,
469
+ "step": 900
470
+ },
471
+ {
472
+ "epoch": 0.06201550387596899,
473
+ "grad_norm": 0.435546875,
474
+ "learning_rate": 0.0007,
475
+ "loss": 2.7462047576904296,
476
+ "step": 920
477
+ },
478
+ {
479
+ "epoch": 0.06336366700370745,
480
+ "grad_norm": 0.4140625,
481
+ "learning_rate": 0.0007,
482
+ "loss": 2.732274627685547,
483
+ "step": 940
484
+ },
485
+ {
486
+ "epoch": 0.06403774856757667,
487
+ "eval_loss": 2.713942289352417,
488
+ "eval_runtime": 15.6062,
489
+ "eval_samples_per_second": 610.463,
490
+ "eval_steps_per_second": 0.641,
491
+ "step": 950
492
+ },
493
+ {
494
+ "epoch": 0.06471183013144591,
495
+ "grad_norm": 0.419921875,
496
+ "learning_rate": 0.0007,
497
+ "loss": 2.705318069458008,
498
+ "step": 960
499
+ },
500
+ {
501
+ "epoch": 0.06605999325918437,
502
+ "grad_norm": 0.4296875,
503
+ "learning_rate": 0.0007,
504
+ "loss": 2.6892818450927733,
505
+ "step": 980
506
+ },
507
+ {
508
+ "epoch": 0.06740815638692282,
509
+ "grad_norm": 0.4453125,
510
+ "learning_rate": 0.0007,
511
+ "loss": 2.687115287780762,
512
+ "step": 1000
513
+ },
514
+ {
515
+ "epoch": 0.06740815638692282,
516
+ "eval_loss": 2.668647289276123,
517
+ "eval_runtime": 15.2597,
518
+ "eval_samples_per_second": 624.326,
519
+ "eval_steps_per_second": 0.655,
520
+ "step": 1000
521
+ },
522
+ {
523
+ "epoch": 0.06875631951466127,
524
+ "grad_norm": 0.388671875,
525
+ "learning_rate": 0.0007,
526
+ "loss": 2.6533071517944338,
527
+ "step": 1020
528
+ },
529
+ {
530
+ "epoch": 0.07010448264239973,
531
+ "grad_norm": 0.392578125,
532
+ "learning_rate": 0.0007,
533
+ "loss": 2.6390922546386717,
534
+ "step": 1040
535
+ },
536
+ {
537
+ "epoch": 0.07077856420626896,
538
+ "eval_loss": 2.6291816234588623,
539
+ "eval_runtime": 15.2457,
540
+ "eval_samples_per_second": 624.896,
541
+ "eval_steps_per_second": 0.656,
542
+ "step": 1050
543
+ },
544
+ {
545
+ "epoch": 0.07145264577013818,
546
+ "grad_norm": 0.4375,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.624461555480957,
549
+ "step": 1060
550
+ },
551
+ {
552
+ "epoch": 0.07280080889787664,
553
+ "grad_norm": 0.396484375,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.6044427871704103,
556
+ "step": 1080
557
+ },
558
+ {
559
+ "epoch": 0.0741489720256151,
560
+ "grad_norm": 0.423828125,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.597646713256836,
563
+ "step": 1100
564
+ },
565
+ {
566
+ "epoch": 0.0741489720256151,
567
+ "eval_loss": 2.589696168899536,
568
+ "eval_runtime": 15.3967,
569
+ "eval_samples_per_second": 618.768,
570
+ "eval_steps_per_second": 0.649,
571
+ "step": 1100
572
+ }
573
+ ],
574
+ "logging_steps": 20,
575
+ "max_steps": 2500,
576
+ "num_input_tokens_seen": 0,
577
+ "num_train_epochs": 1,
578
+ "save_steps": 100,
579
+ "stateful_callbacks": {
580
+ "TrainerControl": {
581
+ "args": {
582
+ "should_epoch_stop": false,
583
+ "should_evaluate": false,
584
+ "should_log": false,
585
+ "should_save": true,
586
+ "should_training_stop": false
587
+ },
588
+ "attributes": {}
589
+ }
590
+ },
591
+ "total_flos": 3548912182886400.0,
592
+ "train_batch_size": 64,
593
+ "trial_name": null,
594
+ "trial_params": null
595
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1100/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd90512e56f813612dc668676d6f8cab72da110a42f23c61cafebf91064625a1
3
+ size 4920
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "silu",
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": 100,
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.16.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096,
35
+ "waleed_beta": 10.0
36
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ebf616fa48e557d8696de15b065053ed846c9e3655f30e198e0ab338083693
3
+ size 33967272
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0c7e1bf909a08b8dcf5e0614a3c1728439257cb8c7f6501453301b625539604
3
+ size 68504996
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9022ea8893d375dc2079ce97b2df7f3286392f4c409461865c91e578acc3ba10
3
+ size 14244
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8543afc9166865219cadbba29db865d839ec4102a4f9ddeffc962b9fdba3b028
3
+ size 1064
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/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
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/trainer_state.json ADDED
@@ -0,0 +1,646 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.08088978766430738,
6
+ "eval_steps": 50,
7
+ "global_step": 1200,
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": 2.078125,
15
+ "learning_rate": 2.66e-05,
16
+ "loss": 8.29962387084961,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.3515625,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047650909423828,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.003370407819346141,
28
+ "eval_loss": 7.835535049438477,
29
+ "eval_runtime": 15.2165,
30
+ "eval_samples_per_second": 626.095,
31
+ "eval_steps_per_second": 0.657,
32
+ "step": 50
33
+ },
34
+ {
35
+ "epoch": 0.004044489383215369,
36
+ "grad_norm": 1.2734375,
37
+ "learning_rate": 8.259999999999999e-05,
38
+ "loss": 7.8378746032714846,
39
+ "step": 60
40
+ },
41
+ {
42
+ "epoch": 0.005392652510953826,
43
+ "grad_norm": 1.265625,
44
+ "learning_rate": 0.0001106,
45
+ "loss": 7.5728271484375,
46
+ "step": 80
47
+ },
48
+ {
49
+ "epoch": 0.006740815638692282,
50
+ "grad_norm": 1.1953125,
51
+ "learning_rate": 0.0001386,
52
+ "loss": 7.2332916259765625,
53
+ "step": 100
54
+ },
55
+ {
56
+ "epoch": 0.006740815638692282,
57
+ "eval_loss": 7.039053440093994,
58
+ "eval_runtime": 15.2074,
59
+ "eval_samples_per_second": 626.471,
60
+ "eval_steps_per_second": 0.658,
61
+ "step": 100
62
+ },
63
+ {
64
+ "epoch": 0.008088978766430738,
65
+ "grad_norm": 1.1328125,
66
+ "learning_rate": 0.00016659999999999998,
67
+ "loss": 6.8608039855957035,
68
+ "step": 120
69
+ },
70
+ {
71
+ "epoch": 0.009437141894169195,
72
+ "grad_norm": 0.94921875,
73
+ "learning_rate": 0.00019460000000000001,
74
+ "loss": 6.517842864990234,
75
+ "step": 140
76
+ },
77
+ {
78
+ "epoch": 0.010111223458038422,
79
+ "eval_loss": 6.229617595672607,
80
+ "eval_runtime": 15.1978,
81
+ "eval_samples_per_second": 626.867,
82
+ "eval_steps_per_second": 0.658,
83
+ "step": 150
84
+ },
85
+ {
86
+ "epoch": 0.010785305021907651,
87
+ "grad_norm": 0.80078125,
88
+ "learning_rate": 0.0002226,
89
+ "loss": 6.2399169921875,
90
+ "step": 160
91
+ },
92
+ {
93
+ "epoch": 0.012133468149646108,
94
+ "grad_norm": 0.6640625,
95
+ "learning_rate": 0.00025059999999999997,
96
+ "loss": 6.036548614501953,
97
+ "step": 180
98
+ },
99
+ {
100
+ "epoch": 0.013481631277384564,
101
+ "grad_norm": 0.61328125,
102
+ "learning_rate": 0.0002786,
103
+ "loss": 5.857078552246094,
104
+ "step": 200
105
+ },
106
+ {
107
+ "epoch": 0.013481631277384564,
108
+ "eval_loss": 5.765108585357666,
109
+ "eval_runtime": 15.5271,
110
+ "eval_samples_per_second": 613.574,
111
+ "eval_steps_per_second": 0.644,
112
+ "step": 200
113
+ },
114
+ {
115
+ "epoch": 0.01482979440512302,
116
+ "grad_norm": 0.61328125,
117
+ "learning_rate": 0.00030659999999999997,
118
+ "loss": 5.680564117431641,
119
+ "step": 220
120
+ },
121
+ {
122
+ "epoch": 0.016177957532861477,
123
+ "grad_norm": 2.0625,
124
+ "learning_rate": 0.0003346,
125
+ "loss": 5.472259902954102,
126
+ "step": 240
127
+ },
128
+ {
129
+ "epoch": 0.016852039096730706,
130
+ "eval_loss": 5.240335464477539,
131
+ "eval_runtime": 15.2596,
132
+ "eval_samples_per_second": 624.33,
133
+ "eval_steps_per_second": 0.655,
134
+ "step": 250
135
+ },
136
+ {
137
+ "epoch": 0.01752612066059993,
138
+ "grad_norm": 1.7421875,
139
+ "learning_rate": 0.00036260000000000003,
140
+ "loss": 5.24955940246582,
141
+ "step": 260
142
+ },
143
+ {
144
+ "epoch": 0.01887428378833839,
145
+ "grad_norm": 0.62109375,
146
+ "learning_rate": 0.0003906,
147
+ "loss": 4.995759963989258,
148
+ "step": 280
149
+ },
150
+ {
151
+ "epoch": 0.020222446916076844,
152
+ "grad_norm": 1.390625,
153
+ "learning_rate": 0.0004186,
154
+ "loss": 4.769353866577148,
155
+ "step": 300
156
+ },
157
+ {
158
+ "epoch": 0.020222446916076844,
159
+ "eval_loss": 4.669629096984863,
160
+ "eval_runtime": 15.2595,
161
+ "eval_samples_per_second": 624.332,
162
+ "eval_steps_per_second": 0.655,
163
+ "step": 300
164
+ },
165
+ {
166
+ "epoch": 0.021570610043815303,
167
+ "grad_norm": 0.73828125,
168
+ "learning_rate": 0.0004466,
169
+ "loss": 4.569867324829102,
170
+ "step": 320
171
+ },
172
+ {
173
+ "epoch": 0.022918773171553757,
174
+ "grad_norm": 0.765625,
175
+ "learning_rate": 0.00047460000000000004,
176
+ "loss": 4.389446258544922,
177
+ "step": 340
178
+ },
179
+ {
180
+ "epoch": 0.023592854735422986,
181
+ "eval_loss": 4.231207847595215,
182
+ "eval_runtime": 15.2143,
183
+ "eval_samples_per_second": 626.188,
184
+ "eval_steps_per_second": 0.657,
185
+ "step": 350
186
+ },
187
+ {
188
+ "epoch": 0.024266936299292215,
189
+ "grad_norm": 1.2421875,
190
+ "learning_rate": 0.0005026,
191
+ "loss": 4.243848419189453,
192
+ "step": 360
193
+ },
194
+ {
195
+ "epoch": 0.02561509942703067,
196
+ "grad_norm": 0.7109375,
197
+ "learning_rate": 0.0005306,
198
+ "loss": 4.1090648651123045,
199
+ "step": 380
200
+ },
201
+ {
202
+ "epoch": 0.026963262554769128,
203
+ "grad_norm": 0.484375,
204
+ "learning_rate": 0.0005586,
205
+ "loss": 3.966172790527344,
206
+ "step": 400
207
+ },
208
+ {
209
+ "epoch": 0.026963262554769128,
210
+ "eval_loss": 3.923773765563965,
211
+ "eval_runtime": 15.5308,
212
+ "eval_samples_per_second": 613.426,
213
+ "eval_steps_per_second": 0.644,
214
+ "step": 400
215
+ },
216
+ {
217
+ "epoch": 0.028311425682507583,
218
+ "grad_norm": 0.6640625,
219
+ "learning_rate": 0.0005866,
220
+ "loss": 3.8559597015380858,
221
+ "step": 420
222
+ },
223
+ {
224
+ "epoch": 0.02965958881024604,
225
+ "grad_norm": 0.515625,
226
+ "learning_rate": 0.0006146,
227
+ "loss": 3.7943866729736326,
228
+ "step": 440
229
+ },
230
+ {
231
+ "epoch": 0.030333670374115267,
232
+ "eval_loss": 3.6977553367614746,
233
+ "eval_runtime": 15.2001,
234
+ "eval_samples_per_second": 626.77,
235
+ "eval_steps_per_second": 0.658,
236
+ "step": 450
237
+ },
238
+ {
239
+ "epoch": 0.031007751937984496,
240
+ "grad_norm": 0.51953125,
241
+ "learning_rate": 0.0006426,
242
+ "loss": 3.686443328857422,
243
+ "step": 460
244
+ },
245
+ {
246
+ "epoch": 0.032355915065722954,
247
+ "grad_norm": 0.62109375,
248
+ "learning_rate": 0.0006705999999999999,
249
+ "loss": 3.6123157501220704,
250
+ "step": 480
251
+ },
252
+ {
253
+ "epoch": 0.03370407819346141,
254
+ "grad_norm": 0.70703125,
255
+ "learning_rate": 0.0006986,
256
+ "loss": 3.542654800415039,
257
+ "step": 500
258
+ },
259
+ {
260
+ "epoch": 0.03370407819346141,
261
+ "eval_loss": 3.513820171356201,
262
+ "eval_runtime": 15.2193,
263
+ "eval_samples_per_second": 625.983,
264
+ "eval_steps_per_second": 0.657,
265
+ "step": 500
266
+ },
267
+ {
268
+ "epoch": 0.03505224132119986,
269
+ "grad_norm": 0.48046875,
270
+ "learning_rate": 0.0007,
271
+ "loss": 3.472199249267578,
272
+ "step": 520
273
+ },
274
+ {
275
+ "epoch": 0.03640040444893832,
276
+ "grad_norm": 0.53515625,
277
+ "learning_rate": 0.0007,
278
+ "loss": 3.396291732788086,
279
+ "step": 540
280
+ },
281
+ {
282
+ "epoch": 0.03707448601280755,
283
+ "eval_loss": 3.3581483364105225,
284
+ "eval_runtime": 15.3044,
285
+ "eval_samples_per_second": 622.502,
286
+ "eval_steps_per_second": 0.653,
287
+ "step": 550
288
+ },
289
+ {
290
+ "epoch": 0.03774856757667678,
291
+ "grad_norm": 0.47265625,
292
+ "learning_rate": 0.0007,
293
+ "loss": 3.347300720214844,
294
+ "step": 560
295
+ },
296
+ {
297
+ "epoch": 0.03909673070441524,
298
+ "grad_norm": 0.458984375,
299
+ "learning_rate": 0.0007,
300
+ "loss": 3.2885902404785154,
301
+ "step": 580
302
+ },
303
+ {
304
+ "epoch": 0.04044489383215369,
305
+ "grad_norm": 0.4375,
306
+ "learning_rate": 0.0007,
307
+ "loss": 3.25723876953125,
308
+ "step": 600
309
+ },
310
+ {
311
+ "epoch": 0.04044489383215369,
312
+ "eval_loss": 3.2229573726654053,
313
+ "eval_runtime": 15.2967,
314
+ "eval_samples_per_second": 622.812,
315
+ "eval_steps_per_second": 0.654,
316
+ "step": 600
317
+ },
318
+ {
319
+ "epoch": 0.04179305695989215,
320
+ "grad_norm": 0.458984375,
321
+ "learning_rate": 0.0007,
322
+ "loss": 3.1902334213256838,
323
+ "step": 620
324
+ },
325
+ {
326
+ "epoch": 0.043141220087630605,
327
+ "grad_norm": 0.40625,
328
+ "learning_rate": 0.0007,
329
+ "loss": 3.167861557006836,
330
+ "step": 640
331
+ },
332
+ {
333
+ "epoch": 0.043815301651499834,
334
+ "eval_loss": 3.120636463165283,
335
+ "eval_runtime": 15.3751,
336
+ "eval_samples_per_second": 619.639,
337
+ "eval_steps_per_second": 0.65,
338
+ "step": 650
339
+ },
340
+ {
341
+ "epoch": 0.044489383215369056,
342
+ "grad_norm": 0.494140625,
343
+ "learning_rate": 0.0007,
344
+ "loss": 3.1089324951171875,
345
+ "step": 660
346
+ },
347
+ {
348
+ "epoch": 0.045837546343107514,
349
+ "grad_norm": 0.435546875,
350
+ "learning_rate": 0.0007,
351
+ "loss": 3.0828268051147463,
352
+ "step": 680
353
+ },
354
+ {
355
+ "epoch": 0.04718570947084597,
356
+ "grad_norm": 0.427734375,
357
+ "learning_rate": 0.0007,
358
+ "loss": 3.0396488189697264,
359
+ "step": 700
360
+ },
361
+ {
362
+ "epoch": 0.04718570947084597,
363
+ "eval_loss": 3.0276408195495605,
364
+ "eval_runtime": 15.2735,
365
+ "eval_samples_per_second": 623.761,
366
+ "eval_steps_per_second": 0.655,
367
+ "step": 700
368
+ },
369
+ {
370
+ "epoch": 0.04853387259858443,
371
+ "grad_norm": 0.4375,
372
+ "learning_rate": 0.0007,
373
+ "loss": 3.0034805297851563,
374
+ "step": 720
375
+ },
376
+ {
377
+ "epoch": 0.04988203572632288,
378
+ "grad_norm": 0.52734375,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.96556396484375,
381
+ "step": 740
382
+ },
383
+ {
384
+ "epoch": 0.05055611729019211,
385
+ "eval_loss": 2.950009346008301,
386
+ "eval_runtime": 15.243,
387
+ "eval_samples_per_second": 625.009,
388
+ "eval_steps_per_second": 0.656,
389
+ "step": 750
390
+ },
391
+ {
392
+ "epoch": 0.05123019885406134,
393
+ "grad_norm": 0.451171875,
394
+ "learning_rate": 0.0007,
395
+ "loss": 2.950779151916504,
396
+ "step": 760
397
+ },
398
+ {
399
+ "epoch": 0.0525783619817998,
400
+ "grad_norm": 0.400390625,
401
+ "learning_rate": 0.0007,
402
+ "loss": 2.956203079223633,
403
+ "step": 780
404
+ },
405
+ {
406
+ "epoch": 0.053926525109538256,
407
+ "grad_norm": 0.40625,
408
+ "learning_rate": 0.0007,
409
+ "loss": 2.9017578125,
410
+ "step": 800
411
+ },
412
+ {
413
+ "epoch": 0.053926525109538256,
414
+ "eval_loss": 2.8844006061553955,
415
+ "eval_runtime": 15.2903,
416
+ "eval_samples_per_second": 623.073,
417
+ "eval_steps_per_second": 0.654,
418
+ "step": 800
419
+ },
420
+ {
421
+ "epoch": 0.05527468823727671,
422
+ "grad_norm": 0.41015625,
423
+ "learning_rate": 0.0007,
424
+ "loss": 2.887515640258789,
425
+ "step": 820
426
+ },
427
+ {
428
+ "epoch": 0.056622851365015166,
429
+ "grad_norm": 0.41796875,
430
+ "learning_rate": 0.0007,
431
+ "loss": 2.8556928634643555,
432
+ "step": 840
433
+ },
434
+ {
435
+ "epoch": 0.057296932928884395,
436
+ "eval_loss": 2.821610450744629,
437
+ "eval_runtime": 15.2253,
438
+ "eval_samples_per_second": 625.733,
439
+ "eval_steps_per_second": 0.657,
440
+ "step": 850
441
+ },
442
+ {
443
+ "epoch": 0.057971014492753624,
444
+ "grad_norm": 0.423828125,
445
+ "learning_rate": 0.0007,
446
+ "loss": 2.8130855560302734,
447
+ "step": 860
448
+ },
449
+ {
450
+ "epoch": 0.05931917762049208,
451
+ "grad_norm": 0.38671875,
452
+ "learning_rate": 0.0007,
453
+ "loss": 2.790329170227051,
454
+ "step": 880
455
+ },
456
+ {
457
+ "epoch": 0.06066734074823053,
458
+ "grad_norm": 0.427734375,
459
+ "learning_rate": 0.0007,
460
+ "loss": 2.7760162353515625,
461
+ "step": 900
462
+ },
463
+ {
464
+ "epoch": 0.06066734074823053,
465
+ "eval_loss": 2.7661373615264893,
466
+ "eval_runtime": 15.2736,
467
+ "eval_samples_per_second": 623.755,
468
+ "eval_steps_per_second": 0.655,
469
+ "step": 900
470
+ },
471
+ {
472
+ "epoch": 0.06201550387596899,
473
+ "grad_norm": 0.435546875,
474
+ "learning_rate": 0.0007,
475
+ "loss": 2.7462047576904296,
476
+ "step": 920
477
+ },
478
+ {
479
+ "epoch": 0.06336366700370745,
480
+ "grad_norm": 0.4140625,
481
+ "learning_rate": 0.0007,
482
+ "loss": 2.732274627685547,
483
+ "step": 940
484
+ },
485
+ {
486
+ "epoch": 0.06403774856757667,
487
+ "eval_loss": 2.713942289352417,
488
+ "eval_runtime": 15.6062,
489
+ "eval_samples_per_second": 610.463,
490
+ "eval_steps_per_second": 0.641,
491
+ "step": 950
492
+ },
493
+ {
494
+ "epoch": 0.06471183013144591,
495
+ "grad_norm": 0.419921875,
496
+ "learning_rate": 0.0007,
497
+ "loss": 2.705318069458008,
498
+ "step": 960
499
+ },
500
+ {
501
+ "epoch": 0.06605999325918437,
502
+ "grad_norm": 0.4296875,
503
+ "learning_rate": 0.0007,
504
+ "loss": 2.6892818450927733,
505
+ "step": 980
506
+ },
507
+ {
508
+ "epoch": 0.06740815638692282,
509
+ "grad_norm": 0.4453125,
510
+ "learning_rate": 0.0007,
511
+ "loss": 2.687115287780762,
512
+ "step": 1000
513
+ },
514
+ {
515
+ "epoch": 0.06740815638692282,
516
+ "eval_loss": 2.668647289276123,
517
+ "eval_runtime": 15.2597,
518
+ "eval_samples_per_second": 624.326,
519
+ "eval_steps_per_second": 0.655,
520
+ "step": 1000
521
+ },
522
+ {
523
+ "epoch": 0.06875631951466127,
524
+ "grad_norm": 0.388671875,
525
+ "learning_rate": 0.0007,
526
+ "loss": 2.6533071517944338,
527
+ "step": 1020
528
+ },
529
+ {
530
+ "epoch": 0.07010448264239973,
531
+ "grad_norm": 0.392578125,
532
+ "learning_rate": 0.0007,
533
+ "loss": 2.6390922546386717,
534
+ "step": 1040
535
+ },
536
+ {
537
+ "epoch": 0.07077856420626896,
538
+ "eval_loss": 2.6291816234588623,
539
+ "eval_runtime": 15.2457,
540
+ "eval_samples_per_second": 624.896,
541
+ "eval_steps_per_second": 0.656,
542
+ "step": 1050
543
+ },
544
+ {
545
+ "epoch": 0.07145264577013818,
546
+ "grad_norm": 0.4375,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.624461555480957,
549
+ "step": 1060
550
+ },
551
+ {
552
+ "epoch": 0.07280080889787664,
553
+ "grad_norm": 0.396484375,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.6044427871704103,
556
+ "step": 1080
557
+ },
558
+ {
559
+ "epoch": 0.0741489720256151,
560
+ "grad_norm": 0.423828125,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.597646713256836,
563
+ "step": 1100
564
+ },
565
+ {
566
+ "epoch": 0.0741489720256151,
567
+ "eval_loss": 2.589696168899536,
568
+ "eval_runtime": 15.3967,
569
+ "eval_samples_per_second": 618.768,
570
+ "eval_steps_per_second": 0.649,
571
+ "step": 1100
572
+ },
573
+ {
574
+ "epoch": 0.07549713515335356,
575
+ "grad_norm": 0.369140625,
576
+ "learning_rate": 0.0007,
577
+ "loss": 2.561947250366211,
578
+ "step": 1120
579
+ },
580
+ {
581
+ "epoch": 0.07684529828109202,
582
+ "grad_norm": 0.390625,
583
+ "learning_rate": 0.0007,
584
+ "loss": 2.54862117767334,
585
+ "step": 1140
586
+ },
587
+ {
588
+ "epoch": 0.07751937984496124,
589
+ "eval_loss": 2.559185743331909,
590
+ "eval_runtime": 15.2877,
591
+ "eval_samples_per_second": 623.181,
592
+ "eval_steps_per_second": 0.654,
593
+ "step": 1150
594
+ },
595
+ {
596
+ "epoch": 0.07819346140883048,
597
+ "grad_norm": 0.37890625,
598
+ "learning_rate": 0.0007,
599
+ "loss": 2.5469717025756835,
600
+ "step": 1160
601
+ },
602
+ {
603
+ "epoch": 0.07954162453656892,
604
+ "grad_norm": 0.400390625,
605
+ "learning_rate": 0.0007,
606
+ "loss": 2.540451240539551,
607
+ "step": 1180
608
+ },
609
+ {
610
+ "epoch": 0.08088978766430738,
611
+ "grad_norm": 0.40625,
612
+ "learning_rate": 0.0007,
613
+ "loss": 2.53134708404541,
614
+ "step": 1200
615
+ },
616
+ {
617
+ "epoch": 0.08088978766430738,
618
+ "eval_loss": 2.521665096282959,
619
+ "eval_runtime": 15.2822,
620
+ "eval_samples_per_second": 623.407,
621
+ "eval_steps_per_second": 0.654,
622
+ "step": 1200
623
+ }
624
+ ],
625
+ "logging_steps": 20,
626
+ "max_steps": 2500,
627
+ "num_input_tokens_seen": 0,
628
+ "num_train_epochs": 1,
629
+ "save_steps": 100,
630
+ "stateful_callbacks": {
631
+ "TrainerControl": {
632
+ "args": {
633
+ "should_epoch_stop": false,
634
+ "should_evaluate": false,
635
+ "should_log": false,
636
+ "should_save": true,
637
+ "should_training_stop": false
638
+ },
639
+ "attributes": {}
640
+ }
641
+ },
642
+ "total_flos": 3871540563148800.0,
643
+ "train_batch_size": 64,
644
+ "trial_name": null,
645
+ "trial_params": null
646
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1200/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd90512e56f813612dc668676d6f8cab72da110a42f23c61cafebf91064625a1
3
+ size 4920
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "silu",
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": 100,
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.16.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096,
35
+ "waleed_beta": 10.0
36
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61dfd4dfc2ba6aa8d53a70cfd85ba33c7a4eecef02e2504130589ca4b138aa17
3
+ size 33967272
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5743abb221137552f627f64bed5dec0fdc32d97604516f5c2afbbdb05cde6bf3
3
+ size 68504996
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b91d54eec621bed16563c41eeeb5b197c78337576ee4a7418e37bfff54707b1b
3
+ size 14244
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6166e4e25b157aeaf4765c80f04dab16e63f8ec3cd6ab3637d77b940e7dfcfe1
3
+ size 1064
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/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
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/trainer_state.json ADDED
@@ -0,0 +1,697 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.08763060330299967,
6
+ "eval_steps": 50,
7
+ "global_step": 1300,
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": 2.078125,
15
+ "learning_rate": 2.66e-05,
16
+ "loss": 8.29962387084961,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.3515625,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047650909423828,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.003370407819346141,
28
+ "eval_loss": 7.835535049438477,
29
+ "eval_runtime": 15.2165,
30
+ "eval_samples_per_second": 626.095,
31
+ "eval_steps_per_second": 0.657,
32
+ "step": 50
33
+ },
34
+ {
35
+ "epoch": 0.004044489383215369,
36
+ "grad_norm": 1.2734375,
37
+ "learning_rate": 8.259999999999999e-05,
38
+ "loss": 7.8378746032714846,
39
+ "step": 60
40
+ },
41
+ {
42
+ "epoch": 0.005392652510953826,
43
+ "grad_norm": 1.265625,
44
+ "learning_rate": 0.0001106,
45
+ "loss": 7.5728271484375,
46
+ "step": 80
47
+ },
48
+ {
49
+ "epoch": 0.006740815638692282,
50
+ "grad_norm": 1.1953125,
51
+ "learning_rate": 0.0001386,
52
+ "loss": 7.2332916259765625,
53
+ "step": 100
54
+ },
55
+ {
56
+ "epoch": 0.006740815638692282,
57
+ "eval_loss": 7.039053440093994,
58
+ "eval_runtime": 15.2074,
59
+ "eval_samples_per_second": 626.471,
60
+ "eval_steps_per_second": 0.658,
61
+ "step": 100
62
+ },
63
+ {
64
+ "epoch": 0.008088978766430738,
65
+ "grad_norm": 1.1328125,
66
+ "learning_rate": 0.00016659999999999998,
67
+ "loss": 6.8608039855957035,
68
+ "step": 120
69
+ },
70
+ {
71
+ "epoch": 0.009437141894169195,
72
+ "grad_norm": 0.94921875,
73
+ "learning_rate": 0.00019460000000000001,
74
+ "loss": 6.517842864990234,
75
+ "step": 140
76
+ },
77
+ {
78
+ "epoch": 0.010111223458038422,
79
+ "eval_loss": 6.229617595672607,
80
+ "eval_runtime": 15.1978,
81
+ "eval_samples_per_second": 626.867,
82
+ "eval_steps_per_second": 0.658,
83
+ "step": 150
84
+ },
85
+ {
86
+ "epoch": 0.010785305021907651,
87
+ "grad_norm": 0.80078125,
88
+ "learning_rate": 0.0002226,
89
+ "loss": 6.2399169921875,
90
+ "step": 160
91
+ },
92
+ {
93
+ "epoch": 0.012133468149646108,
94
+ "grad_norm": 0.6640625,
95
+ "learning_rate": 0.00025059999999999997,
96
+ "loss": 6.036548614501953,
97
+ "step": 180
98
+ },
99
+ {
100
+ "epoch": 0.013481631277384564,
101
+ "grad_norm": 0.61328125,
102
+ "learning_rate": 0.0002786,
103
+ "loss": 5.857078552246094,
104
+ "step": 200
105
+ },
106
+ {
107
+ "epoch": 0.013481631277384564,
108
+ "eval_loss": 5.765108585357666,
109
+ "eval_runtime": 15.5271,
110
+ "eval_samples_per_second": 613.574,
111
+ "eval_steps_per_second": 0.644,
112
+ "step": 200
113
+ },
114
+ {
115
+ "epoch": 0.01482979440512302,
116
+ "grad_norm": 0.61328125,
117
+ "learning_rate": 0.00030659999999999997,
118
+ "loss": 5.680564117431641,
119
+ "step": 220
120
+ },
121
+ {
122
+ "epoch": 0.016177957532861477,
123
+ "grad_norm": 2.0625,
124
+ "learning_rate": 0.0003346,
125
+ "loss": 5.472259902954102,
126
+ "step": 240
127
+ },
128
+ {
129
+ "epoch": 0.016852039096730706,
130
+ "eval_loss": 5.240335464477539,
131
+ "eval_runtime": 15.2596,
132
+ "eval_samples_per_second": 624.33,
133
+ "eval_steps_per_second": 0.655,
134
+ "step": 250
135
+ },
136
+ {
137
+ "epoch": 0.01752612066059993,
138
+ "grad_norm": 1.7421875,
139
+ "learning_rate": 0.00036260000000000003,
140
+ "loss": 5.24955940246582,
141
+ "step": 260
142
+ },
143
+ {
144
+ "epoch": 0.01887428378833839,
145
+ "grad_norm": 0.62109375,
146
+ "learning_rate": 0.0003906,
147
+ "loss": 4.995759963989258,
148
+ "step": 280
149
+ },
150
+ {
151
+ "epoch": 0.020222446916076844,
152
+ "grad_norm": 1.390625,
153
+ "learning_rate": 0.0004186,
154
+ "loss": 4.769353866577148,
155
+ "step": 300
156
+ },
157
+ {
158
+ "epoch": 0.020222446916076844,
159
+ "eval_loss": 4.669629096984863,
160
+ "eval_runtime": 15.2595,
161
+ "eval_samples_per_second": 624.332,
162
+ "eval_steps_per_second": 0.655,
163
+ "step": 300
164
+ },
165
+ {
166
+ "epoch": 0.021570610043815303,
167
+ "grad_norm": 0.73828125,
168
+ "learning_rate": 0.0004466,
169
+ "loss": 4.569867324829102,
170
+ "step": 320
171
+ },
172
+ {
173
+ "epoch": 0.022918773171553757,
174
+ "grad_norm": 0.765625,
175
+ "learning_rate": 0.00047460000000000004,
176
+ "loss": 4.389446258544922,
177
+ "step": 340
178
+ },
179
+ {
180
+ "epoch": 0.023592854735422986,
181
+ "eval_loss": 4.231207847595215,
182
+ "eval_runtime": 15.2143,
183
+ "eval_samples_per_second": 626.188,
184
+ "eval_steps_per_second": 0.657,
185
+ "step": 350
186
+ },
187
+ {
188
+ "epoch": 0.024266936299292215,
189
+ "grad_norm": 1.2421875,
190
+ "learning_rate": 0.0005026,
191
+ "loss": 4.243848419189453,
192
+ "step": 360
193
+ },
194
+ {
195
+ "epoch": 0.02561509942703067,
196
+ "grad_norm": 0.7109375,
197
+ "learning_rate": 0.0005306,
198
+ "loss": 4.1090648651123045,
199
+ "step": 380
200
+ },
201
+ {
202
+ "epoch": 0.026963262554769128,
203
+ "grad_norm": 0.484375,
204
+ "learning_rate": 0.0005586,
205
+ "loss": 3.966172790527344,
206
+ "step": 400
207
+ },
208
+ {
209
+ "epoch": 0.026963262554769128,
210
+ "eval_loss": 3.923773765563965,
211
+ "eval_runtime": 15.5308,
212
+ "eval_samples_per_second": 613.426,
213
+ "eval_steps_per_second": 0.644,
214
+ "step": 400
215
+ },
216
+ {
217
+ "epoch": 0.028311425682507583,
218
+ "grad_norm": 0.6640625,
219
+ "learning_rate": 0.0005866,
220
+ "loss": 3.8559597015380858,
221
+ "step": 420
222
+ },
223
+ {
224
+ "epoch": 0.02965958881024604,
225
+ "grad_norm": 0.515625,
226
+ "learning_rate": 0.0006146,
227
+ "loss": 3.7943866729736326,
228
+ "step": 440
229
+ },
230
+ {
231
+ "epoch": 0.030333670374115267,
232
+ "eval_loss": 3.6977553367614746,
233
+ "eval_runtime": 15.2001,
234
+ "eval_samples_per_second": 626.77,
235
+ "eval_steps_per_second": 0.658,
236
+ "step": 450
237
+ },
238
+ {
239
+ "epoch": 0.031007751937984496,
240
+ "grad_norm": 0.51953125,
241
+ "learning_rate": 0.0006426,
242
+ "loss": 3.686443328857422,
243
+ "step": 460
244
+ },
245
+ {
246
+ "epoch": 0.032355915065722954,
247
+ "grad_norm": 0.62109375,
248
+ "learning_rate": 0.0006705999999999999,
249
+ "loss": 3.6123157501220704,
250
+ "step": 480
251
+ },
252
+ {
253
+ "epoch": 0.03370407819346141,
254
+ "grad_norm": 0.70703125,
255
+ "learning_rate": 0.0006986,
256
+ "loss": 3.542654800415039,
257
+ "step": 500
258
+ },
259
+ {
260
+ "epoch": 0.03370407819346141,
261
+ "eval_loss": 3.513820171356201,
262
+ "eval_runtime": 15.2193,
263
+ "eval_samples_per_second": 625.983,
264
+ "eval_steps_per_second": 0.657,
265
+ "step": 500
266
+ },
267
+ {
268
+ "epoch": 0.03505224132119986,
269
+ "grad_norm": 0.48046875,
270
+ "learning_rate": 0.0007,
271
+ "loss": 3.472199249267578,
272
+ "step": 520
273
+ },
274
+ {
275
+ "epoch": 0.03640040444893832,
276
+ "grad_norm": 0.53515625,
277
+ "learning_rate": 0.0007,
278
+ "loss": 3.396291732788086,
279
+ "step": 540
280
+ },
281
+ {
282
+ "epoch": 0.03707448601280755,
283
+ "eval_loss": 3.3581483364105225,
284
+ "eval_runtime": 15.3044,
285
+ "eval_samples_per_second": 622.502,
286
+ "eval_steps_per_second": 0.653,
287
+ "step": 550
288
+ },
289
+ {
290
+ "epoch": 0.03774856757667678,
291
+ "grad_norm": 0.47265625,
292
+ "learning_rate": 0.0007,
293
+ "loss": 3.347300720214844,
294
+ "step": 560
295
+ },
296
+ {
297
+ "epoch": 0.03909673070441524,
298
+ "grad_norm": 0.458984375,
299
+ "learning_rate": 0.0007,
300
+ "loss": 3.2885902404785154,
301
+ "step": 580
302
+ },
303
+ {
304
+ "epoch": 0.04044489383215369,
305
+ "grad_norm": 0.4375,
306
+ "learning_rate": 0.0007,
307
+ "loss": 3.25723876953125,
308
+ "step": 600
309
+ },
310
+ {
311
+ "epoch": 0.04044489383215369,
312
+ "eval_loss": 3.2229573726654053,
313
+ "eval_runtime": 15.2967,
314
+ "eval_samples_per_second": 622.812,
315
+ "eval_steps_per_second": 0.654,
316
+ "step": 600
317
+ },
318
+ {
319
+ "epoch": 0.04179305695989215,
320
+ "grad_norm": 0.458984375,
321
+ "learning_rate": 0.0007,
322
+ "loss": 3.1902334213256838,
323
+ "step": 620
324
+ },
325
+ {
326
+ "epoch": 0.043141220087630605,
327
+ "grad_norm": 0.40625,
328
+ "learning_rate": 0.0007,
329
+ "loss": 3.167861557006836,
330
+ "step": 640
331
+ },
332
+ {
333
+ "epoch": 0.043815301651499834,
334
+ "eval_loss": 3.120636463165283,
335
+ "eval_runtime": 15.3751,
336
+ "eval_samples_per_second": 619.639,
337
+ "eval_steps_per_second": 0.65,
338
+ "step": 650
339
+ },
340
+ {
341
+ "epoch": 0.044489383215369056,
342
+ "grad_norm": 0.494140625,
343
+ "learning_rate": 0.0007,
344
+ "loss": 3.1089324951171875,
345
+ "step": 660
346
+ },
347
+ {
348
+ "epoch": 0.045837546343107514,
349
+ "grad_norm": 0.435546875,
350
+ "learning_rate": 0.0007,
351
+ "loss": 3.0828268051147463,
352
+ "step": 680
353
+ },
354
+ {
355
+ "epoch": 0.04718570947084597,
356
+ "grad_norm": 0.427734375,
357
+ "learning_rate": 0.0007,
358
+ "loss": 3.0396488189697264,
359
+ "step": 700
360
+ },
361
+ {
362
+ "epoch": 0.04718570947084597,
363
+ "eval_loss": 3.0276408195495605,
364
+ "eval_runtime": 15.2735,
365
+ "eval_samples_per_second": 623.761,
366
+ "eval_steps_per_second": 0.655,
367
+ "step": 700
368
+ },
369
+ {
370
+ "epoch": 0.04853387259858443,
371
+ "grad_norm": 0.4375,
372
+ "learning_rate": 0.0007,
373
+ "loss": 3.0034805297851563,
374
+ "step": 720
375
+ },
376
+ {
377
+ "epoch": 0.04988203572632288,
378
+ "grad_norm": 0.52734375,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.96556396484375,
381
+ "step": 740
382
+ },
383
+ {
384
+ "epoch": 0.05055611729019211,
385
+ "eval_loss": 2.950009346008301,
386
+ "eval_runtime": 15.243,
387
+ "eval_samples_per_second": 625.009,
388
+ "eval_steps_per_second": 0.656,
389
+ "step": 750
390
+ },
391
+ {
392
+ "epoch": 0.05123019885406134,
393
+ "grad_norm": 0.451171875,
394
+ "learning_rate": 0.0007,
395
+ "loss": 2.950779151916504,
396
+ "step": 760
397
+ },
398
+ {
399
+ "epoch": 0.0525783619817998,
400
+ "grad_norm": 0.400390625,
401
+ "learning_rate": 0.0007,
402
+ "loss": 2.956203079223633,
403
+ "step": 780
404
+ },
405
+ {
406
+ "epoch": 0.053926525109538256,
407
+ "grad_norm": 0.40625,
408
+ "learning_rate": 0.0007,
409
+ "loss": 2.9017578125,
410
+ "step": 800
411
+ },
412
+ {
413
+ "epoch": 0.053926525109538256,
414
+ "eval_loss": 2.8844006061553955,
415
+ "eval_runtime": 15.2903,
416
+ "eval_samples_per_second": 623.073,
417
+ "eval_steps_per_second": 0.654,
418
+ "step": 800
419
+ },
420
+ {
421
+ "epoch": 0.05527468823727671,
422
+ "grad_norm": 0.41015625,
423
+ "learning_rate": 0.0007,
424
+ "loss": 2.887515640258789,
425
+ "step": 820
426
+ },
427
+ {
428
+ "epoch": 0.056622851365015166,
429
+ "grad_norm": 0.41796875,
430
+ "learning_rate": 0.0007,
431
+ "loss": 2.8556928634643555,
432
+ "step": 840
433
+ },
434
+ {
435
+ "epoch": 0.057296932928884395,
436
+ "eval_loss": 2.821610450744629,
437
+ "eval_runtime": 15.2253,
438
+ "eval_samples_per_second": 625.733,
439
+ "eval_steps_per_second": 0.657,
440
+ "step": 850
441
+ },
442
+ {
443
+ "epoch": 0.057971014492753624,
444
+ "grad_norm": 0.423828125,
445
+ "learning_rate": 0.0007,
446
+ "loss": 2.8130855560302734,
447
+ "step": 860
448
+ },
449
+ {
450
+ "epoch": 0.05931917762049208,
451
+ "grad_norm": 0.38671875,
452
+ "learning_rate": 0.0007,
453
+ "loss": 2.790329170227051,
454
+ "step": 880
455
+ },
456
+ {
457
+ "epoch": 0.06066734074823053,
458
+ "grad_norm": 0.427734375,
459
+ "learning_rate": 0.0007,
460
+ "loss": 2.7760162353515625,
461
+ "step": 900
462
+ },
463
+ {
464
+ "epoch": 0.06066734074823053,
465
+ "eval_loss": 2.7661373615264893,
466
+ "eval_runtime": 15.2736,
467
+ "eval_samples_per_second": 623.755,
468
+ "eval_steps_per_second": 0.655,
469
+ "step": 900
470
+ },
471
+ {
472
+ "epoch": 0.06201550387596899,
473
+ "grad_norm": 0.435546875,
474
+ "learning_rate": 0.0007,
475
+ "loss": 2.7462047576904296,
476
+ "step": 920
477
+ },
478
+ {
479
+ "epoch": 0.06336366700370745,
480
+ "grad_norm": 0.4140625,
481
+ "learning_rate": 0.0007,
482
+ "loss": 2.732274627685547,
483
+ "step": 940
484
+ },
485
+ {
486
+ "epoch": 0.06403774856757667,
487
+ "eval_loss": 2.713942289352417,
488
+ "eval_runtime": 15.6062,
489
+ "eval_samples_per_second": 610.463,
490
+ "eval_steps_per_second": 0.641,
491
+ "step": 950
492
+ },
493
+ {
494
+ "epoch": 0.06471183013144591,
495
+ "grad_norm": 0.419921875,
496
+ "learning_rate": 0.0007,
497
+ "loss": 2.705318069458008,
498
+ "step": 960
499
+ },
500
+ {
501
+ "epoch": 0.06605999325918437,
502
+ "grad_norm": 0.4296875,
503
+ "learning_rate": 0.0007,
504
+ "loss": 2.6892818450927733,
505
+ "step": 980
506
+ },
507
+ {
508
+ "epoch": 0.06740815638692282,
509
+ "grad_norm": 0.4453125,
510
+ "learning_rate": 0.0007,
511
+ "loss": 2.687115287780762,
512
+ "step": 1000
513
+ },
514
+ {
515
+ "epoch": 0.06740815638692282,
516
+ "eval_loss": 2.668647289276123,
517
+ "eval_runtime": 15.2597,
518
+ "eval_samples_per_second": 624.326,
519
+ "eval_steps_per_second": 0.655,
520
+ "step": 1000
521
+ },
522
+ {
523
+ "epoch": 0.06875631951466127,
524
+ "grad_norm": 0.388671875,
525
+ "learning_rate": 0.0007,
526
+ "loss": 2.6533071517944338,
527
+ "step": 1020
528
+ },
529
+ {
530
+ "epoch": 0.07010448264239973,
531
+ "grad_norm": 0.392578125,
532
+ "learning_rate": 0.0007,
533
+ "loss": 2.6390922546386717,
534
+ "step": 1040
535
+ },
536
+ {
537
+ "epoch": 0.07077856420626896,
538
+ "eval_loss": 2.6291816234588623,
539
+ "eval_runtime": 15.2457,
540
+ "eval_samples_per_second": 624.896,
541
+ "eval_steps_per_second": 0.656,
542
+ "step": 1050
543
+ },
544
+ {
545
+ "epoch": 0.07145264577013818,
546
+ "grad_norm": 0.4375,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.624461555480957,
549
+ "step": 1060
550
+ },
551
+ {
552
+ "epoch": 0.07280080889787664,
553
+ "grad_norm": 0.396484375,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.6044427871704103,
556
+ "step": 1080
557
+ },
558
+ {
559
+ "epoch": 0.0741489720256151,
560
+ "grad_norm": 0.423828125,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.597646713256836,
563
+ "step": 1100
564
+ },
565
+ {
566
+ "epoch": 0.0741489720256151,
567
+ "eval_loss": 2.589696168899536,
568
+ "eval_runtime": 15.3967,
569
+ "eval_samples_per_second": 618.768,
570
+ "eval_steps_per_second": 0.649,
571
+ "step": 1100
572
+ },
573
+ {
574
+ "epoch": 0.07549713515335356,
575
+ "grad_norm": 0.369140625,
576
+ "learning_rate": 0.0007,
577
+ "loss": 2.561947250366211,
578
+ "step": 1120
579
+ },
580
+ {
581
+ "epoch": 0.07684529828109202,
582
+ "grad_norm": 0.390625,
583
+ "learning_rate": 0.0007,
584
+ "loss": 2.54862117767334,
585
+ "step": 1140
586
+ },
587
+ {
588
+ "epoch": 0.07751937984496124,
589
+ "eval_loss": 2.559185743331909,
590
+ "eval_runtime": 15.2877,
591
+ "eval_samples_per_second": 623.181,
592
+ "eval_steps_per_second": 0.654,
593
+ "step": 1150
594
+ },
595
+ {
596
+ "epoch": 0.07819346140883048,
597
+ "grad_norm": 0.37890625,
598
+ "learning_rate": 0.0007,
599
+ "loss": 2.5469717025756835,
600
+ "step": 1160
601
+ },
602
+ {
603
+ "epoch": 0.07954162453656892,
604
+ "grad_norm": 0.400390625,
605
+ "learning_rate": 0.0007,
606
+ "loss": 2.540451240539551,
607
+ "step": 1180
608
+ },
609
+ {
610
+ "epoch": 0.08088978766430738,
611
+ "grad_norm": 0.40625,
612
+ "learning_rate": 0.0007,
613
+ "loss": 2.53134708404541,
614
+ "step": 1200
615
+ },
616
+ {
617
+ "epoch": 0.08088978766430738,
618
+ "eval_loss": 2.521665096282959,
619
+ "eval_runtime": 15.2822,
620
+ "eval_samples_per_second": 623.407,
621
+ "eval_steps_per_second": 0.654,
622
+ "step": 1200
623
+ },
624
+ {
625
+ "epoch": 0.08223795079204584,
626
+ "grad_norm": 0.37109375,
627
+ "learning_rate": 0.0007,
628
+ "loss": 2.513539123535156,
629
+ "step": 1220
630
+ },
631
+ {
632
+ "epoch": 0.0835861139197843,
633
+ "grad_norm": 0.36328125,
634
+ "learning_rate": 0.0007,
635
+ "loss": 2.5009738922119142,
636
+ "step": 1240
637
+ },
638
+ {
639
+ "epoch": 0.08426019548365352,
640
+ "eval_loss": 2.489459753036499,
641
+ "eval_runtime": 15.2493,
642
+ "eval_samples_per_second": 624.749,
643
+ "eval_steps_per_second": 0.656,
644
+ "step": 1250
645
+ },
646
+ {
647
+ "epoch": 0.08493427704752275,
648
+ "grad_norm": 0.388671875,
649
+ "learning_rate": 0.0007,
650
+ "loss": 2.4894077301025392,
651
+ "step": 1260
652
+ },
653
+ {
654
+ "epoch": 0.08628244017526121,
655
+ "grad_norm": 0.369140625,
656
+ "learning_rate": 0.0007,
657
+ "loss": 2.4697439193725588,
658
+ "step": 1280
659
+ },
660
+ {
661
+ "epoch": 0.08763060330299967,
662
+ "grad_norm": 0.423828125,
663
+ "learning_rate": 0.0007,
664
+ "loss": 2.461294174194336,
665
+ "step": 1300
666
+ },
667
+ {
668
+ "epoch": 0.08763060330299967,
669
+ "eval_loss": 2.4627816677093506,
670
+ "eval_runtime": 15.421,
671
+ "eval_samples_per_second": 617.795,
672
+ "eval_steps_per_second": 0.648,
673
+ "step": 1300
674
+ }
675
+ ],
676
+ "logging_steps": 20,
677
+ "max_steps": 2500,
678
+ "num_input_tokens_seen": 0,
679
+ "num_train_epochs": 1,
680
+ "save_steps": 100,
681
+ "stateful_callbacks": {
682
+ "TrainerControl": {
683
+ "args": {
684
+ "should_epoch_stop": false,
685
+ "should_evaluate": false,
686
+ "should_log": false,
687
+ "should_save": true,
688
+ "should_training_stop": false
689
+ },
690
+ "attributes": {}
691
+ }
692
+ },
693
+ "total_flos": 4194168943411200.0,
694
+ "train_batch_size": 64,
695
+ "trial_name": null,
696
+ "trial_params": null
697
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-1300/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd90512e56f813612dc668676d6f8cab72da110a42f23c61cafebf91064625a1
3
+ size 4920
zain/Activation/out/glu-silu-100L_run/checkpoint-800/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "silu",
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": 100,
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.16.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096,
35
+ "waleed_beta": 10.0
36
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-800/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8578f79eb539aac9aa27fd9a695b168cd0d40df5da3932e5d25940cb9ab0911
3
+ size 33967272
zain/Activation/out/glu-silu-100L_run/checkpoint-800/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11a2194606800c872fab1c82c9f562373be9741e99f2f5f0380a2e416fd1b2c7
3
+ size 68504996
zain/Activation/out/glu-silu-100L_run/checkpoint-800/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f83fe17529e572dff2734bb21512b28dd7cf5d20ef0e84688f5068ffbf24e765
3
+ size 14244
zain/Activation/out/glu-silu-100L_run/checkpoint-800/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4ae273f6e52d9470fdb0cf70fb543f74693bb1ddd058314f4d5c7d3c01c667e
3
+ size 1064
zain/Activation/out/glu-silu-100L_run/checkpoint-800/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-silu-100L_run/checkpoint-800/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
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-800/trainer_state.json ADDED
@@ -0,0 +1,442 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.053926525109538256,
6
+ "eval_steps": 50,
7
+ "global_step": 800,
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": 2.078125,
15
+ "learning_rate": 2.66e-05,
16
+ "loss": 8.29962387084961,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.3515625,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047650909423828,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.003370407819346141,
28
+ "eval_loss": 7.835535049438477,
29
+ "eval_runtime": 15.2165,
30
+ "eval_samples_per_second": 626.095,
31
+ "eval_steps_per_second": 0.657,
32
+ "step": 50
33
+ },
34
+ {
35
+ "epoch": 0.004044489383215369,
36
+ "grad_norm": 1.2734375,
37
+ "learning_rate": 8.259999999999999e-05,
38
+ "loss": 7.8378746032714846,
39
+ "step": 60
40
+ },
41
+ {
42
+ "epoch": 0.005392652510953826,
43
+ "grad_norm": 1.265625,
44
+ "learning_rate": 0.0001106,
45
+ "loss": 7.5728271484375,
46
+ "step": 80
47
+ },
48
+ {
49
+ "epoch": 0.006740815638692282,
50
+ "grad_norm": 1.1953125,
51
+ "learning_rate": 0.0001386,
52
+ "loss": 7.2332916259765625,
53
+ "step": 100
54
+ },
55
+ {
56
+ "epoch": 0.006740815638692282,
57
+ "eval_loss": 7.039053440093994,
58
+ "eval_runtime": 15.2074,
59
+ "eval_samples_per_second": 626.471,
60
+ "eval_steps_per_second": 0.658,
61
+ "step": 100
62
+ },
63
+ {
64
+ "epoch": 0.008088978766430738,
65
+ "grad_norm": 1.1328125,
66
+ "learning_rate": 0.00016659999999999998,
67
+ "loss": 6.8608039855957035,
68
+ "step": 120
69
+ },
70
+ {
71
+ "epoch": 0.009437141894169195,
72
+ "grad_norm": 0.94921875,
73
+ "learning_rate": 0.00019460000000000001,
74
+ "loss": 6.517842864990234,
75
+ "step": 140
76
+ },
77
+ {
78
+ "epoch": 0.010111223458038422,
79
+ "eval_loss": 6.229617595672607,
80
+ "eval_runtime": 15.1978,
81
+ "eval_samples_per_second": 626.867,
82
+ "eval_steps_per_second": 0.658,
83
+ "step": 150
84
+ },
85
+ {
86
+ "epoch": 0.010785305021907651,
87
+ "grad_norm": 0.80078125,
88
+ "learning_rate": 0.0002226,
89
+ "loss": 6.2399169921875,
90
+ "step": 160
91
+ },
92
+ {
93
+ "epoch": 0.012133468149646108,
94
+ "grad_norm": 0.6640625,
95
+ "learning_rate": 0.00025059999999999997,
96
+ "loss": 6.036548614501953,
97
+ "step": 180
98
+ },
99
+ {
100
+ "epoch": 0.013481631277384564,
101
+ "grad_norm": 0.61328125,
102
+ "learning_rate": 0.0002786,
103
+ "loss": 5.857078552246094,
104
+ "step": 200
105
+ },
106
+ {
107
+ "epoch": 0.013481631277384564,
108
+ "eval_loss": 5.765108585357666,
109
+ "eval_runtime": 15.5271,
110
+ "eval_samples_per_second": 613.574,
111
+ "eval_steps_per_second": 0.644,
112
+ "step": 200
113
+ },
114
+ {
115
+ "epoch": 0.01482979440512302,
116
+ "grad_norm": 0.61328125,
117
+ "learning_rate": 0.00030659999999999997,
118
+ "loss": 5.680564117431641,
119
+ "step": 220
120
+ },
121
+ {
122
+ "epoch": 0.016177957532861477,
123
+ "grad_norm": 2.0625,
124
+ "learning_rate": 0.0003346,
125
+ "loss": 5.472259902954102,
126
+ "step": 240
127
+ },
128
+ {
129
+ "epoch": 0.016852039096730706,
130
+ "eval_loss": 5.240335464477539,
131
+ "eval_runtime": 15.2596,
132
+ "eval_samples_per_second": 624.33,
133
+ "eval_steps_per_second": 0.655,
134
+ "step": 250
135
+ },
136
+ {
137
+ "epoch": 0.01752612066059993,
138
+ "grad_norm": 1.7421875,
139
+ "learning_rate": 0.00036260000000000003,
140
+ "loss": 5.24955940246582,
141
+ "step": 260
142
+ },
143
+ {
144
+ "epoch": 0.01887428378833839,
145
+ "grad_norm": 0.62109375,
146
+ "learning_rate": 0.0003906,
147
+ "loss": 4.995759963989258,
148
+ "step": 280
149
+ },
150
+ {
151
+ "epoch": 0.020222446916076844,
152
+ "grad_norm": 1.390625,
153
+ "learning_rate": 0.0004186,
154
+ "loss": 4.769353866577148,
155
+ "step": 300
156
+ },
157
+ {
158
+ "epoch": 0.020222446916076844,
159
+ "eval_loss": 4.669629096984863,
160
+ "eval_runtime": 15.2595,
161
+ "eval_samples_per_second": 624.332,
162
+ "eval_steps_per_second": 0.655,
163
+ "step": 300
164
+ },
165
+ {
166
+ "epoch": 0.021570610043815303,
167
+ "grad_norm": 0.73828125,
168
+ "learning_rate": 0.0004466,
169
+ "loss": 4.569867324829102,
170
+ "step": 320
171
+ },
172
+ {
173
+ "epoch": 0.022918773171553757,
174
+ "grad_norm": 0.765625,
175
+ "learning_rate": 0.00047460000000000004,
176
+ "loss": 4.389446258544922,
177
+ "step": 340
178
+ },
179
+ {
180
+ "epoch": 0.023592854735422986,
181
+ "eval_loss": 4.231207847595215,
182
+ "eval_runtime": 15.2143,
183
+ "eval_samples_per_second": 626.188,
184
+ "eval_steps_per_second": 0.657,
185
+ "step": 350
186
+ },
187
+ {
188
+ "epoch": 0.024266936299292215,
189
+ "grad_norm": 1.2421875,
190
+ "learning_rate": 0.0005026,
191
+ "loss": 4.243848419189453,
192
+ "step": 360
193
+ },
194
+ {
195
+ "epoch": 0.02561509942703067,
196
+ "grad_norm": 0.7109375,
197
+ "learning_rate": 0.0005306,
198
+ "loss": 4.1090648651123045,
199
+ "step": 380
200
+ },
201
+ {
202
+ "epoch": 0.026963262554769128,
203
+ "grad_norm": 0.484375,
204
+ "learning_rate": 0.0005586,
205
+ "loss": 3.966172790527344,
206
+ "step": 400
207
+ },
208
+ {
209
+ "epoch": 0.026963262554769128,
210
+ "eval_loss": 3.923773765563965,
211
+ "eval_runtime": 15.5308,
212
+ "eval_samples_per_second": 613.426,
213
+ "eval_steps_per_second": 0.644,
214
+ "step": 400
215
+ },
216
+ {
217
+ "epoch": 0.028311425682507583,
218
+ "grad_norm": 0.6640625,
219
+ "learning_rate": 0.0005866,
220
+ "loss": 3.8559597015380858,
221
+ "step": 420
222
+ },
223
+ {
224
+ "epoch": 0.02965958881024604,
225
+ "grad_norm": 0.515625,
226
+ "learning_rate": 0.0006146,
227
+ "loss": 3.7943866729736326,
228
+ "step": 440
229
+ },
230
+ {
231
+ "epoch": 0.030333670374115267,
232
+ "eval_loss": 3.6977553367614746,
233
+ "eval_runtime": 15.2001,
234
+ "eval_samples_per_second": 626.77,
235
+ "eval_steps_per_second": 0.658,
236
+ "step": 450
237
+ },
238
+ {
239
+ "epoch": 0.031007751937984496,
240
+ "grad_norm": 0.51953125,
241
+ "learning_rate": 0.0006426,
242
+ "loss": 3.686443328857422,
243
+ "step": 460
244
+ },
245
+ {
246
+ "epoch": 0.032355915065722954,
247
+ "grad_norm": 0.62109375,
248
+ "learning_rate": 0.0006705999999999999,
249
+ "loss": 3.6123157501220704,
250
+ "step": 480
251
+ },
252
+ {
253
+ "epoch": 0.03370407819346141,
254
+ "grad_norm": 0.70703125,
255
+ "learning_rate": 0.0006986,
256
+ "loss": 3.542654800415039,
257
+ "step": 500
258
+ },
259
+ {
260
+ "epoch": 0.03370407819346141,
261
+ "eval_loss": 3.513820171356201,
262
+ "eval_runtime": 15.2193,
263
+ "eval_samples_per_second": 625.983,
264
+ "eval_steps_per_second": 0.657,
265
+ "step": 500
266
+ },
267
+ {
268
+ "epoch": 0.03505224132119986,
269
+ "grad_norm": 0.48046875,
270
+ "learning_rate": 0.0007,
271
+ "loss": 3.472199249267578,
272
+ "step": 520
273
+ },
274
+ {
275
+ "epoch": 0.03640040444893832,
276
+ "grad_norm": 0.53515625,
277
+ "learning_rate": 0.0007,
278
+ "loss": 3.396291732788086,
279
+ "step": 540
280
+ },
281
+ {
282
+ "epoch": 0.03707448601280755,
283
+ "eval_loss": 3.3581483364105225,
284
+ "eval_runtime": 15.3044,
285
+ "eval_samples_per_second": 622.502,
286
+ "eval_steps_per_second": 0.653,
287
+ "step": 550
288
+ },
289
+ {
290
+ "epoch": 0.03774856757667678,
291
+ "grad_norm": 0.47265625,
292
+ "learning_rate": 0.0007,
293
+ "loss": 3.347300720214844,
294
+ "step": 560
295
+ },
296
+ {
297
+ "epoch": 0.03909673070441524,
298
+ "grad_norm": 0.458984375,
299
+ "learning_rate": 0.0007,
300
+ "loss": 3.2885902404785154,
301
+ "step": 580
302
+ },
303
+ {
304
+ "epoch": 0.04044489383215369,
305
+ "grad_norm": 0.4375,
306
+ "learning_rate": 0.0007,
307
+ "loss": 3.25723876953125,
308
+ "step": 600
309
+ },
310
+ {
311
+ "epoch": 0.04044489383215369,
312
+ "eval_loss": 3.2229573726654053,
313
+ "eval_runtime": 15.2967,
314
+ "eval_samples_per_second": 622.812,
315
+ "eval_steps_per_second": 0.654,
316
+ "step": 600
317
+ },
318
+ {
319
+ "epoch": 0.04179305695989215,
320
+ "grad_norm": 0.458984375,
321
+ "learning_rate": 0.0007,
322
+ "loss": 3.1902334213256838,
323
+ "step": 620
324
+ },
325
+ {
326
+ "epoch": 0.043141220087630605,
327
+ "grad_norm": 0.40625,
328
+ "learning_rate": 0.0007,
329
+ "loss": 3.167861557006836,
330
+ "step": 640
331
+ },
332
+ {
333
+ "epoch": 0.043815301651499834,
334
+ "eval_loss": 3.120636463165283,
335
+ "eval_runtime": 15.3751,
336
+ "eval_samples_per_second": 619.639,
337
+ "eval_steps_per_second": 0.65,
338
+ "step": 650
339
+ },
340
+ {
341
+ "epoch": 0.044489383215369056,
342
+ "grad_norm": 0.494140625,
343
+ "learning_rate": 0.0007,
344
+ "loss": 3.1089324951171875,
345
+ "step": 660
346
+ },
347
+ {
348
+ "epoch": 0.045837546343107514,
349
+ "grad_norm": 0.435546875,
350
+ "learning_rate": 0.0007,
351
+ "loss": 3.0828268051147463,
352
+ "step": 680
353
+ },
354
+ {
355
+ "epoch": 0.04718570947084597,
356
+ "grad_norm": 0.427734375,
357
+ "learning_rate": 0.0007,
358
+ "loss": 3.0396488189697264,
359
+ "step": 700
360
+ },
361
+ {
362
+ "epoch": 0.04718570947084597,
363
+ "eval_loss": 3.0276408195495605,
364
+ "eval_runtime": 15.2735,
365
+ "eval_samples_per_second": 623.761,
366
+ "eval_steps_per_second": 0.655,
367
+ "step": 700
368
+ },
369
+ {
370
+ "epoch": 0.04853387259858443,
371
+ "grad_norm": 0.4375,
372
+ "learning_rate": 0.0007,
373
+ "loss": 3.0034805297851563,
374
+ "step": 720
375
+ },
376
+ {
377
+ "epoch": 0.04988203572632288,
378
+ "grad_norm": 0.52734375,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.96556396484375,
381
+ "step": 740
382
+ },
383
+ {
384
+ "epoch": 0.05055611729019211,
385
+ "eval_loss": 2.950009346008301,
386
+ "eval_runtime": 15.243,
387
+ "eval_samples_per_second": 625.009,
388
+ "eval_steps_per_second": 0.656,
389
+ "step": 750
390
+ },
391
+ {
392
+ "epoch": 0.05123019885406134,
393
+ "grad_norm": 0.451171875,
394
+ "learning_rate": 0.0007,
395
+ "loss": 2.950779151916504,
396
+ "step": 760
397
+ },
398
+ {
399
+ "epoch": 0.0525783619817998,
400
+ "grad_norm": 0.400390625,
401
+ "learning_rate": 0.0007,
402
+ "loss": 2.956203079223633,
403
+ "step": 780
404
+ },
405
+ {
406
+ "epoch": 0.053926525109538256,
407
+ "grad_norm": 0.40625,
408
+ "learning_rate": 0.0007,
409
+ "loss": 2.9017578125,
410
+ "step": 800
411
+ },
412
+ {
413
+ "epoch": 0.053926525109538256,
414
+ "eval_loss": 2.8844006061553955,
415
+ "eval_runtime": 15.2903,
416
+ "eval_samples_per_second": 623.073,
417
+ "eval_steps_per_second": 0.654,
418
+ "step": 800
419
+ }
420
+ ],
421
+ "logging_steps": 20,
422
+ "max_steps": 2500,
423
+ "num_input_tokens_seen": 0,
424
+ "num_train_epochs": 1,
425
+ "save_steps": 100,
426
+ "stateful_callbacks": {
427
+ "TrainerControl": {
428
+ "args": {
429
+ "should_epoch_stop": false,
430
+ "should_evaluate": false,
431
+ "should_log": false,
432
+ "should_save": true,
433
+ "should_training_stop": false
434
+ },
435
+ "attributes": {}
436
+ }
437
+ },
438
+ "total_flos": 2581027042099200.0,
439
+ "train_batch_size": 64,
440
+ "trial_name": null,
441
+ "trial_params": null
442
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-800/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd90512e56f813612dc668676d6f8cab72da110a42f23c61cafebf91064625a1
3
+ size 4920
zain/Activation/out/glu-silu-100L_run/checkpoint-900/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "silu",
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": 100,
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.16.0.dev0",
33
+ "use_cache": false,
34
+ "vocab_size": 4096,
35
+ "waleed_beta": 10.0
36
+ }
zain/Activation/out/glu-silu-100L_run/checkpoint-900/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd0cf64cf9569bab9ee756c09a1068b2ae357e614a4057b015ccf219a4192800
3
+ size 33967272
zain/Activation/out/glu-silu-100L_run/checkpoint-900/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:950402bd16f4d17c67311a853b37e49303a008513071aab32b7fb48001bdfeaa
3
+ size 68504996
zain/Activation/out/glu-silu-100L_run/checkpoint-900/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86a2aa7bca84d2b8e7dd4e04a286714ba9169af11c46c739950a52df4c45259f
3
+ size 14244
zain/Activation/out/glu-silu-100L_run/checkpoint-900/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7af41faf99717dd67953037b442647180e36de84b90ea94fe9ee61f85ee5881
3
+ size 1064