w-ahmad commited on
Commit
4a712d9
·
verified ·
1 Parent(s): 235a3a5

Auto upload zain 2026-08-14T21:14:29.212142 (part 2)

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +1 -0
  2. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2700/tokenizer_config.json +13 -0
  3. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2700/trainer_state.json +995 -0
  4. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2700/training_args.bin +3 -0
  5. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/config.json +36 -0
  6. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/model.safetensors +3 -0
  7. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/optimizer.pt +3 -0
  8. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/rng_state.pth +3 -0
  9. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/scheduler.pt +3 -0
  10. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/tokenizer.json +0 -0
  11. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/tokenizer_config.json +13 -0
  12. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/trainer_state.json +1030 -0
  13. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/training_args.bin +3 -0
  14. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/config.json +36 -0
  15. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/model.safetensors +3 -0
  16. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/optimizer.pt +3 -0
  17. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/rng_state.pth +3 -0
  18. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/scheduler.pt +3 -0
  19. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/tokenizer.json +0 -0
  20. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/tokenizer_config.json +13 -0
  21. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/trainer_state.json +1065 -0
  22. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/training_args.bin +3 -0
  23. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/config.json +36 -0
  24. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/model.safetensors +3 -0
  25. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/optimizer.pt +3 -0
  26. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/rng_state.pth +3 -0
  27. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/scheduler.pt +3 -0
  28. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/tokenizer.json +0 -0
  29. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/tokenizer_config.json +13 -0
  30. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/trainer_state.json +1100 -0
  31. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/training_args.bin +3 -0
  32. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/config.json +36 -0
  33. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/model.safetensors +3 -0
  34. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/optimizer.pt +3 -0
  35. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/rng_state.pth +3 -0
  36. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/scheduler.pt +3 -0
  37. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/tokenizer.json +0 -0
  38. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/tokenizer_config.json +13 -0
  39. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/trainer_state.json +1135 -0
  40. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/training_args.bin +3 -0
  41. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/config.json +36 -0
  42. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/model.safetensors +3 -0
  43. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/optimizer.pt +3 -0
  44. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/rng_state.pth +3 -0
  45. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/scheduler.pt +3 -0
  46. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/tokenizer.json +0 -0
  47. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/tokenizer_config.json +13 -0
  48. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/trainer_state.json +1170 -0
  49. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/training_args.bin +3 -0
  50. zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3300/config.json +36 -0
.gitattributes CHANGED
@@ -55,3 +55,4 @@ zain/Activation/wandb/run-20260814_210109-qauftiuh/run-qauftiuh.wandb filter=lfs
55
  zain/Activation/out/glu-relu-100L_trash_run/training_log.jsonl filter=lfs diff=lfs merge=lfs -text
56
  zain/Activation/wandb/run-20260814_210155-rxpj34pj/run-rxpj34pj.wandb filter=lfs diff=lfs merge=lfs -text
57
  zain/Activation/wandb/run-20260814_210809-a6574zx8/run-a6574zx8.wandb filter=lfs diff=lfs merge=lfs -text
 
 
55
  zain/Activation/out/glu-relu-100L_trash_run/training_log.jsonl filter=lfs diff=lfs merge=lfs -text
56
  zain/Activation/wandb/run-20260814_210155-rxpj34pj/run-rxpj34pj.wandb filter=lfs diff=lfs merge=lfs -text
57
  zain/Activation/wandb/run-20260814_210809-a6574zx8/run-a6574zx8.wandb filter=lfs diff=lfs merge=lfs -text
58
+ zain/Activation/out/glu-gelu-100L_trash_run/training_log.jsonl filter=lfs diff=lfs merge=lfs -text
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2700/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-gelu-100L_trash_run/checkpoint-2700/trainer_state.json ADDED
@@ -0,0 +1,995 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.1820020222446916,
6
+ "eval_steps": 2498,
7
+ "global_step": 2700,
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.299579620361328,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.359375,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047666931152344,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.2734375,
29
+ "learning_rate": 8.259999999999999e-05,
30
+ "loss": 7.838191986083984,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 1.265625,
36
+ "learning_rate": 0.0001106,
37
+ "loss": 7.57310791015625,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.1953125,
43
+ "learning_rate": 0.0001386,
44
+ "loss": 7.233531951904297,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 1.1328125,
50
+ "learning_rate": 0.00016659999999999998,
51
+ "loss": 6.860941314697266,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 0.95703125,
57
+ "learning_rate": 0.00019460000000000001,
58
+ "loss": 6.51712646484375,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 0.81640625,
64
+ "learning_rate": 0.0002226,
65
+ "loss": 6.234881591796875,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 0.73828125,
71
+ "learning_rate": 0.00025059999999999997,
72
+ "loss": 6.024795913696289,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 1.0390625,
78
+ "learning_rate": 0.0002786,
79
+ "loss": 5.837321472167969,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.01482979440512302,
84
+ "grad_norm": 0.63671875,
85
+ "learning_rate": 0.00030659999999999997,
86
+ "loss": 5.647708129882813,
87
+ "step": 220
88
+ },
89
+ {
90
+ "epoch": 0.016177957532861477,
91
+ "grad_norm": 0.8984375,
92
+ "learning_rate": 0.0003346,
93
+ "loss": 5.43688735961914,
94
+ "step": 240
95
+ },
96
+ {
97
+ "epoch": 0.01752612066059993,
98
+ "grad_norm": 0.77734375,
99
+ "learning_rate": 0.00036260000000000003,
100
+ "loss": 5.210599517822265,
101
+ "step": 260
102
+ },
103
+ {
104
+ "epoch": 0.01887428378833839,
105
+ "grad_norm": 0.890625,
106
+ "learning_rate": 0.0003906,
107
+ "loss": 4.956959533691406,
108
+ "step": 280
109
+ },
110
+ {
111
+ "epoch": 0.020222446916076844,
112
+ "grad_norm": 0.84765625,
113
+ "learning_rate": 0.0004186,
114
+ "loss": 4.726874160766601,
115
+ "step": 300
116
+ },
117
+ {
118
+ "epoch": 0.021570610043815303,
119
+ "grad_norm": 0.87890625,
120
+ "learning_rate": 0.0004466,
121
+ "loss": 4.527945327758789,
122
+ "step": 320
123
+ },
124
+ {
125
+ "epoch": 0.022918773171553757,
126
+ "grad_norm": 1.3515625,
127
+ "learning_rate": 0.00047460000000000004,
128
+ "loss": 4.3517810821533205,
129
+ "step": 340
130
+ },
131
+ {
132
+ "epoch": 0.024266936299292215,
133
+ "grad_norm": 0.76171875,
134
+ "learning_rate": 0.0005026,
135
+ "loss": 4.212622451782226,
136
+ "step": 360
137
+ },
138
+ {
139
+ "epoch": 0.02561509942703067,
140
+ "grad_norm": 0.6640625,
141
+ "learning_rate": 0.0005306,
142
+ "loss": 4.076633453369141,
143
+ "step": 380
144
+ },
145
+ {
146
+ "epoch": 0.026963262554769128,
147
+ "grad_norm": 0.498046875,
148
+ "learning_rate": 0.0005586,
149
+ "loss": 3.9346649169921877,
150
+ "step": 400
151
+ },
152
+ {
153
+ "epoch": 0.028311425682507583,
154
+ "grad_norm": 0.85546875,
155
+ "learning_rate": 0.0005866,
156
+ "loss": 3.8310710906982424,
157
+ "step": 420
158
+ },
159
+ {
160
+ "epoch": 0.02965958881024604,
161
+ "grad_norm": 0.59375,
162
+ "learning_rate": 0.0006146,
163
+ "loss": 3.7676155090332033,
164
+ "step": 440
165
+ },
166
+ {
167
+ "epoch": 0.031007751937984496,
168
+ "grad_norm": 0.6171875,
169
+ "learning_rate": 0.0006426,
170
+ "loss": 3.6599285125732424,
171
+ "step": 460
172
+ },
173
+ {
174
+ "epoch": 0.032355915065722954,
175
+ "grad_norm": 0.5546875,
176
+ "learning_rate": 0.0006705999999999999,
177
+ "loss": 3.5886669158935547,
178
+ "step": 480
179
+ },
180
+ {
181
+ "epoch": 0.03370407819346141,
182
+ "grad_norm": 0.546875,
183
+ "learning_rate": 0.0006986,
184
+ "loss": 3.512950134277344,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 0.03505224132119986,
189
+ "grad_norm": 0.53125,
190
+ "learning_rate": 0.0007,
191
+ "loss": 3.444244384765625,
192
+ "step": 520
193
+ },
194
+ {
195
+ "epoch": 0.03640040444893832,
196
+ "grad_norm": 0.5078125,
197
+ "learning_rate": 0.0007,
198
+ "loss": 3.369425964355469,
199
+ "step": 540
200
+ },
201
+ {
202
+ "epoch": 0.03774856757667678,
203
+ "grad_norm": 0.5625,
204
+ "learning_rate": 0.0007,
205
+ "loss": 3.318771743774414,
206
+ "step": 560
207
+ },
208
+ {
209
+ "epoch": 0.03909673070441524,
210
+ "grad_norm": 0.470703125,
211
+ "learning_rate": 0.0007,
212
+ "loss": 3.2595756530761717,
213
+ "step": 580
214
+ },
215
+ {
216
+ "epoch": 0.04044489383215369,
217
+ "grad_norm": 0.53125,
218
+ "learning_rate": 0.0007,
219
+ "loss": 3.2241649627685547,
220
+ "step": 600
221
+ },
222
+ {
223
+ "epoch": 0.04179305695989215,
224
+ "grad_norm": 0.484375,
225
+ "learning_rate": 0.0007,
226
+ "loss": 3.1619989395141603,
227
+ "step": 620
228
+ },
229
+ {
230
+ "epoch": 0.043141220087630605,
231
+ "grad_norm": 0.5625,
232
+ "learning_rate": 0.0007,
233
+ "loss": 3.14036865234375,
234
+ "step": 640
235
+ },
236
+ {
237
+ "epoch": 0.044489383215369056,
238
+ "grad_norm": 0.41796875,
239
+ "learning_rate": 0.0007,
240
+ "loss": 3.083884620666504,
241
+ "step": 660
242
+ },
243
+ {
244
+ "epoch": 0.045837546343107514,
245
+ "grad_norm": 0.451171875,
246
+ "learning_rate": 0.0007,
247
+ "loss": 3.0591588973999024,
248
+ "step": 680
249
+ },
250
+ {
251
+ "epoch": 0.04718570947084597,
252
+ "grad_norm": 0.423828125,
253
+ "learning_rate": 0.0007,
254
+ "loss": 3.015497016906738,
255
+ "step": 700
256
+ },
257
+ {
258
+ "epoch": 0.04853387259858443,
259
+ "grad_norm": 0.41796875,
260
+ "learning_rate": 0.0007,
261
+ "loss": 2.9805700302124025,
262
+ "step": 720
263
+ },
264
+ {
265
+ "epoch": 0.04988203572632288,
266
+ "grad_norm": 0.46875,
267
+ "learning_rate": 0.0007,
268
+ "loss": 2.9426918029785156,
269
+ "step": 740
270
+ },
271
+ {
272
+ "epoch": 0.05123019885406134,
273
+ "grad_norm": 0.46484375,
274
+ "learning_rate": 0.0007,
275
+ "loss": 2.9274158477783203,
276
+ "step": 760
277
+ },
278
+ {
279
+ "epoch": 0.0525783619817998,
280
+ "grad_norm": 0.421875,
281
+ "learning_rate": 0.0007,
282
+ "loss": 2.931459426879883,
283
+ "step": 780
284
+ },
285
+ {
286
+ "epoch": 0.053926525109538256,
287
+ "grad_norm": 0.486328125,
288
+ "learning_rate": 0.0007,
289
+ "loss": 2.877642059326172,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.05527468823727671,
294
+ "grad_norm": 0.40234375,
295
+ "learning_rate": 0.0007,
296
+ "loss": 2.8646697998046875,
297
+ "step": 820
298
+ },
299
+ {
300
+ "epoch": 0.056622851365015166,
301
+ "grad_norm": 0.451171875,
302
+ "learning_rate": 0.0007,
303
+ "loss": 2.8328128814697267,
304
+ "step": 840
305
+ },
306
+ {
307
+ "epoch": 0.057971014492753624,
308
+ "grad_norm": 0.396484375,
309
+ "learning_rate": 0.0007,
310
+ "loss": 2.790274429321289,
311
+ "step": 860
312
+ },
313
+ {
314
+ "epoch": 0.05931917762049208,
315
+ "grad_norm": 0.423828125,
316
+ "learning_rate": 0.0007,
317
+ "loss": 2.7668125152587892,
318
+ "step": 880
319
+ },
320
+ {
321
+ "epoch": 0.06066734074823053,
322
+ "grad_norm": 0.427734375,
323
+ "learning_rate": 0.0007,
324
+ "loss": 2.7516881942749025,
325
+ "step": 900
326
+ },
327
+ {
328
+ "epoch": 0.06201550387596899,
329
+ "grad_norm": 0.42578125,
330
+ "learning_rate": 0.0007,
331
+ "loss": 2.722590446472168,
332
+ "step": 920
333
+ },
334
+ {
335
+ "epoch": 0.06336366700370745,
336
+ "grad_norm": 0.4375,
337
+ "learning_rate": 0.0007,
338
+ "loss": 2.708496856689453,
339
+ "step": 940
340
+ },
341
+ {
342
+ "epoch": 0.06471183013144591,
343
+ "grad_norm": 0.400390625,
344
+ "learning_rate": 0.0007,
345
+ "loss": 2.681654930114746,
346
+ "step": 960
347
+ },
348
+ {
349
+ "epoch": 0.06605999325918437,
350
+ "grad_norm": 0.396484375,
351
+ "learning_rate": 0.0007,
352
+ "loss": 2.6650571823120117,
353
+ "step": 980
354
+ },
355
+ {
356
+ "epoch": 0.06740815638692282,
357
+ "grad_norm": 0.41015625,
358
+ "learning_rate": 0.0007,
359
+ "loss": 2.6637720108032226,
360
+ "step": 1000
361
+ },
362
+ {
363
+ "epoch": 0.06875631951466127,
364
+ "grad_norm": 0.396484375,
365
+ "learning_rate": 0.0007,
366
+ "loss": 2.6311159133911133,
367
+ "step": 1020
368
+ },
369
+ {
370
+ "epoch": 0.07010448264239973,
371
+ "grad_norm": 0.408203125,
372
+ "learning_rate": 0.0007,
373
+ "loss": 2.615707015991211,
374
+ "step": 1040
375
+ },
376
+ {
377
+ "epoch": 0.07145264577013818,
378
+ "grad_norm": 0.404296875,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.6003103256225586,
381
+ "step": 1060
382
+ },
383
+ {
384
+ "epoch": 0.07280080889787664,
385
+ "grad_norm": 0.41015625,
386
+ "learning_rate": 0.0007,
387
+ "loss": 2.581348991394043,
388
+ "step": 1080
389
+ },
390
+ {
391
+ "epoch": 0.0741489720256151,
392
+ "grad_norm": 0.404296875,
393
+ "learning_rate": 0.0007,
394
+ "loss": 2.574792671203613,
395
+ "step": 1100
396
+ },
397
+ {
398
+ "epoch": 0.07549713515335356,
399
+ "grad_norm": 0.384765625,
400
+ "learning_rate": 0.0007,
401
+ "loss": 2.538400650024414,
402
+ "step": 1120
403
+ },
404
+ {
405
+ "epoch": 0.07684529828109202,
406
+ "grad_norm": 0.41796875,
407
+ "learning_rate": 0.0007,
408
+ "loss": 2.5249162673950196,
409
+ "step": 1140
410
+ },
411
+ {
412
+ "epoch": 0.07819346140883048,
413
+ "grad_norm": 0.404296875,
414
+ "learning_rate": 0.0007,
415
+ "loss": 2.5237232208251954,
416
+ "step": 1160
417
+ },
418
+ {
419
+ "epoch": 0.07954162453656892,
420
+ "grad_norm": 0.396484375,
421
+ "learning_rate": 0.0007,
422
+ "loss": 2.517318344116211,
423
+ "step": 1180
424
+ },
425
+ {
426
+ "epoch": 0.08088978766430738,
427
+ "grad_norm": 0.41796875,
428
+ "learning_rate": 0.0007,
429
+ "loss": 2.507699203491211,
430
+ "step": 1200
431
+ },
432
+ {
433
+ "epoch": 0.08223795079204584,
434
+ "grad_norm": 0.3828125,
435
+ "learning_rate": 0.0007,
436
+ "loss": 2.4932661056518555,
437
+ "step": 1220
438
+ },
439
+ {
440
+ "epoch": 0.0835861139197843,
441
+ "grad_norm": 0.365234375,
442
+ "learning_rate": 0.0007,
443
+ "loss": 2.4781061172485352,
444
+ "step": 1240
445
+ },
446
+ {
447
+ "epoch": 0.08493427704752275,
448
+ "grad_norm": 0.388671875,
449
+ "learning_rate": 0.0007,
450
+ "loss": 2.465612030029297,
451
+ "step": 1260
452
+ },
453
+ {
454
+ "epoch": 0.08628244017526121,
455
+ "grad_norm": 0.41796875,
456
+ "learning_rate": 0.0007,
457
+ "loss": 2.4463497161865235,
458
+ "step": 1280
459
+ },
460
+ {
461
+ "epoch": 0.08763060330299967,
462
+ "grad_norm": 0.390625,
463
+ "learning_rate": 0.0007,
464
+ "loss": 2.4374414443969727,
465
+ "step": 1300
466
+ },
467
+ {
468
+ "epoch": 0.08897876643073811,
469
+ "grad_norm": 0.35546875,
470
+ "learning_rate": 0.0007,
471
+ "loss": 2.4308101654052736,
472
+ "step": 1320
473
+ },
474
+ {
475
+ "epoch": 0.09032692955847657,
476
+ "grad_norm": 0.40234375,
477
+ "learning_rate": 0.0007,
478
+ "loss": 2.416130256652832,
479
+ "step": 1340
480
+ },
481
+ {
482
+ "epoch": 0.09167509268621503,
483
+ "grad_norm": 0.4296875,
484
+ "learning_rate": 0.0007,
485
+ "loss": 2.41524600982666,
486
+ "step": 1360
487
+ },
488
+ {
489
+ "epoch": 0.09302325581395349,
490
+ "grad_norm": 0.359375,
491
+ "learning_rate": 0.0007,
492
+ "loss": 2.392221450805664,
493
+ "step": 1380
494
+ },
495
+ {
496
+ "epoch": 0.09437141894169195,
497
+ "grad_norm": 0.373046875,
498
+ "learning_rate": 0.0007,
499
+ "loss": 2.383866310119629,
500
+ "step": 1400
501
+ },
502
+ {
503
+ "epoch": 0.0957195820694304,
504
+ "grad_norm": 0.396484375,
505
+ "learning_rate": 0.0007,
506
+ "loss": 2.394974708557129,
507
+ "step": 1420
508
+ },
509
+ {
510
+ "epoch": 0.09706774519716886,
511
+ "grad_norm": 0.396484375,
512
+ "learning_rate": 0.0007,
513
+ "loss": 2.3713239669799804,
514
+ "step": 1440
515
+ },
516
+ {
517
+ "epoch": 0.09841590832490732,
518
+ "grad_norm": 0.37109375,
519
+ "learning_rate": 0.0007,
520
+ "loss": 2.3597253799438476,
521
+ "step": 1460
522
+ },
523
+ {
524
+ "epoch": 0.09976407145264576,
525
+ "grad_norm": 0.34765625,
526
+ "learning_rate": 0.0007,
527
+ "loss": 2.3452421188354493,
528
+ "step": 1480
529
+ },
530
+ {
531
+ "epoch": 0.10111223458038422,
532
+ "grad_norm": 0.376953125,
533
+ "learning_rate": 0.0007,
534
+ "loss": 2.3336454391479493,
535
+ "step": 1500
536
+ },
537
+ {
538
+ "epoch": 0.10246039770812268,
539
+ "grad_norm": 0.39453125,
540
+ "learning_rate": 0.0007,
541
+ "loss": 2.3358110427856444,
542
+ "step": 1520
543
+ },
544
+ {
545
+ "epoch": 0.10380856083586114,
546
+ "grad_norm": 0.40625,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.324386787414551,
549
+ "step": 1540
550
+ },
551
+ {
552
+ "epoch": 0.1051567239635996,
553
+ "grad_norm": 0.376953125,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.313844108581543,
556
+ "step": 1560
557
+ },
558
+ {
559
+ "epoch": 0.10650488709133805,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.3294256210327147,
563
+ "step": 1580
564
+ },
565
+ {
566
+ "epoch": 0.10785305021907651,
567
+ "grad_norm": 0.40234375,
568
+ "learning_rate": 0.0007,
569
+ "loss": 2.297942352294922,
570
+ "step": 1600
571
+ },
572
+ {
573
+ "epoch": 0.10920121334681497,
574
+ "grad_norm": 0.357421875,
575
+ "learning_rate": 0.0007,
576
+ "loss": 2.296767997741699,
577
+ "step": 1620
578
+ },
579
+ {
580
+ "epoch": 0.11054937647455342,
581
+ "grad_norm": 0.37890625,
582
+ "learning_rate": 0.0007,
583
+ "loss": 2.3081844329833983,
584
+ "step": 1640
585
+ },
586
+ {
587
+ "epoch": 0.11189753960229187,
588
+ "grad_norm": 0.35546875,
589
+ "learning_rate": 0.0007,
590
+ "loss": 2.2883626937866213,
591
+ "step": 1660
592
+ },
593
+ {
594
+ "epoch": 0.11324570273003033,
595
+ "grad_norm": 0.375,
596
+ "learning_rate": 0.0007,
597
+ "loss": 2.272566223144531,
598
+ "step": 1680
599
+ },
600
+ {
601
+ "epoch": 0.11459386585776879,
602
+ "grad_norm": 0.373046875,
603
+ "learning_rate": 0.0007,
604
+ "loss": 2.281250762939453,
605
+ "step": 1700
606
+ },
607
+ {
608
+ "epoch": 0.11594202898550725,
609
+ "grad_norm": 0.41796875,
610
+ "learning_rate": 0.0007,
611
+ "loss": 2.257417106628418,
612
+ "step": 1720
613
+ },
614
+ {
615
+ "epoch": 0.1172901921132457,
616
+ "grad_norm": 0.365234375,
617
+ "learning_rate": 0.0007,
618
+ "loss": 2.2473636627197267,
619
+ "step": 1740
620
+ },
621
+ {
622
+ "epoch": 0.11863835524098416,
623
+ "grad_norm": 0.380859375,
624
+ "learning_rate": 0.0007,
625
+ "loss": 2.253912925720215,
626
+ "step": 1760
627
+ },
628
+ {
629
+ "epoch": 0.11998651836872262,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 0.0007,
632
+ "loss": 2.2672780990600585,
633
+ "step": 1780
634
+ },
635
+ {
636
+ "epoch": 0.12133468149646107,
637
+ "grad_norm": 0.3671875,
638
+ "learning_rate": 0.0007,
639
+ "loss": 2.261806106567383,
640
+ "step": 1800
641
+ },
642
+ {
643
+ "epoch": 0.12268284462419952,
644
+ "grad_norm": 0.375,
645
+ "learning_rate": 0.0007,
646
+ "loss": 2.2513105392456056,
647
+ "step": 1820
648
+ },
649
+ {
650
+ "epoch": 0.12403100775193798,
651
+ "grad_norm": 0.369140625,
652
+ "learning_rate": 0.0007,
653
+ "loss": 2.2201774597167967,
654
+ "step": 1840
655
+ },
656
+ {
657
+ "epoch": 0.12537917087967643,
658
+ "grad_norm": 0.35546875,
659
+ "learning_rate": 0.0007,
660
+ "loss": 2.2213924407958983,
661
+ "step": 1860
662
+ },
663
+ {
664
+ "epoch": 0.1267273340074149,
665
+ "grad_norm": 0.361328125,
666
+ "learning_rate": 0.0007,
667
+ "loss": 2.21158447265625,
668
+ "step": 1880
669
+ },
670
+ {
671
+ "epoch": 0.12807549713515334,
672
+ "grad_norm": 0.357421875,
673
+ "learning_rate": 0.0007,
674
+ "loss": 2.218313217163086,
675
+ "step": 1900
676
+ },
677
+ {
678
+ "epoch": 0.12942366026289182,
679
+ "grad_norm": 0.361328125,
680
+ "learning_rate": 0.0007,
681
+ "loss": 2.1961637496948243,
682
+ "step": 1920
683
+ },
684
+ {
685
+ "epoch": 0.13077182339063026,
686
+ "grad_norm": 0.369140625,
687
+ "learning_rate": 0.0007,
688
+ "loss": 2.2057926177978517,
689
+ "step": 1940
690
+ },
691
+ {
692
+ "epoch": 0.13211998651836873,
693
+ "grad_norm": 0.345703125,
694
+ "learning_rate": 0.0007,
695
+ "loss": 2.181648826599121,
696
+ "step": 1960
697
+ },
698
+ {
699
+ "epoch": 0.13346814964610718,
700
+ "grad_norm": 0.380859375,
701
+ "learning_rate": 0.0007,
702
+ "loss": 2.203908920288086,
703
+ "step": 1980
704
+ },
705
+ {
706
+ "epoch": 0.13481631277384565,
707
+ "grad_norm": 0.341796875,
708
+ "learning_rate": 0.0007,
709
+ "loss": 2.1758956909179688,
710
+ "step": 2000
711
+ },
712
+ {
713
+ "epoch": 0.1361644759015841,
714
+ "grad_norm": 0.35546875,
715
+ "learning_rate": 0.0007,
716
+ "loss": 2.185748481750488,
717
+ "step": 2020
718
+ },
719
+ {
720
+ "epoch": 0.13751263902932254,
721
+ "grad_norm": 0.41796875,
722
+ "learning_rate": 0.0007,
723
+ "loss": 2.165380668640137,
724
+ "step": 2040
725
+ },
726
+ {
727
+ "epoch": 0.138860802157061,
728
+ "grad_norm": 0.3515625,
729
+ "learning_rate": 0.0007,
730
+ "loss": 2.1852161407470705,
731
+ "step": 2060
732
+ },
733
+ {
734
+ "epoch": 0.14020896528479945,
735
+ "grad_norm": 0.34765625,
736
+ "learning_rate": 0.0007,
737
+ "loss": 2.160782051086426,
738
+ "step": 2080
739
+ },
740
+ {
741
+ "epoch": 0.14155712841253792,
742
+ "grad_norm": 0.337890625,
743
+ "learning_rate": 0.0007,
744
+ "loss": 2.152023124694824,
745
+ "step": 2100
746
+ },
747
+ {
748
+ "epoch": 0.14290529154027637,
749
+ "grad_norm": 0.359375,
750
+ "learning_rate": 0.0007,
751
+ "loss": 2.160308837890625,
752
+ "step": 2120
753
+ },
754
+ {
755
+ "epoch": 0.14425345466801484,
756
+ "grad_norm": 0.34765625,
757
+ "learning_rate": 0.0007,
758
+ "loss": 2.1609643936157226,
759
+ "step": 2140
760
+ },
761
+ {
762
+ "epoch": 0.14560161779575329,
763
+ "grad_norm": 0.357421875,
764
+ "learning_rate": 0.0007,
765
+ "loss": 2.1475265502929686,
766
+ "step": 2160
767
+ },
768
+ {
769
+ "epoch": 0.14694978092349173,
770
+ "grad_norm": 0.345703125,
771
+ "learning_rate": 0.0007,
772
+ "loss": 2.155929183959961,
773
+ "step": 2180
774
+ },
775
+ {
776
+ "epoch": 0.1482979440512302,
777
+ "grad_norm": 0.341796875,
778
+ "learning_rate": 0.0007,
779
+ "loss": 2.1293373107910156,
780
+ "step": 2200
781
+ },
782
+ {
783
+ "epoch": 0.14964610717896865,
784
+ "grad_norm": 0.35546875,
785
+ "learning_rate": 0.0007,
786
+ "loss": 2.137188720703125,
787
+ "step": 2220
788
+ },
789
+ {
790
+ "epoch": 0.15099427030670712,
791
+ "grad_norm": 0.345703125,
792
+ "learning_rate": 0.0007,
793
+ "loss": 2.1462854385375976,
794
+ "step": 2240
795
+ },
796
+ {
797
+ "epoch": 0.15234243343444556,
798
+ "grad_norm": 0.365234375,
799
+ "learning_rate": 0.0007,
800
+ "loss": 2.1255794525146485,
801
+ "step": 2260
802
+ },
803
+ {
804
+ "epoch": 0.15369059656218403,
805
+ "grad_norm": 0.33984375,
806
+ "learning_rate": 0.0007,
807
+ "loss": 2.1365018844604493,
808
+ "step": 2280
809
+ },
810
+ {
811
+ "epoch": 0.15503875968992248,
812
+ "grad_norm": 0.337890625,
813
+ "learning_rate": 0.0007,
814
+ "loss": 2.1164464950561523,
815
+ "step": 2300
816
+ },
817
+ {
818
+ "epoch": 0.15638692281766095,
819
+ "grad_norm": 0.361328125,
820
+ "learning_rate": 0.0007,
821
+ "loss": 2.127320671081543,
822
+ "step": 2320
823
+ },
824
+ {
825
+ "epoch": 0.1577350859453994,
826
+ "grad_norm": 0.3359375,
827
+ "learning_rate": 0.0007,
828
+ "loss": 2.117641067504883,
829
+ "step": 2340
830
+ },
831
+ {
832
+ "epoch": 0.15908324907313784,
833
+ "grad_norm": 0.357421875,
834
+ "learning_rate": 0.0007,
835
+ "loss": 2.119377326965332,
836
+ "step": 2360
837
+ },
838
+ {
839
+ "epoch": 0.1604314122008763,
840
+ "grad_norm": 0.33203125,
841
+ "learning_rate": 0.0007,
842
+ "loss": 2.121909332275391,
843
+ "step": 2380
844
+ },
845
+ {
846
+ "epoch": 0.16177957532861476,
847
+ "grad_norm": 0.361328125,
848
+ "learning_rate": 0.0007,
849
+ "loss": 2.10235538482666,
850
+ "step": 2400
851
+ },
852
+ {
853
+ "epoch": 0.16312773845635323,
854
+ "grad_norm": 0.353515625,
855
+ "learning_rate": 0.0007,
856
+ "loss": 2.1056692123413088,
857
+ "step": 2420
858
+ },
859
+ {
860
+ "epoch": 0.16447590158409167,
861
+ "grad_norm": 0.3359375,
862
+ "learning_rate": 0.0007,
863
+ "loss": 2.093578910827637,
864
+ "step": 2440
865
+ },
866
+ {
867
+ "epoch": 0.16582406471183014,
868
+ "grad_norm": 0.337890625,
869
+ "learning_rate": 0.0007,
870
+ "loss": 2.1059795379638673,
871
+ "step": 2460
872
+ },
873
+ {
874
+ "epoch": 0.1671722278395686,
875
+ "grad_norm": 0.361328125,
876
+ "learning_rate": 0.0007,
877
+ "loss": 2.106010627746582,
878
+ "step": 2480
879
+ },
880
+ {
881
+ "epoch": 0.1683855746545332,
882
+ "eval_loss": 2.100916624069214,
883
+ "eval_runtime": 15.3711,
884
+ "eval_samples_per_second": 619.799,
885
+ "eval_steps_per_second": 0.651,
886
+ "step": 2498
887
+ },
888
+ {
889
+ "epoch": 0.16852039096730703,
890
+ "grad_norm": 0.33984375,
891
+ "learning_rate": 0.0007,
892
+ "loss": 2.087843322753906,
893
+ "step": 2500
894
+ },
895
+ {
896
+ "epoch": 0.16852039096730703,
897
+ "eval_loss": 2.1001250743865967,
898
+ "eval_runtime": 18.4156,
899
+ "eval_samples_per_second": 517.335,
900
+ "eval_steps_per_second": 0.543,
901
+ "step": 2500
902
+ },
903
+ {
904
+ "epoch": 0.1698685540950455,
905
+ "grad_norm": 20840448.0,
906
+ "learning_rate": 3.4200000000000005e-05,
907
+ "loss": 2.092716407775879,
908
+ "step": 2520
909
+ },
910
+ {
911
+ "epoch": 0.17121671722278395,
912
+ "grad_norm": 22937600.0,
913
+ "learning_rate": 7.020000000000001e-05,
914
+ "loss": 2.106246566772461,
915
+ "step": 2540
916
+ },
917
+ {
918
+ "epoch": 0.17256488035052242,
919
+ "grad_norm": 116391936.0,
920
+ "learning_rate": 9e-05,
921
+ "loss": 3.2435539245605467,
922
+ "step": 2560
923
+ },
924
+ {
925
+ "epoch": 0.17391304347826086,
926
+ "grad_norm": 283115520.0,
927
+ "learning_rate": 9e-05,
928
+ "loss": 7.010993957519531,
929
+ "step": 2580
930
+ },
931
+ {
932
+ "epoch": 0.17526120660599934,
933
+ "grad_norm": 583008256.0,
934
+ "learning_rate": 9e-05,
935
+ "loss": 7.576824188232422,
936
+ "step": 2600
937
+ },
938
+ {
939
+ "epoch": 0.17660936973373778,
940
+ "grad_norm": 998244352.0,
941
+ "learning_rate": 9e-05,
942
+ "loss": 7.7992095947265625,
943
+ "step": 2620
944
+ },
945
+ {
946
+ "epoch": 0.17795753286147623,
947
+ "grad_norm": 1501560832.0,
948
+ "learning_rate": 9e-05,
949
+ "loss": 7.901728820800781,
950
+ "step": 2640
951
+ },
952
+ {
953
+ "epoch": 0.1793056959892147,
954
+ "grad_norm": 2055208960.0,
955
+ "learning_rate": 9e-05,
956
+ "loss": 7.956874847412109,
957
+ "step": 2660
958
+ },
959
+ {
960
+ "epoch": 0.18065385911695314,
961
+ "grad_norm": 2650800128.0,
962
+ "learning_rate": 9e-05,
963
+ "loss": 7.995229339599609,
964
+ "step": 2680
965
+ },
966
+ {
967
+ "epoch": 0.1820020222446916,
968
+ "grad_norm": 3288334336.0,
969
+ "learning_rate": 9e-05,
970
+ "loss": 8.019290161132812,
971
+ "step": 2700
972
+ }
973
+ ],
974
+ "logging_steps": 20,
975
+ "max_steps": 3500,
976
+ "num_input_tokens_seen": 0,
977
+ "num_train_epochs": 1,
978
+ "save_steps": 100,
979
+ "stateful_callbacks": {
980
+ "TrainerControl": {
981
+ "args": {
982
+ "should_epoch_stop": false,
983
+ "should_evaluate": false,
984
+ "should_log": false,
985
+ "should_save": true,
986
+ "should_training_stop": false
987
+ },
988
+ "attributes": {}
989
+ }
990
+ },
991
+ "total_flos": 8710966267084800.0,
992
+ "train_batch_size": 64,
993
+ "trial_name": null,
994
+ "trial_params": null
995
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2700/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759f687547d62f080b691b398df9dd337d9e21ff107770065591d2088b29bf88
3
+ size 4920
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
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": 4
36
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58f4082f2f88c026c101e0df304af6056c1ac2a931aa7f8ee54951f931383b49
3
+ size 33967272
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f498a3d898abb05eb07250e74f328a4892f5788c2b82c00b8d99b7eecc768b1
3
+ size 39573050
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cf9097d4513154245c48236b6ec5137b7ee2a21c9f58f2cba798ea275c6026f
3
+ size 14244
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3524231f4a5ff6c2e972dce51861a8bc8472013528f64695abe2c7eea8f6783e
3
+ size 1064
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/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-gelu-100L_trash_run/checkpoint-2800/trainer_state.json ADDED
@@ -0,0 +1,1030 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.1887428378833839,
6
+ "eval_steps": 2498,
7
+ "global_step": 2800,
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.299579620361328,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.359375,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047666931152344,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.2734375,
29
+ "learning_rate": 8.259999999999999e-05,
30
+ "loss": 7.838191986083984,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 1.265625,
36
+ "learning_rate": 0.0001106,
37
+ "loss": 7.57310791015625,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.1953125,
43
+ "learning_rate": 0.0001386,
44
+ "loss": 7.233531951904297,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 1.1328125,
50
+ "learning_rate": 0.00016659999999999998,
51
+ "loss": 6.860941314697266,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 0.95703125,
57
+ "learning_rate": 0.00019460000000000001,
58
+ "loss": 6.51712646484375,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 0.81640625,
64
+ "learning_rate": 0.0002226,
65
+ "loss": 6.234881591796875,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 0.73828125,
71
+ "learning_rate": 0.00025059999999999997,
72
+ "loss": 6.024795913696289,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 1.0390625,
78
+ "learning_rate": 0.0002786,
79
+ "loss": 5.837321472167969,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.01482979440512302,
84
+ "grad_norm": 0.63671875,
85
+ "learning_rate": 0.00030659999999999997,
86
+ "loss": 5.647708129882813,
87
+ "step": 220
88
+ },
89
+ {
90
+ "epoch": 0.016177957532861477,
91
+ "grad_norm": 0.8984375,
92
+ "learning_rate": 0.0003346,
93
+ "loss": 5.43688735961914,
94
+ "step": 240
95
+ },
96
+ {
97
+ "epoch": 0.01752612066059993,
98
+ "grad_norm": 0.77734375,
99
+ "learning_rate": 0.00036260000000000003,
100
+ "loss": 5.210599517822265,
101
+ "step": 260
102
+ },
103
+ {
104
+ "epoch": 0.01887428378833839,
105
+ "grad_norm": 0.890625,
106
+ "learning_rate": 0.0003906,
107
+ "loss": 4.956959533691406,
108
+ "step": 280
109
+ },
110
+ {
111
+ "epoch": 0.020222446916076844,
112
+ "grad_norm": 0.84765625,
113
+ "learning_rate": 0.0004186,
114
+ "loss": 4.726874160766601,
115
+ "step": 300
116
+ },
117
+ {
118
+ "epoch": 0.021570610043815303,
119
+ "grad_norm": 0.87890625,
120
+ "learning_rate": 0.0004466,
121
+ "loss": 4.527945327758789,
122
+ "step": 320
123
+ },
124
+ {
125
+ "epoch": 0.022918773171553757,
126
+ "grad_norm": 1.3515625,
127
+ "learning_rate": 0.00047460000000000004,
128
+ "loss": 4.3517810821533205,
129
+ "step": 340
130
+ },
131
+ {
132
+ "epoch": 0.024266936299292215,
133
+ "grad_norm": 0.76171875,
134
+ "learning_rate": 0.0005026,
135
+ "loss": 4.212622451782226,
136
+ "step": 360
137
+ },
138
+ {
139
+ "epoch": 0.02561509942703067,
140
+ "grad_norm": 0.6640625,
141
+ "learning_rate": 0.0005306,
142
+ "loss": 4.076633453369141,
143
+ "step": 380
144
+ },
145
+ {
146
+ "epoch": 0.026963262554769128,
147
+ "grad_norm": 0.498046875,
148
+ "learning_rate": 0.0005586,
149
+ "loss": 3.9346649169921877,
150
+ "step": 400
151
+ },
152
+ {
153
+ "epoch": 0.028311425682507583,
154
+ "grad_norm": 0.85546875,
155
+ "learning_rate": 0.0005866,
156
+ "loss": 3.8310710906982424,
157
+ "step": 420
158
+ },
159
+ {
160
+ "epoch": 0.02965958881024604,
161
+ "grad_norm": 0.59375,
162
+ "learning_rate": 0.0006146,
163
+ "loss": 3.7676155090332033,
164
+ "step": 440
165
+ },
166
+ {
167
+ "epoch": 0.031007751937984496,
168
+ "grad_norm": 0.6171875,
169
+ "learning_rate": 0.0006426,
170
+ "loss": 3.6599285125732424,
171
+ "step": 460
172
+ },
173
+ {
174
+ "epoch": 0.032355915065722954,
175
+ "grad_norm": 0.5546875,
176
+ "learning_rate": 0.0006705999999999999,
177
+ "loss": 3.5886669158935547,
178
+ "step": 480
179
+ },
180
+ {
181
+ "epoch": 0.03370407819346141,
182
+ "grad_norm": 0.546875,
183
+ "learning_rate": 0.0006986,
184
+ "loss": 3.512950134277344,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 0.03505224132119986,
189
+ "grad_norm": 0.53125,
190
+ "learning_rate": 0.0007,
191
+ "loss": 3.444244384765625,
192
+ "step": 520
193
+ },
194
+ {
195
+ "epoch": 0.03640040444893832,
196
+ "grad_norm": 0.5078125,
197
+ "learning_rate": 0.0007,
198
+ "loss": 3.369425964355469,
199
+ "step": 540
200
+ },
201
+ {
202
+ "epoch": 0.03774856757667678,
203
+ "grad_norm": 0.5625,
204
+ "learning_rate": 0.0007,
205
+ "loss": 3.318771743774414,
206
+ "step": 560
207
+ },
208
+ {
209
+ "epoch": 0.03909673070441524,
210
+ "grad_norm": 0.470703125,
211
+ "learning_rate": 0.0007,
212
+ "loss": 3.2595756530761717,
213
+ "step": 580
214
+ },
215
+ {
216
+ "epoch": 0.04044489383215369,
217
+ "grad_norm": 0.53125,
218
+ "learning_rate": 0.0007,
219
+ "loss": 3.2241649627685547,
220
+ "step": 600
221
+ },
222
+ {
223
+ "epoch": 0.04179305695989215,
224
+ "grad_norm": 0.484375,
225
+ "learning_rate": 0.0007,
226
+ "loss": 3.1619989395141603,
227
+ "step": 620
228
+ },
229
+ {
230
+ "epoch": 0.043141220087630605,
231
+ "grad_norm": 0.5625,
232
+ "learning_rate": 0.0007,
233
+ "loss": 3.14036865234375,
234
+ "step": 640
235
+ },
236
+ {
237
+ "epoch": 0.044489383215369056,
238
+ "grad_norm": 0.41796875,
239
+ "learning_rate": 0.0007,
240
+ "loss": 3.083884620666504,
241
+ "step": 660
242
+ },
243
+ {
244
+ "epoch": 0.045837546343107514,
245
+ "grad_norm": 0.451171875,
246
+ "learning_rate": 0.0007,
247
+ "loss": 3.0591588973999024,
248
+ "step": 680
249
+ },
250
+ {
251
+ "epoch": 0.04718570947084597,
252
+ "grad_norm": 0.423828125,
253
+ "learning_rate": 0.0007,
254
+ "loss": 3.015497016906738,
255
+ "step": 700
256
+ },
257
+ {
258
+ "epoch": 0.04853387259858443,
259
+ "grad_norm": 0.41796875,
260
+ "learning_rate": 0.0007,
261
+ "loss": 2.9805700302124025,
262
+ "step": 720
263
+ },
264
+ {
265
+ "epoch": 0.04988203572632288,
266
+ "grad_norm": 0.46875,
267
+ "learning_rate": 0.0007,
268
+ "loss": 2.9426918029785156,
269
+ "step": 740
270
+ },
271
+ {
272
+ "epoch": 0.05123019885406134,
273
+ "grad_norm": 0.46484375,
274
+ "learning_rate": 0.0007,
275
+ "loss": 2.9274158477783203,
276
+ "step": 760
277
+ },
278
+ {
279
+ "epoch": 0.0525783619817998,
280
+ "grad_norm": 0.421875,
281
+ "learning_rate": 0.0007,
282
+ "loss": 2.931459426879883,
283
+ "step": 780
284
+ },
285
+ {
286
+ "epoch": 0.053926525109538256,
287
+ "grad_norm": 0.486328125,
288
+ "learning_rate": 0.0007,
289
+ "loss": 2.877642059326172,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.05527468823727671,
294
+ "grad_norm": 0.40234375,
295
+ "learning_rate": 0.0007,
296
+ "loss": 2.8646697998046875,
297
+ "step": 820
298
+ },
299
+ {
300
+ "epoch": 0.056622851365015166,
301
+ "grad_norm": 0.451171875,
302
+ "learning_rate": 0.0007,
303
+ "loss": 2.8328128814697267,
304
+ "step": 840
305
+ },
306
+ {
307
+ "epoch": 0.057971014492753624,
308
+ "grad_norm": 0.396484375,
309
+ "learning_rate": 0.0007,
310
+ "loss": 2.790274429321289,
311
+ "step": 860
312
+ },
313
+ {
314
+ "epoch": 0.05931917762049208,
315
+ "grad_norm": 0.423828125,
316
+ "learning_rate": 0.0007,
317
+ "loss": 2.7668125152587892,
318
+ "step": 880
319
+ },
320
+ {
321
+ "epoch": 0.06066734074823053,
322
+ "grad_norm": 0.427734375,
323
+ "learning_rate": 0.0007,
324
+ "loss": 2.7516881942749025,
325
+ "step": 900
326
+ },
327
+ {
328
+ "epoch": 0.06201550387596899,
329
+ "grad_norm": 0.42578125,
330
+ "learning_rate": 0.0007,
331
+ "loss": 2.722590446472168,
332
+ "step": 920
333
+ },
334
+ {
335
+ "epoch": 0.06336366700370745,
336
+ "grad_norm": 0.4375,
337
+ "learning_rate": 0.0007,
338
+ "loss": 2.708496856689453,
339
+ "step": 940
340
+ },
341
+ {
342
+ "epoch": 0.06471183013144591,
343
+ "grad_norm": 0.400390625,
344
+ "learning_rate": 0.0007,
345
+ "loss": 2.681654930114746,
346
+ "step": 960
347
+ },
348
+ {
349
+ "epoch": 0.06605999325918437,
350
+ "grad_norm": 0.396484375,
351
+ "learning_rate": 0.0007,
352
+ "loss": 2.6650571823120117,
353
+ "step": 980
354
+ },
355
+ {
356
+ "epoch": 0.06740815638692282,
357
+ "grad_norm": 0.41015625,
358
+ "learning_rate": 0.0007,
359
+ "loss": 2.6637720108032226,
360
+ "step": 1000
361
+ },
362
+ {
363
+ "epoch": 0.06875631951466127,
364
+ "grad_norm": 0.396484375,
365
+ "learning_rate": 0.0007,
366
+ "loss": 2.6311159133911133,
367
+ "step": 1020
368
+ },
369
+ {
370
+ "epoch": 0.07010448264239973,
371
+ "grad_norm": 0.408203125,
372
+ "learning_rate": 0.0007,
373
+ "loss": 2.615707015991211,
374
+ "step": 1040
375
+ },
376
+ {
377
+ "epoch": 0.07145264577013818,
378
+ "grad_norm": 0.404296875,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.6003103256225586,
381
+ "step": 1060
382
+ },
383
+ {
384
+ "epoch": 0.07280080889787664,
385
+ "grad_norm": 0.41015625,
386
+ "learning_rate": 0.0007,
387
+ "loss": 2.581348991394043,
388
+ "step": 1080
389
+ },
390
+ {
391
+ "epoch": 0.0741489720256151,
392
+ "grad_norm": 0.404296875,
393
+ "learning_rate": 0.0007,
394
+ "loss": 2.574792671203613,
395
+ "step": 1100
396
+ },
397
+ {
398
+ "epoch": 0.07549713515335356,
399
+ "grad_norm": 0.384765625,
400
+ "learning_rate": 0.0007,
401
+ "loss": 2.538400650024414,
402
+ "step": 1120
403
+ },
404
+ {
405
+ "epoch": 0.07684529828109202,
406
+ "grad_norm": 0.41796875,
407
+ "learning_rate": 0.0007,
408
+ "loss": 2.5249162673950196,
409
+ "step": 1140
410
+ },
411
+ {
412
+ "epoch": 0.07819346140883048,
413
+ "grad_norm": 0.404296875,
414
+ "learning_rate": 0.0007,
415
+ "loss": 2.5237232208251954,
416
+ "step": 1160
417
+ },
418
+ {
419
+ "epoch": 0.07954162453656892,
420
+ "grad_norm": 0.396484375,
421
+ "learning_rate": 0.0007,
422
+ "loss": 2.517318344116211,
423
+ "step": 1180
424
+ },
425
+ {
426
+ "epoch": 0.08088978766430738,
427
+ "grad_norm": 0.41796875,
428
+ "learning_rate": 0.0007,
429
+ "loss": 2.507699203491211,
430
+ "step": 1200
431
+ },
432
+ {
433
+ "epoch": 0.08223795079204584,
434
+ "grad_norm": 0.3828125,
435
+ "learning_rate": 0.0007,
436
+ "loss": 2.4932661056518555,
437
+ "step": 1220
438
+ },
439
+ {
440
+ "epoch": 0.0835861139197843,
441
+ "grad_norm": 0.365234375,
442
+ "learning_rate": 0.0007,
443
+ "loss": 2.4781061172485352,
444
+ "step": 1240
445
+ },
446
+ {
447
+ "epoch": 0.08493427704752275,
448
+ "grad_norm": 0.388671875,
449
+ "learning_rate": 0.0007,
450
+ "loss": 2.465612030029297,
451
+ "step": 1260
452
+ },
453
+ {
454
+ "epoch": 0.08628244017526121,
455
+ "grad_norm": 0.41796875,
456
+ "learning_rate": 0.0007,
457
+ "loss": 2.4463497161865235,
458
+ "step": 1280
459
+ },
460
+ {
461
+ "epoch": 0.08763060330299967,
462
+ "grad_norm": 0.390625,
463
+ "learning_rate": 0.0007,
464
+ "loss": 2.4374414443969727,
465
+ "step": 1300
466
+ },
467
+ {
468
+ "epoch": 0.08897876643073811,
469
+ "grad_norm": 0.35546875,
470
+ "learning_rate": 0.0007,
471
+ "loss": 2.4308101654052736,
472
+ "step": 1320
473
+ },
474
+ {
475
+ "epoch": 0.09032692955847657,
476
+ "grad_norm": 0.40234375,
477
+ "learning_rate": 0.0007,
478
+ "loss": 2.416130256652832,
479
+ "step": 1340
480
+ },
481
+ {
482
+ "epoch": 0.09167509268621503,
483
+ "grad_norm": 0.4296875,
484
+ "learning_rate": 0.0007,
485
+ "loss": 2.41524600982666,
486
+ "step": 1360
487
+ },
488
+ {
489
+ "epoch": 0.09302325581395349,
490
+ "grad_norm": 0.359375,
491
+ "learning_rate": 0.0007,
492
+ "loss": 2.392221450805664,
493
+ "step": 1380
494
+ },
495
+ {
496
+ "epoch": 0.09437141894169195,
497
+ "grad_norm": 0.373046875,
498
+ "learning_rate": 0.0007,
499
+ "loss": 2.383866310119629,
500
+ "step": 1400
501
+ },
502
+ {
503
+ "epoch": 0.0957195820694304,
504
+ "grad_norm": 0.396484375,
505
+ "learning_rate": 0.0007,
506
+ "loss": 2.394974708557129,
507
+ "step": 1420
508
+ },
509
+ {
510
+ "epoch": 0.09706774519716886,
511
+ "grad_norm": 0.396484375,
512
+ "learning_rate": 0.0007,
513
+ "loss": 2.3713239669799804,
514
+ "step": 1440
515
+ },
516
+ {
517
+ "epoch": 0.09841590832490732,
518
+ "grad_norm": 0.37109375,
519
+ "learning_rate": 0.0007,
520
+ "loss": 2.3597253799438476,
521
+ "step": 1460
522
+ },
523
+ {
524
+ "epoch": 0.09976407145264576,
525
+ "grad_norm": 0.34765625,
526
+ "learning_rate": 0.0007,
527
+ "loss": 2.3452421188354493,
528
+ "step": 1480
529
+ },
530
+ {
531
+ "epoch": 0.10111223458038422,
532
+ "grad_norm": 0.376953125,
533
+ "learning_rate": 0.0007,
534
+ "loss": 2.3336454391479493,
535
+ "step": 1500
536
+ },
537
+ {
538
+ "epoch": 0.10246039770812268,
539
+ "grad_norm": 0.39453125,
540
+ "learning_rate": 0.0007,
541
+ "loss": 2.3358110427856444,
542
+ "step": 1520
543
+ },
544
+ {
545
+ "epoch": 0.10380856083586114,
546
+ "grad_norm": 0.40625,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.324386787414551,
549
+ "step": 1540
550
+ },
551
+ {
552
+ "epoch": 0.1051567239635996,
553
+ "grad_norm": 0.376953125,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.313844108581543,
556
+ "step": 1560
557
+ },
558
+ {
559
+ "epoch": 0.10650488709133805,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.3294256210327147,
563
+ "step": 1580
564
+ },
565
+ {
566
+ "epoch": 0.10785305021907651,
567
+ "grad_norm": 0.40234375,
568
+ "learning_rate": 0.0007,
569
+ "loss": 2.297942352294922,
570
+ "step": 1600
571
+ },
572
+ {
573
+ "epoch": 0.10920121334681497,
574
+ "grad_norm": 0.357421875,
575
+ "learning_rate": 0.0007,
576
+ "loss": 2.296767997741699,
577
+ "step": 1620
578
+ },
579
+ {
580
+ "epoch": 0.11054937647455342,
581
+ "grad_norm": 0.37890625,
582
+ "learning_rate": 0.0007,
583
+ "loss": 2.3081844329833983,
584
+ "step": 1640
585
+ },
586
+ {
587
+ "epoch": 0.11189753960229187,
588
+ "grad_norm": 0.35546875,
589
+ "learning_rate": 0.0007,
590
+ "loss": 2.2883626937866213,
591
+ "step": 1660
592
+ },
593
+ {
594
+ "epoch": 0.11324570273003033,
595
+ "grad_norm": 0.375,
596
+ "learning_rate": 0.0007,
597
+ "loss": 2.272566223144531,
598
+ "step": 1680
599
+ },
600
+ {
601
+ "epoch": 0.11459386585776879,
602
+ "grad_norm": 0.373046875,
603
+ "learning_rate": 0.0007,
604
+ "loss": 2.281250762939453,
605
+ "step": 1700
606
+ },
607
+ {
608
+ "epoch": 0.11594202898550725,
609
+ "grad_norm": 0.41796875,
610
+ "learning_rate": 0.0007,
611
+ "loss": 2.257417106628418,
612
+ "step": 1720
613
+ },
614
+ {
615
+ "epoch": 0.1172901921132457,
616
+ "grad_norm": 0.365234375,
617
+ "learning_rate": 0.0007,
618
+ "loss": 2.2473636627197267,
619
+ "step": 1740
620
+ },
621
+ {
622
+ "epoch": 0.11863835524098416,
623
+ "grad_norm": 0.380859375,
624
+ "learning_rate": 0.0007,
625
+ "loss": 2.253912925720215,
626
+ "step": 1760
627
+ },
628
+ {
629
+ "epoch": 0.11998651836872262,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 0.0007,
632
+ "loss": 2.2672780990600585,
633
+ "step": 1780
634
+ },
635
+ {
636
+ "epoch": 0.12133468149646107,
637
+ "grad_norm": 0.3671875,
638
+ "learning_rate": 0.0007,
639
+ "loss": 2.261806106567383,
640
+ "step": 1800
641
+ },
642
+ {
643
+ "epoch": 0.12268284462419952,
644
+ "grad_norm": 0.375,
645
+ "learning_rate": 0.0007,
646
+ "loss": 2.2513105392456056,
647
+ "step": 1820
648
+ },
649
+ {
650
+ "epoch": 0.12403100775193798,
651
+ "grad_norm": 0.369140625,
652
+ "learning_rate": 0.0007,
653
+ "loss": 2.2201774597167967,
654
+ "step": 1840
655
+ },
656
+ {
657
+ "epoch": 0.12537917087967643,
658
+ "grad_norm": 0.35546875,
659
+ "learning_rate": 0.0007,
660
+ "loss": 2.2213924407958983,
661
+ "step": 1860
662
+ },
663
+ {
664
+ "epoch": 0.1267273340074149,
665
+ "grad_norm": 0.361328125,
666
+ "learning_rate": 0.0007,
667
+ "loss": 2.21158447265625,
668
+ "step": 1880
669
+ },
670
+ {
671
+ "epoch": 0.12807549713515334,
672
+ "grad_norm": 0.357421875,
673
+ "learning_rate": 0.0007,
674
+ "loss": 2.218313217163086,
675
+ "step": 1900
676
+ },
677
+ {
678
+ "epoch": 0.12942366026289182,
679
+ "grad_norm": 0.361328125,
680
+ "learning_rate": 0.0007,
681
+ "loss": 2.1961637496948243,
682
+ "step": 1920
683
+ },
684
+ {
685
+ "epoch": 0.13077182339063026,
686
+ "grad_norm": 0.369140625,
687
+ "learning_rate": 0.0007,
688
+ "loss": 2.2057926177978517,
689
+ "step": 1940
690
+ },
691
+ {
692
+ "epoch": 0.13211998651836873,
693
+ "grad_norm": 0.345703125,
694
+ "learning_rate": 0.0007,
695
+ "loss": 2.181648826599121,
696
+ "step": 1960
697
+ },
698
+ {
699
+ "epoch": 0.13346814964610718,
700
+ "grad_norm": 0.380859375,
701
+ "learning_rate": 0.0007,
702
+ "loss": 2.203908920288086,
703
+ "step": 1980
704
+ },
705
+ {
706
+ "epoch": 0.13481631277384565,
707
+ "grad_norm": 0.341796875,
708
+ "learning_rate": 0.0007,
709
+ "loss": 2.1758956909179688,
710
+ "step": 2000
711
+ },
712
+ {
713
+ "epoch": 0.1361644759015841,
714
+ "grad_norm": 0.35546875,
715
+ "learning_rate": 0.0007,
716
+ "loss": 2.185748481750488,
717
+ "step": 2020
718
+ },
719
+ {
720
+ "epoch": 0.13751263902932254,
721
+ "grad_norm": 0.41796875,
722
+ "learning_rate": 0.0007,
723
+ "loss": 2.165380668640137,
724
+ "step": 2040
725
+ },
726
+ {
727
+ "epoch": 0.138860802157061,
728
+ "grad_norm": 0.3515625,
729
+ "learning_rate": 0.0007,
730
+ "loss": 2.1852161407470705,
731
+ "step": 2060
732
+ },
733
+ {
734
+ "epoch": 0.14020896528479945,
735
+ "grad_norm": 0.34765625,
736
+ "learning_rate": 0.0007,
737
+ "loss": 2.160782051086426,
738
+ "step": 2080
739
+ },
740
+ {
741
+ "epoch": 0.14155712841253792,
742
+ "grad_norm": 0.337890625,
743
+ "learning_rate": 0.0007,
744
+ "loss": 2.152023124694824,
745
+ "step": 2100
746
+ },
747
+ {
748
+ "epoch": 0.14290529154027637,
749
+ "grad_norm": 0.359375,
750
+ "learning_rate": 0.0007,
751
+ "loss": 2.160308837890625,
752
+ "step": 2120
753
+ },
754
+ {
755
+ "epoch": 0.14425345466801484,
756
+ "grad_norm": 0.34765625,
757
+ "learning_rate": 0.0007,
758
+ "loss": 2.1609643936157226,
759
+ "step": 2140
760
+ },
761
+ {
762
+ "epoch": 0.14560161779575329,
763
+ "grad_norm": 0.357421875,
764
+ "learning_rate": 0.0007,
765
+ "loss": 2.1475265502929686,
766
+ "step": 2160
767
+ },
768
+ {
769
+ "epoch": 0.14694978092349173,
770
+ "grad_norm": 0.345703125,
771
+ "learning_rate": 0.0007,
772
+ "loss": 2.155929183959961,
773
+ "step": 2180
774
+ },
775
+ {
776
+ "epoch": 0.1482979440512302,
777
+ "grad_norm": 0.341796875,
778
+ "learning_rate": 0.0007,
779
+ "loss": 2.1293373107910156,
780
+ "step": 2200
781
+ },
782
+ {
783
+ "epoch": 0.14964610717896865,
784
+ "grad_norm": 0.35546875,
785
+ "learning_rate": 0.0007,
786
+ "loss": 2.137188720703125,
787
+ "step": 2220
788
+ },
789
+ {
790
+ "epoch": 0.15099427030670712,
791
+ "grad_norm": 0.345703125,
792
+ "learning_rate": 0.0007,
793
+ "loss": 2.1462854385375976,
794
+ "step": 2240
795
+ },
796
+ {
797
+ "epoch": 0.15234243343444556,
798
+ "grad_norm": 0.365234375,
799
+ "learning_rate": 0.0007,
800
+ "loss": 2.1255794525146485,
801
+ "step": 2260
802
+ },
803
+ {
804
+ "epoch": 0.15369059656218403,
805
+ "grad_norm": 0.33984375,
806
+ "learning_rate": 0.0007,
807
+ "loss": 2.1365018844604493,
808
+ "step": 2280
809
+ },
810
+ {
811
+ "epoch": 0.15503875968992248,
812
+ "grad_norm": 0.337890625,
813
+ "learning_rate": 0.0007,
814
+ "loss": 2.1164464950561523,
815
+ "step": 2300
816
+ },
817
+ {
818
+ "epoch": 0.15638692281766095,
819
+ "grad_norm": 0.361328125,
820
+ "learning_rate": 0.0007,
821
+ "loss": 2.127320671081543,
822
+ "step": 2320
823
+ },
824
+ {
825
+ "epoch": 0.1577350859453994,
826
+ "grad_norm": 0.3359375,
827
+ "learning_rate": 0.0007,
828
+ "loss": 2.117641067504883,
829
+ "step": 2340
830
+ },
831
+ {
832
+ "epoch": 0.15908324907313784,
833
+ "grad_norm": 0.357421875,
834
+ "learning_rate": 0.0007,
835
+ "loss": 2.119377326965332,
836
+ "step": 2360
837
+ },
838
+ {
839
+ "epoch": 0.1604314122008763,
840
+ "grad_norm": 0.33203125,
841
+ "learning_rate": 0.0007,
842
+ "loss": 2.121909332275391,
843
+ "step": 2380
844
+ },
845
+ {
846
+ "epoch": 0.16177957532861476,
847
+ "grad_norm": 0.361328125,
848
+ "learning_rate": 0.0007,
849
+ "loss": 2.10235538482666,
850
+ "step": 2400
851
+ },
852
+ {
853
+ "epoch": 0.16312773845635323,
854
+ "grad_norm": 0.353515625,
855
+ "learning_rate": 0.0007,
856
+ "loss": 2.1056692123413088,
857
+ "step": 2420
858
+ },
859
+ {
860
+ "epoch": 0.16447590158409167,
861
+ "grad_norm": 0.3359375,
862
+ "learning_rate": 0.0007,
863
+ "loss": 2.093578910827637,
864
+ "step": 2440
865
+ },
866
+ {
867
+ "epoch": 0.16582406471183014,
868
+ "grad_norm": 0.337890625,
869
+ "learning_rate": 0.0007,
870
+ "loss": 2.1059795379638673,
871
+ "step": 2460
872
+ },
873
+ {
874
+ "epoch": 0.1671722278395686,
875
+ "grad_norm": 0.361328125,
876
+ "learning_rate": 0.0007,
877
+ "loss": 2.106010627746582,
878
+ "step": 2480
879
+ },
880
+ {
881
+ "epoch": 0.1683855746545332,
882
+ "eval_loss": 2.100916624069214,
883
+ "eval_runtime": 15.3711,
884
+ "eval_samples_per_second": 619.799,
885
+ "eval_steps_per_second": 0.651,
886
+ "step": 2498
887
+ },
888
+ {
889
+ "epoch": 0.16852039096730703,
890
+ "grad_norm": 0.33984375,
891
+ "learning_rate": 0.0007,
892
+ "loss": 2.087843322753906,
893
+ "step": 2500
894
+ },
895
+ {
896
+ "epoch": 0.16852039096730703,
897
+ "eval_loss": 2.1001250743865967,
898
+ "eval_runtime": 18.4156,
899
+ "eval_samples_per_second": 517.335,
900
+ "eval_steps_per_second": 0.543,
901
+ "step": 2500
902
+ },
903
+ {
904
+ "epoch": 0.1698685540950455,
905
+ "grad_norm": 20840448.0,
906
+ "learning_rate": 3.4200000000000005e-05,
907
+ "loss": 2.092716407775879,
908
+ "step": 2520
909
+ },
910
+ {
911
+ "epoch": 0.17121671722278395,
912
+ "grad_norm": 22937600.0,
913
+ "learning_rate": 7.020000000000001e-05,
914
+ "loss": 2.106246566772461,
915
+ "step": 2540
916
+ },
917
+ {
918
+ "epoch": 0.17256488035052242,
919
+ "grad_norm": 116391936.0,
920
+ "learning_rate": 9e-05,
921
+ "loss": 3.2435539245605467,
922
+ "step": 2560
923
+ },
924
+ {
925
+ "epoch": 0.17391304347826086,
926
+ "grad_norm": 283115520.0,
927
+ "learning_rate": 9e-05,
928
+ "loss": 7.010993957519531,
929
+ "step": 2580
930
+ },
931
+ {
932
+ "epoch": 0.17526120660599934,
933
+ "grad_norm": 583008256.0,
934
+ "learning_rate": 9e-05,
935
+ "loss": 7.576824188232422,
936
+ "step": 2600
937
+ },
938
+ {
939
+ "epoch": 0.17660936973373778,
940
+ "grad_norm": 998244352.0,
941
+ "learning_rate": 9e-05,
942
+ "loss": 7.7992095947265625,
943
+ "step": 2620
944
+ },
945
+ {
946
+ "epoch": 0.17795753286147623,
947
+ "grad_norm": 1501560832.0,
948
+ "learning_rate": 9e-05,
949
+ "loss": 7.901728820800781,
950
+ "step": 2640
951
+ },
952
+ {
953
+ "epoch": 0.1793056959892147,
954
+ "grad_norm": 2055208960.0,
955
+ "learning_rate": 9e-05,
956
+ "loss": 7.956874847412109,
957
+ "step": 2660
958
+ },
959
+ {
960
+ "epoch": 0.18065385911695314,
961
+ "grad_norm": 2650800128.0,
962
+ "learning_rate": 9e-05,
963
+ "loss": 7.995229339599609,
964
+ "step": 2680
965
+ },
966
+ {
967
+ "epoch": 0.1820020222446916,
968
+ "grad_norm": 3288334336.0,
969
+ "learning_rate": 9e-05,
970
+ "loss": 8.019290161132812,
971
+ "step": 2700
972
+ },
973
+ {
974
+ "epoch": 0.18335018537243006,
975
+ "grad_norm": 3942645760.0,
976
+ "learning_rate": 9e-05,
977
+ "loss": 8.043710327148437,
978
+ "step": 2720
979
+ },
980
+ {
981
+ "epoch": 0.18469834850016853,
982
+ "grad_norm": 4630511616.0,
983
+ "learning_rate": 9e-05,
984
+ "loss": 8.043434143066406,
985
+ "step": 2740
986
+ },
987
+ {
988
+ "epoch": 0.18604651162790697,
989
+ "grad_norm": 5301600256.0,
990
+ "learning_rate": 9e-05,
991
+ "loss": 8.049758911132812,
992
+ "step": 2760
993
+ },
994
+ {
995
+ "epoch": 0.18739467475564545,
996
+ "grad_norm": 5972688896.0,
997
+ "learning_rate": 9e-05,
998
+ "loss": 8.061854553222656,
999
+ "step": 2780
1000
+ },
1001
+ {
1002
+ "epoch": 0.1887428378833839,
1003
+ "grad_norm": 6643777536.0,
1004
+ "learning_rate": 9e-05,
1005
+ "loss": 8.073146057128906,
1006
+ "step": 2800
1007
+ }
1008
+ ],
1009
+ "logging_steps": 20,
1010
+ "max_steps": 3500,
1011
+ "num_input_tokens_seen": 0,
1012
+ "num_train_epochs": 1,
1013
+ "save_steps": 100,
1014
+ "stateful_callbacks": {
1015
+ "TrainerControl": {
1016
+ "args": {
1017
+ "should_epoch_stop": false,
1018
+ "should_evaluate": false,
1019
+ "should_log": false,
1020
+ "should_save": true,
1021
+ "should_training_stop": false
1022
+ },
1023
+ "attributes": {}
1024
+ }
1025
+ },
1026
+ "total_flos": 9033594647347200.0,
1027
+ "train_batch_size": 64,
1028
+ "trial_name": null,
1029
+ "trial_params": null
1030
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2800/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759f687547d62f080b691b398df9dd337d9e21ff107770065591d2088b29bf88
3
+ size 4920
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
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": 4
36
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46b1f3b47a281e5aa8b49b5caea9a61f044ab9ef4947eb33d1b656f295584170
3
+ size 33967272
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e4c7c344176ec3b1ae75937a9c708127c748abd8cf8c6ec9b9f9a34962f9bee
3
+ size 39573050
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cf9097d4513154245c48236b6ec5137b7ee2a21c9f58f2cba798ea275c6026f
3
+ size 14244
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66f5b68450b0c18ded7ac53967162b3ffa6171c24c7dd24167f487c9909bda68
3
+ size 1064
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/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-gelu-100L_trash_run/checkpoint-2900/trainer_state.json ADDED
@@ -0,0 +1,1065 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.19548365352207617,
6
+ "eval_steps": 2498,
7
+ "global_step": 2900,
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.299579620361328,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.359375,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047666931152344,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.2734375,
29
+ "learning_rate": 8.259999999999999e-05,
30
+ "loss": 7.838191986083984,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 1.265625,
36
+ "learning_rate": 0.0001106,
37
+ "loss": 7.57310791015625,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.1953125,
43
+ "learning_rate": 0.0001386,
44
+ "loss": 7.233531951904297,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 1.1328125,
50
+ "learning_rate": 0.00016659999999999998,
51
+ "loss": 6.860941314697266,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 0.95703125,
57
+ "learning_rate": 0.00019460000000000001,
58
+ "loss": 6.51712646484375,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 0.81640625,
64
+ "learning_rate": 0.0002226,
65
+ "loss": 6.234881591796875,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 0.73828125,
71
+ "learning_rate": 0.00025059999999999997,
72
+ "loss": 6.024795913696289,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 1.0390625,
78
+ "learning_rate": 0.0002786,
79
+ "loss": 5.837321472167969,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.01482979440512302,
84
+ "grad_norm": 0.63671875,
85
+ "learning_rate": 0.00030659999999999997,
86
+ "loss": 5.647708129882813,
87
+ "step": 220
88
+ },
89
+ {
90
+ "epoch": 0.016177957532861477,
91
+ "grad_norm": 0.8984375,
92
+ "learning_rate": 0.0003346,
93
+ "loss": 5.43688735961914,
94
+ "step": 240
95
+ },
96
+ {
97
+ "epoch": 0.01752612066059993,
98
+ "grad_norm": 0.77734375,
99
+ "learning_rate": 0.00036260000000000003,
100
+ "loss": 5.210599517822265,
101
+ "step": 260
102
+ },
103
+ {
104
+ "epoch": 0.01887428378833839,
105
+ "grad_norm": 0.890625,
106
+ "learning_rate": 0.0003906,
107
+ "loss": 4.956959533691406,
108
+ "step": 280
109
+ },
110
+ {
111
+ "epoch": 0.020222446916076844,
112
+ "grad_norm": 0.84765625,
113
+ "learning_rate": 0.0004186,
114
+ "loss": 4.726874160766601,
115
+ "step": 300
116
+ },
117
+ {
118
+ "epoch": 0.021570610043815303,
119
+ "grad_norm": 0.87890625,
120
+ "learning_rate": 0.0004466,
121
+ "loss": 4.527945327758789,
122
+ "step": 320
123
+ },
124
+ {
125
+ "epoch": 0.022918773171553757,
126
+ "grad_norm": 1.3515625,
127
+ "learning_rate": 0.00047460000000000004,
128
+ "loss": 4.3517810821533205,
129
+ "step": 340
130
+ },
131
+ {
132
+ "epoch": 0.024266936299292215,
133
+ "grad_norm": 0.76171875,
134
+ "learning_rate": 0.0005026,
135
+ "loss": 4.212622451782226,
136
+ "step": 360
137
+ },
138
+ {
139
+ "epoch": 0.02561509942703067,
140
+ "grad_norm": 0.6640625,
141
+ "learning_rate": 0.0005306,
142
+ "loss": 4.076633453369141,
143
+ "step": 380
144
+ },
145
+ {
146
+ "epoch": 0.026963262554769128,
147
+ "grad_norm": 0.498046875,
148
+ "learning_rate": 0.0005586,
149
+ "loss": 3.9346649169921877,
150
+ "step": 400
151
+ },
152
+ {
153
+ "epoch": 0.028311425682507583,
154
+ "grad_norm": 0.85546875,
155
+ "learning_rate": 0.0005866,
156
+ "loss": 3.8310710906982424,
157
+ "step": 420
158
+ },
159
+ {
160
+ "epoch": 0.02965958881024604,
161
+ "grad_norm": 0.59375,
162
+ "learning_rate": 0.0006146,
163
+ "loss": 3.7676155090332033,
164
+ "step": 440
165
+ },
166
+ {
167
+ "epoch": 0.031007751937984496,
168
+ "grad_norm": 0.6171875,
169
+ "learning_rate": 0.0006426,
170
+ "loss": 3.6599285125732424,
171
+ "step": 460
172
+ },
173
+ {
174
+ "epoch": 0.032355915065722954,
175
+ "grad_norm": 0.5546875,
176
+ "learning_rate": 0.0006705999999999999,
177
+ "loss": 3.5886669158935547,
178
+ "step": 480
179
+ },
180
+ {
181
+ "epoch": 0.03370407819346141,
182
+ "grad_norm": 0.546875,
183
+ "learning_rate": 0.0006986,
184
+ "loss": 3.512950134277344,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 0.03505224132119986,
189
+ "grad_norm": 0.53125,
190
+ "learning_rate": 0.0007,
191
+ "loss": 3.444244384765625,
192
+ "step": 520
193
+ },
194
+ {
195
+ "epoch": 0.03640040444893832,
196
+ "grad_norm": 0.5078125,
197
+ "learning_rate": 0.0007,
198
+ "loss": 3.369425964355469,
199
+ "step": 540
200
+ },
201
+ {
202
+ "epoch": 0.03774856757667678,
203
+ "grad_norm": 0.5625,
204
+ "learning_rate": 0.0007,
205
+ "loss": 3.318771743774414,
206
+ "step": 560
207
+ },
208
+ {
209
+ "epoch": 0.03909673070441524,
210
+ "grad_norm": 0.470703125,
211
+ "learning_rate": 0.0007,
212
+ "loss": 3.2595756530761717,
213
+ "step": 580
214
+ },
215
+ {
216
+ "epoch": 0.04044489383215369,
217
+ "grad_norm": 0.53125,
218
+ "learning_rate": 0.0007,
219
+ "loss": 3.2241649627685547,
220
+ "step": 600
221
+ },
222
+ {
223
+ "epoch": 0.04179305695989215,
224
+ "grad_norm": 0.484375,
225
+ "learning_rate": 0.0007,
226
+ "loss": 3.1619989395141603,
227
+ "step": 620
228
+ },
229
+ {
230
+ "epoch": 0.043141220087630605,
231
+ "grad_norm": 0.5625,
232
+ "learning_rate": 0.0007,
233
+ "loss": 3.14036865234375,
234
+ "step": 640
235
+ },
236
+ {
237
+ "epoch": 0.044489383215369056,
238
+ "grad_norm": 0.41796875,
239
+ "learning_rate": 0.0007,
240
+ "loss": 3.083884620666504,
241
+ "step": 660
242
+ },
243
+ {
244
+ "epoch": 0.045837546343107514,
245
+ "grad_norm": 0.451171875,
246
+ "learning_rate": 0.0007,
247
+ "loss": 3.0591588973999024,
248
+ "step": 680
249
+ },
250
+ {
251
+ "epoch": 0.04718570947084597,
252
+ "grad_norm": 0.423828125,
253
+ "learning_rate": 0.0007,
254
+ "loss": 3.015497016906738,
255
+ "step": 700
256
+ },
257
+ {
258
+ "epoch": 0.04853387259858443,
259
+ "grad_norm": 0.41796875,
260
+ "learning_rate": 0.0007,
261
+ "loss": 2.9805700302124025,
262
+ "step": 720
263
+ },
264
+ {
265
+ "epoch": 0.04988203572632288,
266
+ "grad_norm": 0.46875,
267
+ "learning_rate": 0.0007,
268
+ "loss": 2.9426918029785156,
269
+ "step": 740
270
+ },
271
+ {
272
+ "epoch": 0.05123019885406134,
273
+ "grad_norm": 0.46484375,
274
+ "learning_rate": 0.0007,
275
+ "loss": 2.9274158477783203,
276
+ "step": 760
277
+ },
278
+ {
279
+ "epoch": 0.0525783619817998,
280
+ "grad_norm": 0.421875,
281
+ "learning_rate": 0.0007,
282
+ "loss": 2.931459426879883,
283
+ "step": 780
284
+ },
285
+ {
286
+ "epoch": 0.053926525109538256,
287
+ "grad_norm": 0.486328125,
288
+ "learning_rate": 0.0007,
289
+ "loss": 2.877642059326172,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.05527468823727671,
294
+ "grad_norm": 0.40234375,
295
+ "learning_rate": 0.0007,
296
+ "loss": 2.8646697998046875,
297
+ "step": 820
298
+ },
299
+ {
300
+ "epoch": 0.056622851365015166,
301
+ "grad_norm": 0.451171875,
302
+ "learning_rate": 0.0007,
303
+ "loss": 2.8328128814697267,
304
+ "step": 840
305
+ },
306
+ {
307
+ "epoch": 0.057971014492753624,
308
+ "grad_norm": 0.396484375,
309
+ "learning_rate": 0.0007,
310
+ "loss": 2.790274429321289,
311
+ "step": 860
312
+ },
313
+ {
314
+ "epoch": 0.05931917762049208,
315
+ "grad_norm": 0.423828125,
316
+ "learning_rate": 0.0007,
317
+ "loss": 2.7668125152587892,
318
+ "step": 880
319
+ },
320
+ {
321
+ "epoch": 0.06066734074823053,
322
+ "grad_norm": 0.427734375,
323
+ "learning_rate": 0.0007,
324
+ "loss": 2.7516881942749025,
325
+ "step": 900
326
+ },
327
+ {
328
+ "epoch": 0.06201550387596899,
329
+ "grad_norm": 0.42578125,
330
+ "learning_rate": 0.0007,
331
+ "loss": 2.722590446472168,
332
+ "step": 920
333
+ },
334
+ {
335
+ "epoch": 0.06336366700370745,
336
+ "grad_norm": 0.4375,
337
+ "learning_rate": 0.0007,
338
+ "loss": 2.708496856689453,
339
+ "step": 940
340
+ },
341
+ {
342
+ "epoch": 0.06471183013144591,
343
+ "grad_norm": 0.400390625,
344
+ "learning_rate": 0.0007,
345
+ "loss": 2.681654930114746,
346
+ "step": 960
347
+ },
348
+ {
349
+ "epoch": 0.06605999325918437,
350
+ "grad_norm": 0.396484375,
351
+ "learning_rate": 0.0007,
352
+ "loss": 2.6650571823120117,
353
+ "step": 980
354
+ },
355
+ {
356
+ "epoch": 0.06740815638692282,
357
+ "grad_norm": 0.41015625,
358
+ "learning_rate": 0.0007,
359
+ "loss": 2.6637720108032226,
360
+ "step": 1000
361
+ },
362
+ {
363
+ "epoch": 0.06875631951466127,
364
+ "grad_norm": 0.396484375,
365
+ "learning_rate": 0.0007,
366
+ "loss": 2.6311159133911133,
367
+ "step": 1020
368
+ },
369
+ {
370
+ "epoch": 0.07010448264239973,
371
+ "grad_norm": 0.408203125,
372
+ "learning_rate": 0.0007,
373
+ "loss": 2.615707015991211,
374
+ "step": 1040
375
+ },
376
+ {
377
+ "epoch": 0.07145264577013818,
378
+ "grad_norm": 0.404296875,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.6003103256225586,
381
+ "step": 1060
382
+ },
383
+ {
384
+ "epoch": 0.07280080889787664,
385
+ "grad_norm": 0.41015625,
386
+ "learning_rate": 0.0007,
387
+ "loss": 2.581348991394043,
388
+ "step": 1080
389
+ },
390
+ {
391
+ "epoch": 0.0741489720256151,
392
+ "grad_norm": 0.404296875,
393
+ "learning_rate": 0.0007,
394
+ "loss": 2.574792671203613,
395
+ "step": 1100
396
+ },
397
+ {
398
+ "epoch": 0.07549713515335356,
399
+ "grad_norm": 0.384765625,
400
+ "learning_rate": 0.0007,
401
+ "loss": 2.538400650024414,
402
+ "step": 1120
403
+ },
404
+ {
405
+ "epoch": 0.07684529828109202,
406
+ "grad_norm": 0.41796875,
407
+ "learning_rate": 0.0007,
408
+ "loss": 2.5249162673950196,
409
+ "step": 1140
410
+ },
411
+ {
412
+ "epoch": 0.07819346140883048,
413
+ "grad_norm": 0.404296875,
414
+ "learning_rate": 0.0007,
415
+ "loss": 2.5237232208251954,
416
+ "step": 1160
417
+ },
418
+ {
419
+ "epoch": 0.07954162453656892,
420
+ "grad_norm": 0.396484375,
421
+ "learning_rate": 0.0007,
422
+ "loss": 2.517318344116211,
423
+ "step": 1180
424
+ },
425
+ {
426
+ "epoch": 0.08088978766430738,
427
+ "grad_norm": 0.41796875,
428
+ "learning_rate": 0.0007,
429
+ "loss": 2.507699203491211,
430
+ "step": 1200
431
+ },
432
+ {
433
+ "epoch": 0.08223795079204584,
434
+ "grad_norm": 0.3828125,
435
+ "learning_rate": 0.0007,
436
+ "loss": 2.4932661056518555,
437
+ "step": 1220
438
+ },
439
+ {
440
+ "epoch": 0.0835861139197843,
441
+ "grad_norm": 0.365234375,
442
+ "learning_rate": 0.0007,
443
+ "loss": 2.4781061172485352,
444
+ "step": 1240
445
+ },
446
+ {
447
+ "epoch": 0.08493427704752275,
448
+ "grad_norm": 0.388671875,
449
+ "learning_rate": 0.0007,
450
+ "loss": 2.465612030029297,
451
+ "step": 1260
452
+ },
453
+ {
454
+ "epoch": 0.08628244017526121,
455
+ "grad_norm": 0.41796875,
456
+ "learning_rate": 0.0007,
457
+ "loss": 2.4463497161865235,
458
+ "step": 1280
459
+ },
460
+ {
461
+ "epoch": 0.08763060330299967,
462
+ "grad_norm": 0.390625,
463
+ "learning_rate": 0.0007,
464
+ "loss": 2.4374414443969727,
465
+ "step": 1300
466
+ },
467
+ {
468
+ "epoch": 0.08897876643073811,
469
+ "grad_norm": 0.35546875,
470
+ "learning_rate": 0.0007,
471
+ "loss": 2.4308101654052736,
472
+ "step": 1320
473
+ },
474
+ {
475
+ "epoch": 0.09032692955847657,
476
+ "grad_norm": 0.40234375,
477
+ "learning_rate": 0.0007,
478
+ "loss": 2.416130256652832,
479
+ "step": 1340
480
+ },
481
+ {
482
+ "epoch": 0.09167509268621503,
483
+ "grad_norm": 0.4296875,
484
+ "learning_rate": 0.0007,
485
+ "loss": 2.41524600982666,
486
+ "step": 1360
487
+ },
488
+ {
489
+ "epoch": 0.09302325581395349,
490
+ "grad_norm": 0.359375,
491
+ "learning_rate": 0.0007,
492
+ "loss": 2.392221450805664,
493
+ "step": 1380
494
+ },
495
+ {
496
+ "epoch": 0.09437141894169195,
497
+ "grad_norm": 0.373046875,
498
+ "learning_rate": 0.0007,
499
+ "loss": 2.383866310119629,
500
+ "step": 1400
501
+ },
502
+ {
503
+ "epoch": 0.0957195820694304,
504
+ "grad_norm": 0.396484375,
505
+ "learning_rate": 0.0007,
506
+ "loss": 2.394974708557129,
507
+ "step": 1420
508
+ },
509
+ {
510
+ "epoch": 0.09706774519716886,
511
+ "grad_norm": 0.396484375,
512
+ "learning_rate": 0.0007,
513
+ "loss": 2.3713239669799804,
514
+ "step": 1440
515
+ },
516
+ {
517
+ "epoch": 0.09841590832490732,
518
+ "grad_norm": 0.37109375,
519
+ "learning_rate": 0.0007,
520
+ "loss": 2.3597253799438476,
521
+ "step": 1460
522
+ },
523
+ {
524
+ "epoch": 0.09976407145264576,
525
+ "grad_norm": 0.34765625,
526
+ "learning_rate": 0.0007,
527
+ "loss": 2.3452421188354493,
528
+ "step": 1480
529
+ },
530
+ {
531
+ "epoch": 0.10111223458038422,
532
+ "grad_norm": 0.376953125,
533
+ "learning_rate": 0.0007,
534
+ "loss": 2.3336454391479493,
535
+ "step": 1500
536
+ },
537
+ {
538
+ "epoch": 0.10246039770812268,
539
+ "grad_norm": 0.39453125,
540
+ "learning_rate": 0.0007,
541
+ "loss": 2.3358110427856444,
542
+ "step": 1520
543
+ },
544
+ {
545
+ "epoch": 0.10380856083586114,
546
+ "grad_norm": 0.40625,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.324386787414551,
549
+ "step": 1540
550
+ },
551
+ {
552
+ "epoch": 0.1051567239635996,
553
+ "grad_norm": 0.376953125,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.313844108581543,
556
+ "step": 1560
557
+ },
558
+ {
559
+ "epoch": 0.10650488709133805,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.3294256210327147,
563
+ "step": 1580
564
+ },
565
+ {
566
+ "epoch": 0.10785305021907651,
567
+ "grad_norm": 0.40234375,
568
+ "learning_rate": 0.0007,
569
+ "loss": 2.297942352294922,
570
+ "step": 1600
571
+ },
572
+ {
573
+ "epoch": 0.10920121334681497,
574
+ "grad_norm": 0.357421875,
575
+ "learning_rate": 0.0007,
576
+ "loss": 2.296767997741699,
577
+ "step": 1620
578
+ },
579
+ {
580
+ "epoch": 0.11054937647455342,
581
+ "grad_norm": 0.37890625,
582
+ "learning_rate": 0.0007,
583
+ "loss": 2.3081844329833983,
584
+ "step": 1640
585
+ },
586
+ {
587
+ "epoch": 0.11189753960229187,
588
+ "grad_norm": 0.35546875,
589
+ "learning_rate": 0.0007,
590
+ "loss": 2.2883626937866213,
591
+ "step": 1660
592
+ },
593
+ {
594
+ "epoch": 0.11324570273003033,
595
+ "grad_norm": 0.375,
596
+ "learning_rate": 0.0007,
597
+ "loss": 2.272566223144531,
598
+ "step": 1680
599
+ },
600
+ {
601
+ "epoch": 0.11459386585776879,
602
+ "grad_norm": 0.373046875,
603
+ "learning_rate": 0.0007,
604
+ "loss": 2.281250762939453,
605
+ "step": 1700
606
+ },
607
+ {
608
+ "epoch": 0.11594202898550725,
609
+ "grad_norm": 0.41796875,
610
+ "learning_rate": 0.0007,
611
+ "loss": 2.257417106628418,
612
+ "step": 1720
613
+ },
614
+ {
615
+ "epoch": 0.1172901921132457,
616
+ "grad_norm": 0.365234375,
617
+ "learning_rate": 0.0007,
618
+ "loss": 2.2473636627197267,
619
+ "step": 1740
620
+ },
621
+ {
622
+ "epoch": 0.11863835524098416,
623
+ "grad_norm": 0.380859375,
624
+ "learning_rate": 0.0007,
625
+ "loss": 2.253912925720215,
626
+ "step": 1760
627
+ },
628
+ {
629
+ "epoch": 0.11998651836872262,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 0.0007,
632
+ "loss": 2.2672780990600585,
633
+ "step": 1780
634
+ },
635
+ {
636
+ "epoch": 0.12133468149646107,
637
+ "grad_norm": 0.3671875,
638
+ "learning_rate": 0.0007,
639
+ "loss": 2.261806106567383,
640
+ "step": 1800
641
+ },
642
+ {
643
+ "epoch": 0.12268284462419952,
644
+ "grad_norm": 0.375,
645
+ "learning_rate": 0.0007,
646
+ "loss": 2.2513105392456056,
647
+ "step": 1820
648
+ },
649
+ {
650
+ "epoch": 0.12403100775193798,
651
+ "grad_norm": 0.369140625,
652
+ "learning_rate": 0.0007,
653
+ "loss": 2.2201774597167967,
654
+ "step": 1840
655
+ },
656
+ {
657
+ "epoch": 0.12537917087967643,
658
+ "grad_norm": 0.35546875,
659
+ "learning_rate": 0.0007,
660
+ "loss": 2.2213924407958983,
661
+ "step": 1860
662
+ },
663
+ {
664
+ "epoch": 0.1267273340074149,
665
+ "grad_norm": 0.361328125,
666
+ "learning_rate": 0.0007,
667
+ "loss": 2.21158447265625,
668
+ "step": 1880
669
+ },
670
+ {
671
+ "epoch": 0.12807549713515334,
672
+ "grad_norm": 0.357421875,
673
+ "learning_rate": 0.0007,
674
+ "loss": 2.218313217163086,
675
+ "step": 1900
676
+ },
677
+ {
678
+ "epoch": 0.12942366026289182,
679
+ "grad_norm": 0.361328125,
680
+ "learning_rate": 0.0007,
681
+ "loss": 2.1961637496948243,
682
+ "step": 1920
683
+ },
684
+ {
685
+ "epoch": 0.13077182339063026,
686
+ "grad_norm": 0.369140625,
687
+ "learning_rate": 0.0007,
688
+ "loss": 2.2057926177978517,
689
+ "step": 1940
690
+ },
691
+ {
692
+ "epoch": 0.13211998651836873,
693
+ "grad_norm": 0.345703125,
694
+ "learning_rate": 0.0007,
695
+ "loss": 2.181648826599121,
696
+ "step": 1960
697
+ },
698
+ {
699
+ "epoch": 0.13346814964610718,
700
+ "grad_norm": 0.380859375,
701
+ "learning_rate": 0.0007,
702
+ "loss": 2.203908920288086,
703
+ "step": 1980
704
+ },
705
+ {
706
+ "epoch": 0.13481631277384565,
707
+ "grad_norm": 0.341796875,
708
+ "learning_rate": 0.0007,
709
+ "loss": 2.1758956909179688,
710
+ "step": 2000
711
+ },
712
+ {
713
+ "epoch": 0.1361644759015841,
714
+ "grad_norm": 0.35546875,
715
+ "learning_rate": 0.0007,
716
+ "loss": 2.185748481750488,
717
+ "step": 2020
718
+ },
719
+ {
720
+ "epoch": 0.13751263902932254,
721
+ "grad_norm": 0.41796875,
722
+ "learning_rate": 0.0007,
723
+ "loss": 2.165380668640137,
724
+ "step": 2040
725
+ },
726
+ {
727
+ "epoch": 0.138860802157061,
728
+ "grad_norm": 0.3515625,
729
+ "learning_rate": 0.0007,
730
+ "loss": 2.1852161407470705,
731
+ "step": 2060
732
+ },
733
+ {
734
+ "epoch": 0.14020896528479945,
735
+ "grad_norm": 0.34765625,
736
+ "learning_rate": 0.0007,
737
+ "loss": 2.160782051086426,
738
+ "step": 2080
739
+ },
740
+ {
741
+ "epoch": 0.14155712841253792,
742
+ "grad_norm": 0.337890625,
743
+ "learning_rate": 0.0007,
744
+ "loss": 2.152023124694824,
745
+ "step": 2100
746
+ },
747
+ {
748
+ "epoch": 0.14290529154027637,
749
+ "grad_norm": 0.359375,
750
+ "learning_rate": 0.0007,
751
+ "loss": 2.160308837890625,
752
+ "step": 2120
753
+ },
754
+ {
755
+ "epoch": 0.14425345466801484,
756
+ "grad_norm": 0.34765625,
757
+ "learning_rate": 0.0007,
758
+ "loss": 2.1609643936157226,
759
+ "step": 2140
760
+ },
761
+ {
762
+ "epoch": 0.14560161779575329,
763
+ "grad_norm": 0.357421875,
764
+ "learning_rate": 0.0007,
765
+ "loss": 2.1475265502929686,
766
+ "step": 2160
767
+ },
768
+ {
769
+ "epoch": 0.14694978092349173,
770
+ "grad_norm": 0.345703125,
771
+ "learning_rate": 0.0007,
772
+ "loss": 2.155929183959961,
773
+ "step": 2180
774
+ },
775
+ {
776
+ "epoch": 0.1482979440512302,
777
+ "grad_norm": 0.341796875,
778
+ "learning_rate": 0.0007,
779
+ "loss": 2.1293373107910156,
780
+ "step": 2200
781
+ },
782
+ {
783
+ "epoch": 0.14964610717896865,
784
+ "grad_norm": 0.35546875,
785
+ "learning_rate": 0.0007,
786
+ "loss": 2.137188720703125,
787
+ "step": 2220
788
+ },
789
+ {
790
+ "epoch": 0.15099427030670712,
791
+ "grad_norm": 0.345703125,
792
+ "learning_rate": 0.0007,
793
+ "loss": 2.1462854385375976,
794
+ "step": 2240
795
+ },
796
+ {
797
+ "epoch": 0.15234243343444556,
798
+ "grad_norm": 0.365234375,
799
+ "learning_rate": 0.0007,
800
+ "loss": 2.1255794525146485,
801
+ "step": 2260
802
+ },
803
+ {
804
+ "epoch": 0.15369059656218403,
805
+ "grad_norm": 0.33984375,
806
+ "learning_rate": 0.0007,
807
+ "loss": 2.1365018844604493,
808
+ "step": 2280
809
+ },
810
+ {
811
+ "epoch": 0.15503875968992248,
812
+ "grad_norm": 0.337890625,
813
+ "learning_rate": 0.0007,
814
+ "loss": 2.1164464950561523,
815
+ "step": 2300
816
+ },
817
+ {
818
+ "epoch": 0.15638692281766095,
819
+ "grad_norm": 0.361328125,
820
+ "learning_rate": 0.0007,
821
+ "loss": 2.127320671081543,
822
+ "step": 2320
823
+ },
824
+ {
825
+ "epoch": 0.1577350859453994,
826
+ "grad_norm": 0.3359375,
827
+ "learning_rate": 0.0007,
828
+ "loss": 2.117641067504883,
829
+ "step": 2340
830
+ },
831
+ {
832
+ "epoch": 0.15908324907313784,
833
+ "grad_norm": 0.357421875,
834
+ "learning_rate": 0.0007,
835
+ "loss": 2.119377326965332,
836
+ "step": 2360
837
+ },
838
+ {
839
+ "epoch": 0.1604314122008763,
840
+ "grad_norm": 0.33203125,
841
+ "learning_rate": 0.0007,
842
+ "loss": 2.121909332275391,
843
+ "step": 2380
844
+ },
845
+ {
846
+ "epoch": 0.16177957532861476,
847
+ "grad_norm": 0.361328125,
848
+ "learning_rate": 0.0007,
849
+ "loss": 2.10235538482666,
850
+ "step": 2400
851
+ },
852
+ {
853
+ "epoch": 0.16312773845635323,
854
+ "grad_norm": 0.353515625,
855
+ "learning_rate": 0.0007,
856
+ "loss": 2.1056692123413088,
857
+ "step": 2420
858
+ },
859
+ {
860
+ "epoch": 0.16447590158409167,
861
+ "grad_norm": 0.3359375,
862
+ "learning_rate": 0.0007,
863
+ "loss": 2.093578910827637,
864
+ "step": 2440
865
+ },
866
+ {
867
+ "epoch": 0.16582406471183014,
868
+ "grad_norm": 0.337890625,
869
+ "learning_rate": 0.0007,
870
+ "loss": 2.1059795379638673,
871
+ "step": 2460
872
+ },
873
+ {
874
+ "epoch": 0.1671722278395686,
875
+ "grad_norm": 0.361328125,
876
+ "learning_rate": 0.0007,
877
+ "loss": 2.106010627746582,
878
+ "step": 2480
879
+ },
880
+ {
881
+ "epoch": 0.1683855746545332,
882
+ "eval_loss": 2.100916624069214,
883
+ "eval_runtime": 15.3711,
884
+ "eval_samples_per_second": 619.799,
885
+ "eval_steps_per_second": 0.651,
886
+ "step": 2498
887
+ },
888
+ {
889
+ "epoch": 0.16852039096730703,
890
+ "grad_norm": 0.33984375,
891
+ "learning_rate": 0.0007,
892
+ "loss": 2.087843322753906,
893
+ "step": 2500
894
+ },
895
+ {
896
+ "epoch": 0.16852039096730703,
897
+ "eval_loss": 2.1001250743865967,
898
+ "eval_runtime": 18.4156,
899
+ "eval_samples_per_second": 517.335,
900
+ "eval_steps_per_second": 0.543,
901
+ "step": 2500
902
+ },
903
+ {
904
+ "epoch": 0.1698685540950455,
905
+ "grad_norm": 20840448.0,
906
+ "learning_rate": 3.4200000000000005e-05,
907
+ "loss": 2.092716407775879,
908
+ "step": 2520
909
+ },
910
+ {
911
+ "epoch": 0.17121671722278395,
912
+ "grad_norm": 22937600.0,
913
+ "learning_rate": 7.020000000000001e-05,
914
+ "loss": 2.106246566772461,
915
+ "step": 2540
916
+ },
917
+ {
918
+ "epoch": 0.17256488035052242,
919
+ "grad_norm": 116391936.0,
920
+ "learning_rate": 9e-05,
921
+ "loss": 3.2435539245605467,
922
+ "step": 2560
923
+ },
924
+ {
925
+ "epoch": 0.17391304347826086,
926
+ "grad_norm": 283115520.0,
927
+ "learning_rate": 9e-05,
928
+ "loss": 7.010993957519531,
929
+ "step": 2580
930
+ },
931
+ {
932
+ "epoch": 0.17526120660599934,
933
+ "grad_norm": 583008256.0,
934
+ "learning_rate": 9e-05,
935
+ "loss": 7.576824188232422,
936
+ "step": 2600
937
+ },
938
+ {
939
+ "epoch": 0.17660936973373778,
940
+ "grad_norm": 998244352.0,
941
+ "learning_rate": 9e-05,
942
+ "loss": 7.7992095947265625,
943
+ "step": 2620
944
+ },
945
+ {
946
+ "epoch": 0.17795753286147623,
947
+ "grad_norm": 1501560832.0,
948
+ "learning_rate": 9e-05,
949
+ "loss": 7.901728820800781,
950
+ "step": 2640
951
+ },
952
+ {
953
+ "epoch": 0.1793056959892147,
954
+ "grad_norm": 2055208960.0,
955
+ "learning_rate": 9e-05,
956
+ "loss": 7.956874847412109,
957
+ "step": 2660
958
+ },
959
+ {
960
+ "epoch": 0.18065385911695314,
961
+ "grad_norm": 2650800128.0,
962
+ "learning_rate": 9e-05,
963
+ "loss": 7.995229339599609,
964
+ "step": 2680
965
+ },
966
+ {
967
+ "epoch": 0.1820020222446916,
968
+ "grad_norm": 3288334336.0,
969
+ "learning_rate": 9e-05,
970
+ "loss": 8.019290161132812,
971
+ "step": 2700
972
+ },
973
+ {
974
+ "epoch": 0.18335018537243006,
975
+ "grad_norm": 3942645760.0,
976
+ "learning_rate": 9e-05,
977
+ "loss": 8.043710327148437,
978
+ "step": 2720
979
+ },
980
+ {
981
+ "epoch": 0.18469834850016853,
982
+ "grad_norm": 4630511616.0,
983
+ "learning_rate": 9e-05,
984
+ "loss": 8.043434143066406,
985
+ "step": 2740
986
+ },
987
+ {
988
+ "epoch": 0.18604651162790697,
989
+ "grad_norm": 5301600256.0,
990
+ "learning_rate": 9e-05,
991
+ "loss": 8.049758911132812,
992
+ "step": 2760
993
+ },
994
+ {
995
+ "epoch": 0.18739467475564545,
996
+ "grad_norm": 5972688896.0,
997
+ "learning_rate": 9e-05,
998
+ "loss": 8.061854553222656,
999
+ "step": 2780
1000
+ },
1001
+ {
1002
+ "epoch": 0.1887428378833839,
1003
+ "grad_norm": 6643777536.0,
1004
+ "learning_rate": 9e-05,
1005
+ "loss": 8.073146057128906,
1006
+ "step": 2800
1007
+ },
1008
+ {
1009
+ "epoch": 0.19009100101112233,
1010
+ "grad_norm": 7281311744.0,
1011
+ "learning_rate": 9e-05,
1012
+ "loss": 8.076551818847657,
1013
+ "step": 2820
1014
+ },
1015
+ {
1016
+ "epoch": 0.1914391641388608,
1017
+ "grad_norm": 7918845952.0,
1018
+ "learning_rate": 9e-05,
1019
+ "loss": 8.078580474853515,
1020
+ "step": 2840
1021
+ },
1022
+ {
1023
+ "epoch": 0.19278732726659925,
1024
+ "grad_norm": 8556380160.0,
1025
+ "learning_rate": 9e-05,
1026
+ "loss": 8.073567199707032,
1027
+ "step": 2860
1028
+ },
1029
+ {
1030
+ "epoch": 0.19413549039433772,
1031
+ "grad_norm": 9126805504.0,
1032
+ "learning_rate": 9e-05,
1033
+ "loss": 8.08355941772461,
1034
+ "step": 2880
1035
+ },
1036
+ {
1037
+ "epoch": 0.19548365352207617,
1038
+ "grad_norm": 9730785280.0,
1039
+ "learning_rate": 9e-05,
1040
+ "loss": 8.085595703125,
1041
+ "step": 2900
1042
+ }
1043
+ ],
1044
+ "logging_steps": 20,
1045
+ "max_steps": 3500,
1046
+ "num_input_tokens_seen": 0,
1047
+ "num_train_epochs": 1,
1048
+ "save_steps": 100,
1049
+ "stateful_callbacks": {
1050
+ "TrainerControl": {
1051
+ "args": {
1052
+ "should_epoch_stop": false,
1053
+ "should_evaluate": false,
1054
+ "should_log": false,
1055
+ "should_save": true,
1056
+ "should_training_stop": false
1057
+ },
1058
+ "attributes": {}
1059
+ }
1060
+ },
1061
+ "total_flos": 9356223027609600.0,
1062
+ "train_batch_size": 64,
1063
+ "trial_name": null,
1064
+ "trial_params": null
1065
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-2900/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759f687547d62f080b691b398df9dd337d9e21ff107770065591d2088b29bf88
3
+ size 4920
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
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": 4
36
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c206a196f35a15b24090dba6aa11c1ce9e25cba82e1dc48eee4e54150e1bac3
3
+ size 33967272
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b6ddac25130191cf09616e808c04db1512eeea40be2fdfe894b460b668f3b60
3
+ size 39573050
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cf9097d4513154245c48236b6ec5137b7ee2a21c9f58f2cba798ea275c6026f
3
+ size 14244
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5138b027fc3060283bbcf135873185cb5a8f65583d5c7597645469d2c1529ff7
3
+ size 1064
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/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-gelu-100L_trash_run/checkpoint-3000/trainer_state.json ADDED
@@ -0,0 +1,1100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.20222446916076844,
6
+ "eval_steps": 2498,
7
+ "global_step": 3000,
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.299579620361328,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.359375,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047666931152344,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.2734375,
29
+ "learning_rate": 8.259999999999999e-05,
30
+ "loss": 7.838191986083984,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 1.265625,
36
+ "learning_rate": 0.0001106,
37
+ "loss": 7.57310791015625,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.1953125,
43
+ "learning_rate": 0.0001386,
44
+ "loss": 7.233531951904297,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 1.1328125,
50
+ "learning_rate": 0.00016659999999999998,
51
+ "loss": 6.860941314697266,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 0.95703125,
57
+ "learning_rate": 0.00019460000000000001,
58
+ "loss": 6.51712646484375,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 0.81640625,
64
+ "learning_rate": 0.0002226,
65
+ "loss": 6.234881591796875,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 0.73828125,
71
+ "learning_rate": 0.00025059999999999997,
72
+ "loss": 6.024795913696289,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 1.0390625,
78
+ "learning_rate": 0.0002786,
79
+ "loss": 5.837321472167969,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.01482979440512302,
84
+ "grad_norm": 0.63671875,
85
+ "learning_rate": 0.00030659999999999997,
86
+ "loss": 5.647708129882813,
87
+ "step": 220
88
+ },
89
+ {
90
+ "epoch": 0.016177957532861477,
91
+ "grad_norm": 0.8984375,
92
+ "learning_rate": 0.0003346,
93
+ "loss": 5.43688735961914,
94
+ "step": 240
95
+ },
96
+ {
97
+ "epoch": 0.01752612066059993,
98
+ "grad_norm": 0.77734375,
99
+ "learning_rate": 0.00036260000000000003,
100
+ "loss": 5.210599517822265,
101
+ "step": 260
102
+ },
103
+ {
104
+ "epoch": 0.01887428378833839,
105
+ "grad_norm": 0.890625,
106
+ "learning_rate": 0.0003906,
107
+ "loss": 4.956959533691406,
108
+ "step": 280
109
+ },
110
+ {
111
+ "epoch": 0.020222446916076844,
112
+ "grad_norm": 0.84765625,
113
+ "learning_rate": 0.0004186,
114
+ "loss": 4.726874160766601,
115
+ "step": 300
116
+ },
117
+ {
118
+ "epoch": 0.021570610043815303,
119
+ "grad_norm": 0.87890625,
120
+ "learning_rate": 0.0004466,
121
+ "loss": 4.527945327758789,
122
+ "step": 320
123
+ },
124
+ {
125
+ "epoch": 0.022918773171553757,
126
+ "grad_norm": 1.3515625,
127
+ "learning_rate": 0.00047460000000000004,
128
+ "loss": 4.3517810821533205,
129
+ "step": 340
130
+ },
131
+ {
132
+ "epoch": 0.024266936299292215,
133
+ "grad_norm": 0.76171875,
134
+ "learning_rate": 0.0005026,
135
+ "loss": 4.212622451782226,
136
+ "step": 360
137
+ },
138
+ {
139
+ "epoch": 0.02561509942703067,
140
+ "grad_norm": 0.6640625,
141
+ "learning_rate": 0.0005306,
142
+ "loss": 4.076633453369141,
143
+ "step": 380
144
+ },
145
+ {
146
+ "epoch": 0.026963262554769128,
147
+ "grad_norm": 0.498046875,
148
+ "learning_rate": 0.0005586,
149
+ "loss": 3.9346649169921877,
150
+ "step": 400
151
+ },
152
+ {
153
+ "epoch": 0.028311425682507583,
154
+ "grad_norm": 0.85546875,
155
+ "learning_rate": 0.0005866,
156
+ "loss": 3.8310710906982424,
157
+ "step": 420
158
+ },
159
+ {
160
+ "epoch": 0.02965958881024604,
161
+ "grad_norm": 0.59375,
162
+ "learning_rate": 0.0006146,
163
+ "loss": 3.7676155090332033,
164
+ "step": 440
165
+ },
166
+ {
167
+ "epoch": 0.031007751937984496,
168
+ "grad_norm": 0.6171875,
169
+ "learning_rate": 0.0006426,
170
+ "loss": 3.6599285125732424,
171
+ "step": 460
172
+ },
173
+ {
174
+ "epoch": 0.032355915065722954,
175
+ "grad_norm": 0.5546875,
176
+ "learning_rate": 0.0006705999999999999,
177
+ "loss": 3.5886669158935547,
178
+ "step": 480
179
+ },
180
+ {
181
+ "epoch": 0.03370407819346141,
182
+ "grad_norm": 0.546875,
183
+ "learning_rate": 0.0006986,
184
+ "loss": 3.512950134277344,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 0.03505224132119986,
189
+ "grad_norm": 0.53125,
190
+ "learning_rate": 0.0007,
191
+ "loss": 3.444244384765625,
192
+ "step": 520
193
+ },
194
+ {
195
+ "epoch": 0.03640040444893832,
196
+ "grad_norm": 0.5078125,
197
+ "learning_rate": 0.0007,
198
+ "loss": 3.369425964355469,
199
+ "step": 540
200
+ },
201
+ {
202
+ "epoch": 0.03774856757667678,
203
+ "grad_norm": 0.5625,
204
+ "learning_rate": 0.0007,
205
+ "loss": 3.318771743774414,
206
+ "step": 560
207
+ },
208
+ {
209
+ "epoch": 0.03909673070441524,
210
+ "grad_norm": 0.470703125,
211
+ "learning_rate": 0.0007,
212
+ "loss": 3.2595756530761717,
213
+ "step": 580
214
+ },
215
+ {
216
+ "epoch": 0.04044489383215369,
217
+ "grad_norm": 0.53125,
218
+ "learning_rate": 0.0007,
219
+ "loss": 3.2241649627685547,
220
+ "step": 600
221
+ },
222
+ {
223
+ "epoch": 0.04179305695989215,
224
+ "grad_norm": 0.484375,
225
+ "learning_rate": 0.0007,
226
+ "loss": 3.1619989395141603,
227
+ "step": 620
228
+ },
229
+ {
230
+ "epoch": 0.043141220087630605,
231
+ "grad_norm": 0.5625,
232
+ "learning_rate": 0.0007,
233
+ "loss": 3.14036865234375,
234
+ "step": 640
235
+ },
236
+ {
237
+ "epoch": 0.044489383215369056,
238
+ "grad_norm": 0.41796875,
239
+ "learning_rate": 0.0007,
240
+ "loss": 3.083884620666504,
241
+ "step": 660
242
+ },
243
+ {
244
+ "epoch": 0.045837546343107514,
245
+ "grad_norm": 0.451171875,
246
+ "learning_rate": 0.0007,
247
+ "loss": 3.0591588973999024,
248
+ "step": 680
249
+ },
250
+ {
251
+ "epoch": 0.04718570947084597,
252
+ "grad_norm": 0.423828125,
253
+ "learning_rate": 0.0007,
254
+ "loss": 3.015497016906738,
255
+ "step": 700
256
+ },
257
+ {
258
+ "epoch": 0.04853387259858443,
259
+ "grad_norm": 0.41796875,
260
+ "learning_rate": 0.0007,
261
+ "loss": 2.9805700302124025,
262
+ "step": 720
263
+ },
264
+ {
265
+ "epoch": 0.04988203572632288,
266
+ "grad_norm": 0.46875,
267
+ "learning_rate": 0.0007,
268
+ "loss": 2.9426918029785156,
269
+ "step": 740
270
+ },
271
+ {
272
+ "epoch": 0.05123019885406134,
273
+ "grad_norm": 0.46484375,
274
+ "learning_rate": 0.0007,
275
+ "loss": 2.9274158477783203,
276
+ "step": 760
277
+ },
278
+ {
279
+ "epoch": 0.0525783619817998,
280
+ "grad_norm": 0.421875,
281
+ "learning_rate": 0.0007,
282
+ "loss": 2.931459426879883,
283
+ "step": 780
284
+ },
285
+ {
286
+ "epoch": 0.053926525109538256,
287
+ "grad_norm": 0.486328125,
288
+ "learning_rate": 0.0007,
289
+ "loss": 2.877642059326172,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.05527468823727671,
294
+ "grad_norm": 0.40234375,
295
+ "learning_rate": 0.0007,
296
+ "loss": 2.8646697998046875,
297
+ "step": 820
298
+ },
299
+ {
300
+ "epoch": 0.056622851365015166,
301
+ "grad_norm": 0.451171875,
302
+ "learning_rate": 0.0007,
303
+ "loss": 2.8328128814697267,
304
+ "step": 840
305
+ },
306
+ {
307
+ "epoch": 0.057971014492753624,
308
+ "grad_norm": 0.396484375,
309
+ "learning_rate": 0.0007,
310
+ "loss": 2.790274429321289,
311
+ "step": 860
312
+ },
313
+ {
314
+ "epoch": 0.05931917762049208,
315
+ "grad_norm": 0.423828125,
316
+ "learning_rate": 0.0007,
317
+ "loss": 2.7668125152587892,
318
+ "step": 880
319
+ },
320
+ {
321
+ "epoch": 0.06066734074823053,
322
+ "grad_norm": 0.427734375,
323
+ "learning_rate": 0.0007,
324
+ "loss": 2.7516881942749025,
325
+ "step": 900
326
+ },
327
+ {
328
+ "epoch": 0.06201550387596899,
329
+ "grad_norm": 0.42578125,
330
+ "learning_rate": 0.0007,
331
+ "loss": 2.722590446472168,
332
+ "step": 920
333
+ },
334
+ {
335
+ "epoch": 0.06336366700370745,
336
+ "grad_norm": 0.4375,
337
+ "learning_rate": 0.0007,
338
+ "loss": 2.708496856689453,
339
+ "step": 940
340
+ },
341
+ {
342
+ "epoch": 0.06471183013144591,
343
+ "grad_norm": 0.400390625,
344
+ "learning_rate": 0.0007,
345
+ "loss": 2.681654930114746,
346
+ "step": 960
347
+ },
348
+ {
349
+ "epoch": 0.06605999325918437,
350
+ "grad_norm": 0.396484375,
351
+ "learning_rate": 0.0007,
352
+ "loss": 2.6650571823120117,
353
+ "step": 980
354
+ },
355
+ {
356
+ "epoch": 0.06740815638692282,
357
+ "grad_norm": 0.41015625,
358
+ "learning_rate": 0.0007,
359
+ "loss": 2.6637720108032226,
360
+ "step": 1000
361
+ },
362
+ {
363
+ "epoch": 0.06875631951466127,
364
+ "grad_norm": 0.396484375,
365
+ "learning_rate": 0.0007,
366
+ "loss": 2.6311159133911133,
367
+ "step": 1020
368
+ },
369
+ {
370
+ "epoch": 0.07010448264239973,
371
+ "grad_norm": 0.408203125,
372
+ "learning_rate": 0.0007,
373
+ "loss": 2.615707015991211,
374
+ "step": 1040
375
+ },
376
+ {
377
+ "epoch": 0.07145264577013818,
378
+ "grad_norm": 0.404296875,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.6003103256225586,
381
+ "step": 1060
382
+ },
383
+ {
384
+ "epoch": 0.07280080889787664,
385
+ "grad_norm": 0.41015625,
386
+ "learning_rate": 0.0007,
387
+ "loss": 2.581348991394043,
388
+ "step": 1080
389
+ },
390
+ {
391
+ "epoch": 0.0741489720256151,
392
+ "grad_norm": 0.404296875,
393
+ "learning_rate": 0.0007,
394
+ "loss": 2.574792671203613,
395
+ "step": 1100
396
+ },
397
+ {
398
+ "epoch": 0.07549713515335356,
399
+ "grad_norm": 0.384765625,
400
+ "learning_rate": 0.0007,
401
+ "loss": 2.538400650024414,
402
+ "step": 1120
403
+ },
404
+ {
405
+ "epoch": 0.07684529828109202,
406
+ "grad_norm": 0.41796875,
407
+ "learning_rate": 0.0007,
408
+ "loss": 2.5249162673950196,
409
+ "step": 1140
410
+ },
411
+ {
412
+ "epoch": 0.07819346140883048,
413
+ "grad_norm": 0.404296875,
414
+ "learning_rate": 0.0007,
415
+ "loss": 2.5237232208251954,
416
+ "step": 1160
417
+ },
418
+ {
419
+ "epoch": 0.07954162453656892,
420
+ "grad_norm": 0.396484375,
421
+ "learning_rate": 0.0007,
422
+ "loss": 2.517318344116211,
423
+ "step": 1180
424
+ },
425
+ {
426
+ "epoch": 0.08088978766430738,
427
+ "grad_norm": 0.41796875,
428
+ "learning_rate": 0.0007,
429
+ "loss": 2.507699203491211,
430
+ "step": 1200
431
+ },
432
+ {
433
+ "epoch": 0.08223795079204584,
434
+ "grad_norm": 0.3828125,
435
+ "learning_rate": 0.0007,
436
+ "loss": 2.4932661056518555,
437
+ "step": 1220
438
+ },
439
+ {
440
+ "epoch": 0.0835861139197843,
441
+ "grad_norm": 0.365234375,
442
+ "learning_rate": 0.0007,
443
+ "loss": 2.4781061172485352,
444
+ "step": 1240
445
+ },
446
+ {
447
+ "epoch": 0.08493427704752275,
448
+ "grad_norm": 0.388671875,
449
+ "learning_rate": 0.0007,
450
+ "loss": 2.465612030029297,
451
+ "step": 1260
452
+ },
453
+ {
454
+ "epoch": 0.08628244017526121,
455
+ "grad_norm": 0.41796875,
456
+ "learning_rate": 0.0007,
457
+ "loss": 2.4463497161865235,
458
+ "step": 1280
459
+ },
460
+ {
461
+ "epoch": 0.08763060330299967,
462
+ "grad_norm": 0.390625,
463
+ "learning_rate": 0.0007,
464
+ "loss": 2.4374414443969727,
465
+ "step": 1300
466
+ },
467
+ {
468
+ "epoch": 0.08897876643073811,
469
+ "grad_norm": 0.35546875,
470
+ "learning_rate": 0.0007,
471
+ "loss": 2.4308101654052736,
472
+ "step": 1320
473
+ },
474
+ {
475
+ "epoch": 0.09032692955847657,
476
+ "grad_norm": 0.40234375,
477
+ "learning_rate": 0.0007,
478
+ "loss": 2.416130256652832,
479
+ "step": 1340
480
+ },
481
+ {
482
+ "epoch": 0.09167509268621503,
483
+ "grad_norm": 0.4296875,
484
+ "learning_rate": 0.0007,
485
+ "loss": 2.41524600982666,
486
+ "step": 1360
487
+ },
488
+ {
489
+ "epoch": 0.09302325581395349,
490
+ "grad_norm": 0.359375,
491
+ "learning_rate": 0.0007,
492
+ "loss": 2.392221450805664,
493
+ "step": 1380
494
+ },
495
+ {
496
+ "epoch": 0.09437141894169195,
497
+ "grad_norm": 0.373046875,
498
+ "learning_rate": 0.0007,
499
+ "loss": 2.383866310119629,
500
+ "step": 1400
501
+ },
502
+ {
503
+ "epoch": 0.0957195820694304,
504
+ "grad_norm": 0.396484375,
505
+ "learning_rate": 0.0007,
506
+ "loss": 2.394974708557129,
507
+ "step": 1420
508
+ },
509
+ {
510
+ "epoch": 0.09706774519716886,
511
+ "grad_norm": 0.396484375,
512
+ "learning_rate": 0.0007,
513
+ "loss": 2.3713239669799804,
514
+ "step": 1440
515
+ },
516
+ {
517
+ "epoch": 0.09841590832490732,
518
+ "grad_norm": 0.37109375,
519
+ "learning_rate": 0.0007,
520
+ "loss": 2.3597253799438476,
521
+ "step": 1460
522
+ },
523
+ {
524
+ "epoch": 0.09976407145264576,
525
+ "grad_norm": 0.34765625,
526
+ "learning_rate": 0.0007,
527
+ "loss": 2.3452421188354493,
528
+ "step": 1480
529
+ },
530
+ {
531
+ "epoch": 0.10111223458038422,
532
+ "grad_norm": 0.376953125,
533
+ "learning_rate": 0.0007,
534
+ "loss": 2.3336454391479493,
535
+ "step": 1500
536
+ },
537
+ {
538
+ "epoch": 0.10246039770812268,
539
+ "grad_norm": 0.39453125,
540
+ "learning_rate": 0.0007,
541
+ "loss": 2.3358110427856444,
542
+ "step": 1520
543
+ },
544
+ {
545
+ "epoch": 0.10380856083586114,
546
+ "grad_norm": 0.40625,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.324386787414551,
549
+ "step": 1540
550
+ },
551
+ {
552
+ "epoch": 0.1051567239635996,
553
+ "grad_norm": 0.376953125,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.313844108581543,
556
+ "step": 1560
557
+ },
558
+ {
559
+ "epoch": 0.10650488709133805,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.3294256210327147,
563
+ "step": 1580
564
+ },
565
+ {
566
+ "epoch": 0.10785305021907651,
567
+ "grad_norm": 0.40234375,
568
+ "learning_rate": 0.0007,
569
+ "loss": 2.297942352294922,
570
+ "step": 1600
571
+ },
572
+ {
573
+ "epoch": 0.10920121334681497,
574
+ "grad_norm": 0.357421875,
575
+ "learning_rate": 0.0007,
576
+ "loss": 2.296767997741699,
577
+ "step": 1620
578
+ },
579
+ {
580
+ "epoch": 0.11054937647455342,
581
+ "grad_norm": 0.37890625,
582
+ "learning_rate": 0.0007,
583
+ "loss": 2.3081844329833983,
584
+ "step": 1640
585
+ },
586
+ {
587
+ "epoch": 0.11189753960229187,
588
+ "grad_norm": 0.35546875,
589
+ "learning_rate": 0.0007,
590
+ "loss": 2.2883626937866213,
591
+ "step": 1660
592
+ },
593
+ {
594
+ "epoch": 0.11324570273003033,
595
+ "grad_norm": 0.375,
596
+ "learning_rate": 0.0007,
597
+ "loss": 2.272566223144531,
598
+ "step": 1680
599
+ },
600
+ {
601
+ "epoch": 0.11459386585776879,
602
+ "grad_norm": 0.373046875,
603
+ "learning_rate": 0.0007,
604
+ "loss": 2.281250762939453,
605
+ "step": 1700
606
+ },
607
+ {
608
+ "epoch": 0.11594202898550725,
609
+ "grad_norm": 0.41796875,
610
+ "learning_rate": 0.0007,
611
+ "loss": 2.257417106628418,
612
+ "step": 1720
613
+ },
614
+ {
615
+ "epoch": 0.1172901921132457,
616
+ "grad_norm": 0.365234375,
617
+ "learning_rate": 0.0007,
618
+ "loss": 2.2473636627197267,
619
+ "step": 1740
620
+ },
621
+ {
622
+ "epoch": 0.11863835524098416,
623
+ "grad_norm": 0.380859375,
624
+ "learning_rate": 0.0007,
625
+ "loss": 2.253912925720215,
626
+ "step": 1760
627
+ },
628
+ {
629
+ "epoch": 0.11998651836872262,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 0.0007,
632
+ "loss": 2.2672780990600585,
633
+ "step": 1780
634
+ },
635
+ {
636
+ "epoch": 0.12133468149646107,
637
+ "grad_norm": 0.3671875,
638
+ "learning_rate": 0.0007,
639
+ "loss": 2.261806106567383,
640
+ "step": 1800
641
+ },
642
+ {
643
+ "epoch": 0.12268284462419952,
644
+ "grad_norm": 0.375,
645
+ "learning_rate": 0.0007,
646
+ "loss": 2.2513105392456056,
647
+ "step": 1820
648
+ },
649
+ {
650
+ "epoch": 0.12403100775193798,
651
+ "grad_norm": 0.369140625,
652
+ "learning_rate": 0.0007,
653
+ "loss": 2.2201774597167967,
654
+ "step": 1840
655
+ },
656
+ {
657
+ "epoch": 0.12537917087967643,
658
+ "grad_norm": 0.35546875,
659
+ "learning_rate": 0.0007,
660
+ "loss": 2.2213924407958983,
661
+ "step": 1860
662
+ },
663
+ {
664
+ "epoch": 0.1267273340074149,
665
+ "grad_norm": 0.361328125,
666
+ "learning_rate": 0.0007,
667
+ "loss": 2.21158447265625,
668
+ "step": 1880
669
+ },
670
+ {
671
+ "epoch": 0.12807549713515334,
672
+ "grad_norm": 0.357421875,
673
+ "learning_rate": 0.0007,
674
+ "loss": 2.218313217163086,
675
+ "step": 1900
676
+ },
677
+ {
678
+ "epoch": 0.12942366026289182,
679
+ "grad_norm": 0.361328125,
680
+ "learning_rate": 0.0007,
681
+ "loss": 2.1961637496948243,
682
+ "step": 1920
683
+ },
684
+ {
685
+ "epoch": 0.13077182339063026,
686
+ "grad_norm": 0.369140625,
687
+ "learning_rate": 0.0007,
688
+ "loss": 2.2057926177978517,
689
+ "step": 1940
690
+ },
691
+ {
692
+ "epoch": 0.13211998651836873,
693
+ "grad_norm": 0.345703125,
694
+ "learning_rate": 0.0007,
695
+ "loss": 2.181648826599121,
696
+ "step": 1960
697
+ },
698
+ {
699
+ "epoch": 0.13346814964610718,
700
+ "grad_norm": 0.380859375,
701
+ "learning_rate": 0.0007,
702
+ "loss": 2.203908920288086,
703
+ "step": 1980
704
+ },
705
+ {
706
+ "epoch": 0.13481631277384565,
707
+ "grad_norm": 0.341796875,
708
+ "learning_rate": 0.0007,
709
+ "loss": 2.1758956909179688,
710
+ "step": 2000
711
+ },
712
+ {
713
+ "epoch": 0.1361644759015841,
714
+ "grad_norm": 0.35546875,
715
+ "learning_rate": 0.0007,
716
+ "loss": 2.185748481750488,
717
+ "step": 2020
718
+ },
719
+ {
720
+ "epoch": 0.13751263902932254,
721
+ "grad_norm": 0.41796875,
722
+ "learning_rate": 0.0007,
723
+ "loss": 2.165380668640137,
724
+ "step": 2040
725
+ },
726
+ {
727
+ "epoch": 0.138860802157061,
728
+ "grad_norm": 0.3515625,
729
+ "learning_rate": 0.0007,
730
+ "loss": 2.1852161407470705,
731
+ "step": 2060
732
+ },
733
+ {
734
+ "epoch": 0.14020896528479945,
735
+ "grad_norm": 0.34765625,
736
+ "learning_rate": 0.0007,
737
+ "loss": 2.160782051086426,
738
+ "step": 2080
739
+ },
740
+ {
741
+ "epoch": 0.14155712841253792,
742
+ "grad_norm": 0.337890625,
743
+ "learning_rate": 0.0007,
744
+ "loss": 2.152023124694824,
745
+ "step": 2100
746
+ },
747
+ {
748
+ "epoch": 0.14290529154027637,
749
+ "grad_norm": 0.359375,
750
+ "learning_rate": 0.0007,
751
+ "loss": 2.160308837890625,
752
+ "step": 2120
753
+ },
754
+ {
755
+ "epoch": 0.14425345466801484,
756
+ "grad_norm": 0.34765625,
757
+ "learning_rate": 0.0007,
758
+ "loss": 2.1609643936157226,
759
+ "step": 2140
760
+ },
761
+ {
762
+ "epoch": 0.14560161779575329,
763
+ "grad_norm": 0.357421875,
764
+ "learning_rate": 0.0007,
765
+ "loss": 2.1475265502929686,
766
+ "step": 2160
767
+ },
768
+ {
769
+ "epoch": 0.14694978092349173,
770
+ "grad_norm": 0.345703125,
771
+ "learning_rate": 0.0007,
772
+ "loss": 2.155929183959961,
773
+ "step": 2180
774
+ },
775
+ {
776
+ "epoch": 0.1482979440512302,
777
+ "grad_norm": 0.341796875,
778
+ "learning_rate": 0.0007,
779
+ "loss": 2.1293373107910156,
780
+ "step": 2200
781
+ },
782
+ {
783
+ "epoch": 0.14964610717896865,
784
+ "grad_norm": 0.35546875,
785
+ "learning_rate": 0.0007,
786
+ "loss": 2.137188720703125,
787
+ "step": 2220
788
+ },
789
+ {
790
+ "epoch": 0.15099427030670712,
791
+ "grad_norm": 0.345703125,
792
+ "learning_rate": 0.0007,
793
+ "loss": 2.1462854385375976,
794
+ "step": 2240
795
+ },
796
+ {
797
+ "epoch": 0.15234243343444556,
798
+ "grad_norm": 0.365234375,
799
+ "learning_rate": 0.0007,
800
+ "loss": 2.1255794525146485,
801
+ "step": 2260
802
+ },
803
+ {
804
+ "epoch": 0.15369059656218403,
805
+ "grad_norm": 0.33984375,
806
+ "learning_rate": 0.0007,
807
+ "loss": 2.1365018844604493,
808
+ "step": 2280
809
+ },
810
+ {
811
+ "epoch": 0.15503875968992248,
812
+ "grad_norm": 0.337890625,
813
+ "learning_rate": 0.0007,
814
+ "loss": 2.1164464950561523,
815
+ "step": 2300
816
+ },
817
+ {
818
+ "epoch": 0.15638692281766095,
819
+ "grad_norm": 0.361328125,
820
+ "learning_rate": 0.0007,
821
+ "loss": 2.127320671081543,
822
+ "step": 2320
823
+ },
824
+ {
825
+ "epoch": 0.1577350859453994,
826
+ "grad_norm": 0.3359375,
827
+ "learning_rate": 0.0007,
828
+ "loss": 2.117641067504883,
829
+ "step": 2340
830
+ },
831
+ {
832
+ "epoch": 0.15908324907313784,
833
+ "grad_norm": 0.357421875,
834
+ "learning_rate": 0.0007,
835
+ "loss": 2.119377326965332,
836
+ "step": 2360
837
+ },
838
+ {
839
+ "epoch": 0.1604314122008763,
840
+ "grad_norm": 0.33203125,
841
+ "learning_rate": 0.0007,
842
+ "loss": 2.121909332275391,
843
+ "step": 2380
844
+ },
845
+ {
846
+ "epoch": 0.16177957532861476,
847
+ "grad_norm": 0.361328125,
848
+ "learning_rate": 0.0007,
849
+ "loss": 2.10235538482666,
850
+ "step": 2400
851
+ },
852
+ {
853
+ "epoch": 0.16312773845635323,
854
+ "grad_norm": 0.353515625,
855
+ "learning_rate": 0.0007,
856
+ "loss": 2.1056692123413088,
857
+ "step": 2420
858
+ },
859
+ {
860
+ "epoch": 0.16447590158409167,
861
+ "grad_norm": 0.3359375,
862
+ "learning_rate": 0.0007,
863
+ "loss": 2.093578910827637,
864
+ "step": 2440
865
+ },
866
+ {
867
+ "epoch": 0.16582406471183014,
868
+ "grad_norm": 0.337890625,
869
+ "learning_rate": 0.0007,
870
+ "loss": 2.1059795379638673,
871
+ "step": 2460
872
+ },
873
+ {
874
+ "epoch": 0.1671722278395686,
875
+ "grad_norm": 0.361328125,
876
+ "learning_rate": 0.0007,
877
+ "loss": 2.106010627746582,
878
+ "step": 2480
879
+ },
880
+ {
881
+ "epoch": 0.1683855746545332,
882
+ "eval_loss": 2.100916624069214,
883
+ "eval_runtime": 15.3711,
884
+ "eval_samples_per_second": 619.799,
885
+ "eval_steps_per_second": 0.651,
886
+ "step": 2498
887
+ },
888
+ {
889
+ "epoch": 0.16852039096730703,
890
+ "grad_norm": 0.33984375,
891
+ "learning_rate": 0.0007,
892
+ "loss": 2.087843322753906,
893
+ "step": 2500
894
+ },
895
+ {
896
+ "epoch": 0.16852039096730703,
897
+ "eval_loss": 2.1001250743865967,
898
+ "eval_runtime": 18.4156,
899
+ "eval_samples_per_second": 517.335,
900
+ "eval_steps_per_second": 0.543,
901
+ "step": 2500
902
+ },
903
+ {
904
+ "epoch": 0.1698685540950455,
905
+ "grad_norm": 20840448.0,
906
+ "learning_rate": 3.4200000000000005e-05,
907
+ "loss": 2.092716407775879,
908
+ "step": 2520
909
+ },
910
+ {
911
+ "epoch": 0.17121671722278395,
912
+ "grad_norm": 22937600.0,
913
+ "learning_rate": 7.020000000000001e-05,
914
+ "loss": 2.106246566772461,
915
+ "step": 2540
916
+ },
917
+ {
918
+ "epoch": 0.17256488035052242,
919
+ "grad_norm": 116391936.0,
920
+ "learning_rate": 9e-05,
921
+ "loss": 3.2435539245605467,
922
+ "step": 2560
923
+ },
924
+ {
925
+ "epoch": 0.17391304347826086,
926
+ "grad_norm": 283115520.0,
927
+ "learning_rate": 9e-05,
928
+ "loss": 7.010993957519531,
929
+ "step": 2580
930
+ },
931
+ {
932
+ "epoch": 0.17526120660599934,
933
+ "grad_norm": 583008256.0,
934
+ "learning_rate": 9e-05,
935
+ "loss": 7.576824188232422,
936
+ "step": 2600
937
+ },
938
+ {
939
+ "epoch": 0.17660936973373778,
940
+ "grad_norm": 998244352.0,
941
+ "learning_rate": 9e-05,
942
+ "loss": 7.7992095947265625,
943
+ "step": 2620
944
+ },
945
+ {
946
+ "epoch": 0.17795753286147623,
947
+ "grad_norm": 1501560832.0,
948
+ "learning_rate": 9e-05,
949
+ "loss": 7.901728820800781,
950
+ "step": 2640
951
+ },
952
+ {
953
+ "epoch": 0.1793056959892147,
954
+ "grad_norm": 2055208960.0,
955
+ "learning_rate": 9e-05,
956
+ "loss": 7.956874847412109,
957
+ "step": 2660
958
+ },
959
+ {
960
+ "epoch": 0.18065385911695314,
961
+ "grad_norm": 2650800128.0,
962
+ "learning_rate": 9e-05,
963
+ "loss": 7.995229339599609,
964
+ "step": 2680
965
+ },
966
+ {
967
+ "epoch": 0.1820020222446916,
968
+ "grad_norm": 3288334336.0,
969
+ "learning_rate": 9e-05,
970
+ "loss": 8.019290161132812,
971
+ "step": 2700
972
+ },
973
+ {
974
+ "epoch": 0.18335018537243006,
975
+ "grad_norm": 3942645760.0,
976
+ "learning_rate": 9e-05,
977
+ "loss": 8.043710327148437,
978
+ "step": 2720
979
+ },
980
+ {
981
+ "epoch": 0.18469834850016853,
982
+ "grad_norm": 4630511616.0,
983
+ "learning_rate": 9e-05,
984
+ "loss": 8.043434143066406,
985
+ "step": 2740
986
+ },
987
+ {
988
+ "epoch": 0.18604651162790697,
989
+ "grad_norm": 5301600256.0,
990
+ "learning_rate": 9e-05,
991
+ "loss": 8.049758911132812,
992
+ "step": 2760
993
+ },
994
+ {
995
+ "epoch": 0.18739467475564545,
996
+ "grad_norm": 5972688896.0,
997
+ "learning_rate": 9e-05,
998
+ "loss": 8.061854553222656,
999
+ "step": 2780
1000
+ },
1001
+ {
1002
+ "epoch": 0.1887428378833839,
1003
+ "grad_norm": 6643777536.0,
1004
+ "learning_rate": 9e-05,
1005
+ "loss": 8.073146057128906,
1006
+ "step": 2800
1007
+ },
1008
+ {
1009
+ "epoch": 0.19009100101112233,
1010
+ "grad_norm": 7281311744.0,
1011
+ "learning_rate": 9e-05,
1012
+ "loss": 8.076551818847657,
1013
+ "step": 2820
1014
+ },
1015
+ {
1016
+ "epoch": 0.1914391641388608,
1017
+ "grad_norm": 7918845952.0,
1018
+ "learning_rate": 9e-05,
1019
+ "loss": 8.078580474853515,
1020
+ "step": 2840
1021
+ },
1022
+ {
1023
+ "epoch": 0.19278732726659925,
1024
+ "grad_norm": 8556380160.0,
1025
+ "learning_rate": 9e-05,
1026
+ "loss": 8.073567199707032,
1027
+ "step": 2860
1028
+ },
1029
+ {
1030
+ "epoch": 0.19413549039433772,
1031
+ "grad_norm": 9126805504.0,
1032
+ "learning_rate": 9e-05,
1033
+ "loss": 8.08355941772461,
1034
+ "step": 2880
1035
+ },
1036
+ {
1037
+ "epoch": 0.19548365352207617,
1038
+ "grad_norm": 9730785280.0,
1039
+ "learning_rate": 9e-05,
1040
+ "loss": 8.085595703125,
1041
+ "step": 2900
1042
+ },
1043
+ {
1044
+ "epoch": 0.19683181664981464,
1045
+ "grad_norm": 10267656192.0,
1046
+ "learning_rate": 9e-05,
1047
+ "loss": 8.09107666015625,
1048
+ "step": 2920
1049
+ },
1050
+ {
1051
+ "epoch": 0.19817997977755308,
1052
+ "grad_norm": 10804527104.0,
1053
+ "learning_rate": 9e-05,
1054
+ "loss": 8.093254089355469,
1055
+ "step": 2940
1056
+ },
1057
+ {
1058
+ "epoch": 0.19952814290529153,
1059
+ "grad_norm": 11341398016.0,
1060
+ "learning_rate": 9e-05,
1061
+ "loss": 8.094740295410157,
1062
+ "step": 2960
1063
+ },
1064
+ {
1065
+ "epoch": 0.20087630603303,
1066
+ "grad_norm": 11811160064.0,
1067
+ "learning_rate": 9e-05,
1068
+ "loss": 8.100899505615235,
1069
+ "step": 2980
1070
+ },
1071
+ {
1072
+ "epoch": 0.20222446916076844,
1073
+ "grad_norm": 12213813248.0,
1074
+ "learning_rate": 9e-05,
1075
+ "loss": 8.100372314453125,
1076
+ "step": 3000
1077
+ }
1078
+ ],
1079
+ "logging_steps": 20,
1080
+ "max_steps": 3500,
1081
+ "num_input_tokens_seen": 0,
1082
+ "num_train_epochs": 1,
1083
+ "save_steps": 100,
1084
+ "stateful_callbacks": {
1085
+ "TrainerControl": {
1086
+ "args": {
1087
+ "should_epoch_stop": false,
1088
+ "should_evaluate": false,
1089
+ "should_log": false,
1090
+ "should_save": true,
1091
+ "should_training_stop": false
1092
+ },
1093
+ "attributes": {}
1094
+ }
1095
+ },
1096
+ "total_flos": 9678851407872000.0,
1097
+ "train_batch_size": 64,
1098
+ "trial_name": null,
1099
+ "trial_params": null
1100
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759f687547d62f080b691b398df9dd337d9e21ff107770065591d2088b29bf88
3
+ size 4920
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
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": 4
36
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc5bbd91ac869535b1b64045e5d36b2cdf13cf8c802dc719e7f456ce2a89c062
3
+ size 33967272
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4973864ab280b7e8794d952ad3f82533c29fccb1d9db6020f6a86b5d69e66d3
3
+ size 39573050
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cf9097d4513154245c48236b6ec5137b7ee2a21c9f58f2cba798ea275c6026f
3
+ size 14244
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9810d1ccc64353d007b5d64784152aae7a20a39b3430ca0793162cf90f4756c6
3
+ size 1064
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/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-gelu-100L_trash_run/checkpoint-3100/trainer_state.json ADDED
@@ -0,0 +1,1135 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.20896528479946072,
6
+ "eval_steps": 2498,
7
+ "global_step": 3100,
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.299579620361328,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.359375,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047666931152344,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.2734375,
29
+ "learning_rate": 8.259999999999999e-05,
30
+ "loss": 7.838191986083984,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 1.265625,
36
+ "learning_rate": 0.0001106,
37
+ "loss": 7.57310791015625,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.1953125,
43
+ "learning_rate": 0.0001386,
44
+ "loss": 7.233531951904297,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 1.1328125,
50
+ "learning_rate": 0.00016659999999999998,
51
+ "loss": 6.860941314697266,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 0.95703125,
57
+ "learning_rate": 0.00019460000000000001,
58
+ "loss": 6.51712646484375,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 0.81640625,
64
+ "learning_rate": 0.0002226,
65
+ "loss": 6.234881591796875,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 0.73828125,
71
+ "learning_rate": 0.00025059999999999997,
72
+ "loss": 6.024795913696289,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 1.0390625,
78
+ "learning_rate": 0.0002786,
79
+ "loss": 5.837321472167969,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.01482979440512302,
84
+ "grad_norm": 0.63671875,
85
+ "learning_rate": 0.00030659999999999997,
86
+ "loss": 5.647708129882813,
87
+ "step": 220
88
+ },
89
+ {
90
+ "epoch": 0.016177957532861477,
91
+ "grad_norm": 0.8984375,
92
+ "learning_rate": 0.0003346,
93
+ "loss": 5.43688735961914,
94
+ "step": 240
95
+ },
96
+ {
97
+ "epoch": 0.01752612066059993,
98
+ "grad_norm": 0.77734375,
99
+ "learning_rate": 0.00036260000000000003,
100
+ "loss": 5.210599517822265,
101
+ "step": 260
102
+ },
103
+ {
104
+ "epoch": 0.01887428378833839,
105
+ "grad_norm": 0.890625,
106
+ "learning_rate": 0.0003906,
107
+ "loss": 4.956959533691406,
108
+ "step": 280
109
+ },
110
+ {
111
+ "epoch": 0.020222446916076844,
112
+ "grad_norm": 0.84765625,
113
+ "learning_rate": 0.0004186,
114
+ "loss": 4.726874160766601,
115
+ "step": 300
116
+ },
117
+ {
118
+ "epoch": 0.021570610043815303,
119
+ "grad_norm": 0.87890625,
120
+ "learning_rate": 0.0004466,
121
+ "loss": 4.527945327758789,
122
+ "step": 320
123
+ },
124
+ {
125
+ "epoch": 0.022918773171553757,
126
+ "grad_norm": 1.3515625,
127
+ "learning_rate": 0.00047460000000000004,
128
+ "loss": 4.3517810821533205,
129
+ "step": 340
130
+ },
131
+ {
132
+ "epoch": 0.024266936299292215,
133
+ "grad_norm": 0.76171875,
134
+ "learning_rate": 0.0005026,
135
+ "loss": 4.212622451782226,
136
+ "step": 360
137
+ },
138
+ {
139
+ "epoch": 0.02561509942703067,
140
+ "grad_norm": 0.6640625,
141
+ "learning_rate": 0.0005306,
142
+ "loss": 4.076633453369141,
143
+ "step": 380
144
+ },
145
+ {
146
+ "epoch": 0.026963262554769128,
147
+ "grad_norm": 0.498046875,
148
+ "learning_rate": 0.0005586,
149
+ "loss": 3.9346649169921877,
150
+ "step": 400
151
+ },
152
+ {
153
+ "epoch": 0.028311425682507583,
154
+ "grad_norm": 0.85546875,
155
+ "learning_rate": 0.0005866,
156
+ "loss": 3.8310710906982424,
157
+ "step": 420
158
+ },
159
+ {
160
+ "epoch": 0.02965958881024604,
161
+ "grad_norm": 0.59375,
162
+ "learning_rate": 0.0006146,
163
+ "loss": 3.7676155090332033,
164
+ "step": 440
165
+ },
166
+ {
167
+ "epoch": 0.031007751937984496,
168
+ "grad_norm": 0.6171875,
169
+ "learning_rate": 0.0006426,
170
+ "loss": 3.6599285125732424,
171
+ "step": 460
172
+ },
173
+ {
174
+ "epoch": 0.032355915065722954,
175
+ "grad_norm": 0.5546875,
176
+ "learning_rate": 0.0006705999999999999,
177
+ "loss": 3.5886669158935547,
178
+ "step": 480
179
+ },
180
+ {
181
+ "epoch": 0.03370407819346141,
182
+ "grad_norm": 0.546875,
183
+ "learning_rate": 0.0006986,
184
+ "loss": 3.512950134277344,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 0.03505224132119986,
189
+ "grad_norm": 0.53125,
190
+ "learning_rate": 0.0007,
191
+ "loss": 3.444244384765625,
192
+ "step": 520
193
+ },
194
+ {
195
+ "epoch": 0.03640040444893832,
196
+ "grad_norm": 0.5078125,
197
+ "learning_rate": 0.0007,
198
+ "loss": 3.369425964355469,
199
+ "step": 540
200
+ },
201
+ {
202
+ "epoch": 0.03774856757667678,
203
+ "grad_norm": 0.5625,
204
+ "learning_rate": 0.0007,
205
+ "loss": 3.318771743774414,
206
+ "step": 560
207
+ },
208
+ {
209
+ "epoch": 0.03909673070441524,
210
+ "grad_norm": 0.470703125,
211
+ "learning_rate": 0.0007,
212
+ "loss": 3.2595756530761717,
213
+ "step": 580
214
+ },
215
+ {
216
+ "epoch": 0.04044489383215369,
217
+ "grad_norm": 0.53125,
218
+ "learning_rate": 0.0007,
219
+ "loss": 3.2241649627685547,
220
+ "step": 600
221
+ },
222
+ {
223
+ "epoch": 0.04179305695989215,
224
+ "grad_norm": 0.484375,
225
+ "learning_rate": 0.0007,
226
+ "loss": 3.1619989395141603,
227
+ "step": 620
228
+ },
229
+ {
230
+ "epoch": 0.043141220087630605,
231
+ "grad_norm": 0.5625,
232
+ "learning_rate": 0.0007,
233
+ "loss": 3.14036865234375,
234
+ "step": 640
235
+ },
236
+ {
237
+ "epoch": 0.044489383215369056,
238
+ "grad_norm": 0.41796875,
239
+ "learning_rate": 0.0007,
240
+ "loss": 3.083884620666504,
241
+ "step": 660
242
+ },
243
+ {
244
+ "epoch": 0.045837546343107514,
245
+ "grad_norm": 0.451171875,
246
+ "learning_rate": 0.0007,
247
+ "loss": 3.0591588973999024,
248
+ "step": 680
249
+ },
250
+ {
251
+ "epoch": 0.04718570947084597,
252
+ "grad_norm": 0.423828125,
253
+ "learning_rate": 0.0007,
254
+ "loss": 3.015497016906738,
255
+ "step": 700
256
+ },
257
+ {
258
+ "epoch": 0.04853387259858443,
259
+ "grad_norm": 0.41796875,
260
+ "learning_rate": 0.0007,
261
+ "loss": 2.9805700302124025,
262
+ "step": 720
263
+ },
264
+ {
265
+ "epoch": 0.04988203572632288,
266
+ "grad_norm": 0.46875,
267
+ "learning_rate": 0.0007,
268
+ "loss": 2.9426918029785156,
269
+ "step": 740
270
+ },
271
+ {
272
+ "epoch": 0.05123019885406134,
273
+ "grad_norm": 0.46484375,
274
+ "learning_rate": 0.0007,
275
+ "loss": 2.9274158477783203,
276
+ "step": 760
277
+ },
278
+ {
279
+ "epoch": 0.0525783619817998,
280
+ "grad_norm": 0.421875,
281
+ "learning_rate": 0.0007,
282
+ "loss": 2.931459426879883,
283
+ "step": 780
284
+ },
285
+ {
286
+ "epoch": 0.053926525109538256,
287
+ "grad_norm": 0.486328125,
288
+ "learning_rate": 0.0007,
289
+ "loss": 2.877642059326172,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.05527468823727671,
294
+ "grad_norm": 0.40234375,
295
+ "learning_rate": 0.0007,
296
+ "loss": 2.8646697998046875,
297
+ "step": 820
298
+ },
299
+ {
300
+ "epoch": 0.056622851365015166,
301
+ "grad_norm": 0.451171875,
302
+ "learning_rate": 0.0007,
303
+ "loss": 2.8328128814697267,
304
+ "step": 840
305
+ },
306
+ {
307
+ "epoch": 0.057971014492753624,
308
+ "grad_norm": 0.396484375,
309
+ "learning_rate": 0.0007,
310
+ "loss": 2.790274429321289,
311
+ "step": 860
312
+ },
313
+ {
314
+ "epoch": 0.05931917762049208,
315
+ "grad_norm": 0.423828125,
316
+ "learning_rate": 0.0007,
317
+ "loss": 2.7668125152587892,
318
+ "step": 880
319
+ },
320
+ {
321
+ "epoch": 0.06066734074823053,
322
+ "grad_norm": 0.427734375,
323
+ "learning_rate": 0.0007,
324
+ "loss": 2.7516881942749025,
325
+ "step": 900
326
+ },
327
+ {
328
+ "epoch": 0.06201550387596899,
329
+ "grad_norm": 0.42578125,
330
+ "learning_rate": 0.0007,
331
+ "loss": 2.722590446472168,
332
+ "step": 920
333
+ },
334
+ {
335
+ "epoch": 0.06336366700370745,
336
+ "grad_norm": 0.4375,
337
+ "learning_rate": 0.0007,
338
+ "loss": 2.708496856689453,
339
+ "step": 940
340
+ },
341
+ {
342
+ "epoch": 0.06471183013144591,
343
+ "grad_norm": 0.400390625,
344
+ "learning_rate": 0.0007,
345
+ "loss": 2.681654930114746,
346
+ "step": 960
347
+ },
348
+ {
349
+ "epoch": 0.06605999325918437,
350
+ "grad_norm": 0.396484375,
351
+ "learning_rate": 0.0007,
352
+ "loss": 2.6650571823120117,
353
+ "step": 980
354
+ },
355
+ {
356
+ "epoch": 0.06740815638692282,
357
+ "grad_norm": 0.41015625,
358
+ "learning_rate": 0.0007,
359
+ "loss": 2.6637720108032226,
360
+ "step": 1000
361
+ },
362
+ {
363
+ "epoch": 0.06875631951466127,
364
+ "grad_norm": 0.396484375,
365
+ "learning_rate": 0.0007,
366
+ "loss": 2.6311159133911133,
367
+ "step": 1020
368
+ },
369
+ {
370
+ "epoch": 0.07010448264239973,
371
+ "grad_norm": 0.408203125,
372
+ "learning_rate": 0.0007,
373
+ "loss": 2.615707015991211,
374
+ "step": 1040
375
+ },
376
+ {
377
+ "epoch": 0.07145264577013818,
378
+ "grad_norm": 0.404296875,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.6003103256225586,
381
+ "step": 1060
382
+ },
383
+ {
384
+ "epoch": 0.07280080889787664,
385
+ "grad_norm": 0.41015625,
386
+ "learning_rate": 0.0007,
387
+ "loss": 2.581348991394043,
388
+ "step": 1080
389
+ },
390
+ {
391
+ "epoch": 0.0741489720256151,
392
+ "grad_norm": 0.404296875,
393
+ "learning_rate": 0.0007,
394
+ "loss": 2.574792671203613,
395
+ "step": 1100
396
+ },
397
+ {
398
+ "epoch": 0.07549713515335356,
399
+ "grad_norm": 0.384765625,
400
+ "learning_rate": 0.0007,
401
+ "loss": 2.538400650024414,
402
+ "step": 1120
403
+ },
404
+ {
405
+ "epoch": 0.07684529828109202,
406
+ "grad_norm": 0.41796875,
407
+ "learning_rate": 0.0007,
408
+ "loss": 2.5249162673950196,
409
+ "step": 1140
410
+ },
411
+ {
412
+ "epoch": 0.07819346140883048,
413
+ "grad_norm": 0.404296875,
414
+ "learning_rate": 0.0007,
415
+ "loss": 2.5237232208251954,
416
+ "step": 1160
417
+ },
418
+ {
419
+ "epoch": 0.07954162453656892,
420
+ "grad_norm": 0.396484375,
421
+ "learning_rate": 0.0007,
422
+ "loss": 2.517318344116211,
423
+ "step": 1180
424
+ },
425
+ {
426
+ "epoch": 0.08088978766430738,
427
+ "grad_norm": 0.41796875,
428
+ "learning_rate": 0.0007,
429
+ "loss": 2.507699203491211,
430
+ "step": 1200
431
+ },
432
+ {
433
+ "epoch": 0.08223795079204584,
434
+ "grad_norm": 0.3828125,
435
+ "learning_rate": 0.0007,
436
+ "loss": 2.4932661056518555,
437
+ "step": 1220
438
+ },
439
+ {
440
+ "epoch": 0.0835861139197843,
441
+ "grad_norm": 0.365234375,
442
+ "learning_rate": 0.0007,
443
+ "loss": 2.4781061172485352,
444
+ "step": 1240
445
+ },
446
+ {
447
+ "epoch": 0.08493427704752275,
448
+ "grad_norm": 0.388671875,
449
+ "learning_rate": 0.0007,
450
+ "loss": 2.465612030029297,
451
+ "step": 1260
452
+ },
453
+ {
454
+ "epoch": 0.08628244017526121,
455
+ "grad_norm": 0.41796875,
456
+ "learning_rate": 0.0007,
457
+ "loss": 2.4463497161865235,
458
+ "step": 1280
459
+ },
460
+ {
461
+ "epoch": 0.08763060330299967,
462
+ "grad_norm": 0.390625,
463
+ "learning_rate": 0.0007,
464
+ "loss": 2.4374414443969727,
465
+ "step": 1300
466
+ },
467
+ {
468
+ "epoch": 0.08897876643073811,
469
+ "grad_norm": 0.35546875,
470
+ "learning_rate": 0.0007,
471
+ "loss": 2.4308101654052736,
472
+ "step": 1320
473
+ },
474
+ {
475
+ "epoch": 0.09032692955847657,
476
+ "grad_norm": 0.40234375,
477
+ "learning_rate": 0.0007,
478
+ "loss": 2.416130256652832,
479
+ "step": 1340
480
+ },
481
+ {
482
+ "epoch": 0.09167509268621503,
483
+ "grad_norm": 0.4296875,
484
+ "learning_rate": 0.0007,
485
+ "loss": 2.41524600982666,
486
+ "step": 1360
487
+ },
488
+ {
489
+ "epoch": 0.09302325581395349,
490
+ "grad_norm": 0.359375,
491
+ "learning_rate": 0.0007,
492
+ "loss": 2.392221450805664,
493
+ "step": 1380
494
+ },
495
+ {
496
+ "epoch": 0.09437141894169195,
497
+ "grad_norm": 0.373046875,
498
+ "learning_rate": 0.0007,
499
+ "loss": 2.383866310119629,
500
+ "step": 1400
501
+ },
502
+ {
503
+ "epoch": 0.0957195820694304,
504
+ "grad_norm": 0.396484375,
505
+ "learning_rate": 0.0007,
506
+ "loss": 2.394974708557129,
507
+ "step": 1420
508
+ },
509
+ {
510
+ "epoch": 0.09706774519716886,
511
+ "grad_norm": 0.396484375,
512
+ "learning_rate": 0.0007,
513
+ "loss": 2.3713239669799804,
514
+ "step": 1440
515
+ },
516
+ {
517
+ "epoch": 0.09841590832490732,
518
+ "grad_norm": 0.37109375,
519
+ "learning_rate": 0.0007,
520
+ "loss": 2.3597253799438476,
521
+ "step": 1460
522
+ },
523
+ {
524
+ "epoch": 0.09976407145264576,
525
+ "grad_norm": 0.34765625,
526
+ "learning_rate": 0.0007,
527
+ "loss": 2.3452421188354493,
528
+ "step": 1480
529
+ },
530
+ {
531
+ "epoch": 0.10111223458038422,
532
+ "grad_norm": 0.376953125,
533
+ "learning_rate": 0.0007,
534
+ "loss": 2.3336454391479493,
535
+ "step": 1500
536
+ },
537
+ {
538
+ "epoch": 0.10246039770812268,
539
+ "grad_norm": 0.39453125,
540
+ "learning_rate": 0.0007,
541
+ "loss": 2.3358110427856444,
542
+ "step": 1520
543
+ },
544
+ {
545
+ "epoch": 0.10380856083586114,
546
+ "grad_norm": 0.40625,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.324386787414551,
549
+ "step": 1540
550
+ },
551
+ {
552
+ "epoch": 0.1051567239635996,
553
+ "grad_norm": 0.376953125,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.313844108581543,
556
+ "step": 1560
557
+ },
558
+ {
559
+ "epoch": 0.10650488709133805,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.3294256210327147,
563
+ "step": 1580
564
+ },
565
+ {
566
+ "epoch": 0.10785305021907651,
567
+ "grad_norm": 0.40234375,
568
+ "learning_rate": 0.0007,
569
+ "loss": 2.297942352294922,
570
+ "step": 1600
571
+ },
572
+ {
573
+ "epoch": 0.10920121334681497,
574
+ "grad_norm": 0.357421875,
575
+ "learning_rate": 0.0007,
576
+ "loss": 2.296767997741699,
577
+ "step": 1620
578
+ },
579
+ {
580
+ "epoch": 0.11054937647455342,
581
+ "grad_norm": 0.37890625,
582
+ "learning_rate": 0.0007,
583
+ "loss": 2.3081844329833983,
584
+ "step": 1640
585
+ },
586
+ {
587
+ "epoch": 0.11189753960229187,
588
+ "grad_norm": 0.35546875,
589
+ "learning_rate": 0.0007,
590
+ "loss": 2.2883626937866213,
591
+ "step": 1660
592
+ },
593
+ {
594
+ "epoch": 0.11324570273003033,
595
+ "grad_norm": 0.375,
596
+ "learning_rate": 0.0007,
597
+ "loss": 2.272566223144531,
598
+ "step": 1680
599
+ },
600
+ {
601
+ "epoch": 0.11459386585776879,
602
+ "grad_norm": 0.373046875,
603
+ "learning_rate": 0.0007,
604
+ "loss": 2.281250762939453,
605
+ "step": 1700
606
+ },
607
+ {
608
+ "epoch": 0.11594202898550725,
609
+ "grad_norm": 0.41796875,
610
+ "learning_rate": 0.0007,
611
+ "loss": 2.257417106628418,
612
+ "step": 1720
613
+ },
614
+ {
615
+ "epoch": 0.1172901921132457,
616
+ "grad_norm": 0.365234375,
617
+ "learning_rate": 0.0007,
618
+ "loss": 2.2473636627197267,
619
+ "step": 1740
620
+ },
621
+ {
622
+ "epoch": 0.11863835524098416,
623
+ "grad_norm": 0.380859375,
624
+ "learning_rate": 0.0007,
625
+ "loss": 2.253912925720215,
626
+ "step": 1760
627
+ },
628
+ {
629
+ "epoch": 0.11998651836872262,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 0.0007,
632
+ "loss": 2.2672780990600585,
633
+ "step": 1780
634
+ },
635
+ {
636
+ "epoch": 0.12133468149646107,
637
+ "grad_norm": 0.3671875,
638
+ "learning_rate": 0.0007,
639
+ "loss": 2.261806106567383,
640
+ "step": 1800
641
+ },
642
+ {
643
+ "epoch": 0.12268284462419952,
644
+ "grad_norm": 0.375,
645
+ "learning_rate": 0.0007,
646
+ "loss": 2.2513105392456056,
647
+ "step": 1820
648
+ },
649
+ {
650
+ "epoch": 0.12403100775193798,
651
+ "grad_norm": 0.369140625,
652
+ "learning_rate": 0.0007,
653
+ "loss": 2.2201774597167967,
654
+ "step": 1840
655
+ },
656
+ {
657
+ "epoch": 0.12537917087967643,
658
+ "grad_norm": 0.35546875,
659
+ "learning_rate": 0.0007,
660
+ "loss": 2.2213924407958983,
661
+ "step": 1860
662
+ },
663
+ {
664
+ "epoch": 0.1267273340074149,
665
+ "grad_norm": 0.361328125,
666
+ "learning_rate": 0.0007,
667
+ "loss": 2.21158447265625,
668
+ "step": 1880
669
+ },
670
+ {
671
+ "epoch": 0.12807549713515334,
672
+ "grad_norm": 0.357421875,
673
+ "learning_rate": 0.0007,
674
+ "loss": 2.218313217163086,
675
+ "step": 1900
676
+ },
677
+ {
678
+ "epoch": 0.12942366026289182,
679
+ "grad_norm": 0.361328125,
680
+ "learning_rate": 0.0007,
681
+ "loss": 2.1961637496948243,
682
+ "step": 1920
683
+ },
684
+ {
685
+ "epoch": 0.13077182339063026,
686
+ "grad_norm": 0.369140625,
687
+ "learning_rate": 0.0007,
688
+ "loss": 2.2057926177978517,
689
+ "step": 1940
690
+ },
691
+ {
692
+ "epoch": 0.13211998651836873,
693
+ "grad_norm": 0.345703125,
694
+ "learning_rate": 0.0007,
695
+ "loss": 2.181648826599121,
696
+ "step": 1960
697
+ },
698
+ {
699
+ "epoch": 0.13346814964610718,
700
+ "grad_norm": 0.380859375,
701
+ "learning_rate": 0.0007,
702
+ "loss": 2.203908920288086,
703
+ "step": 1980
704
+ },
705
+ {
706
+ "epoch": 0.13481631277384565,
707
+ "grad_norm": 0.341796875,
708
+ "learning_rate": 0.0007,
709
+ "loss": 2.1758956909179688,
710
+ "step": 2000
711
+ },
712
+ {
713
+ "epoch": 0.1361644759015841,
714
+ "grad_norm": 0.35546875,
715
+ "learning_rate": 0.0007,
716
+ "loss": 2.185748481750488,
717
+ "step": 2020
718
+ },
719
+ {
720
+ "epoch": 0.13751263902932254,
721
+ "grad_norm": 0.41796875,
722
+ "learning_rate": 0.0007,
723
+ "loss": 2.165380668640137,
724
+ "step": 2040
725
+ },
726
+ {
727
+ "epoch": 0.138860802157061,
728
+ "grad_norm": 0.3515625,
729
+ "learning_rate": 0.0007,
730
+ "loss": 2.1852161407470705,
731
+ "step": 2060
732
+ },
733
+ {
734
+ "epoch": 0.14020896528479945,
735
+ "grad_norm": 0.34765625,
736
+ "learning_rate": 0.0007,
737
+ "loss": 2.160782051086426,
738
+ "step": 2080
739
+ },
740
+ {
741
+ "epoch": 0.14155712841253792,
742
+ "grad_norm": 0.337890625,
743
+ "learning_rate": 0.0007,
744
+ "loss": 2.152023124694824,
745
+ "step": 2100
746
+ },
747
+ {
748
+ "epoch": 0.14290529154027637,
749
+ "grad_norm": 0.359375,
750
+ "learning_rate": 0.0007,
751
+ "loss": 2.160308837890625,
752
+ "step": 2120
753
+ },
754
+ {
755
+ "epoch": 0.14425345466801484,
756
+ "grad_norm": 0.34765625,
757
+ "learning_rate": 0.0007,
758
+ "loss": 2.1609643936157226,
759
+ "step": 2140
760
+ },
761
+ {
762
+ "epoch": 0.14560161779575329,
763
+ "grad_norm": 0.357421875,
764
+ "learning_rate": 0.0007,
765
+ "loss": 2.1475265502929686,
766
+ "step": 2160
767
+ },
768
+ {
769
+ "epoch": 0.14694978092349173,
770
+ "grad_norm": 0.345703125,
771
+ "learning_rate": 0.0007,
772
+ "loss": 2.155929183959961,
773
+ "step": 2180
774
+ },
775
+ {
776
+ "epoch": 0.1482979440512302,
777
+ "grad_norm": 0.341796875,
778
+ "learning_rate": 0.0007,
779
+ "loss": 2.1293373107910156,
780
+ "step": 2200
781
+ },
782
+ {
783
+ "epoch": 0.14964610717896865,
784
+ "grad_norm": 0.35546875,
785
+ "learning_rate": 0.0007,
786
+ "loss": 2.137188720703125,
787
+ "step": 2220
788
+ },
789
+ {
790
+ "epoch": 0.15099427030670712,
791
+ "grad_norm": 0.345703125,
792
+ "learning_rate": 0.0007,
793
+ "loss": 2.1462854385375976,
794
+ "step": 2240
795
+ },
796
+ {
797
+ "epoch": 0.15234243343444556,
798
+ "grad_norm": 0.365234375,
799
+ "learning_rate": 0.0007,
800
+ "loss": 2.1255794525146485,
801
+ "step": 2260
802
+ },
803
+ {
804
+ "epoch": 0.15369059656218403,
805
+ "grad_norm": 0.33984375,
806
+ "learning_rate": 0.0007,
807
+ "loss": 2.1365018844604493,
808
+ "step": 2280
809
+ },
810
+ {
811
+ "epoch": 0.15503875968992248,
812
+ "grad_norm": 0.337890625,
813
+ "learning_rate": 0.0007,
814
+ "loss": 2.1164464950561523,
815
+ "step": 2300
816
+ },
817
+ {
818
+ "epoch": 0.15638692281766095,
819
+ "grad_norm": 0.361328125,
820
+ "learning_rate": 0.0007,
821
+ "loss": 2.127320671081543,
822
+ "step": 2320
823
+ },
824
+ {
825
+ "epoch": 0.1577350859453994,
826
+ "grad_norm": 0.3359375,
827
+ "learning_rate": 0.0007,
828
+ "loss": 2.117641067504883,
829
+ "step": 2340
830
+ },
831
+ {
832
+ "epoch": 0.15908324907313784,
833
+ "grad_norm": 0.357421875,
834
+ "learning_rate": 0.0007,
835
+ "loss": 2.119377326965332,
836
+ "step": 2360
837
+ },
838
+ {
839
+ "epoch": 0.1604314122008763,
840
+ "grad_norm": 0.33203125,
841
+ "learning_rate": 0.0007,
842
+ "loss": 2.121909332275391,
843
+ "step": 2380
844
+ },
845
+ {
846
+ "epoch": 0.16177957532861476,
847
+ "grad_norm": 0.361328125,
848
+ "learning_rate": 0.0007,
849
+ "loss": 2.10235538482666,
850
+ "step": 2400
851
+ },
852
+ {
853
+ "epoch": 0.16312773845635323,
854
+ "grad_norm": 0.353515625,
855
+ "learning_rate": 0.0007,
856
+ "loss": 2.1056692123413088,
857
+ "step": 2420
858
+ },
859
+ {
860
+ "epoch": 0.16447590158409167,
861
+ "grad_norm": 0.3359375,
862
+ "learning_rate": 0.0007,
863
+ "loss": 2.093578910827637,
864
+ "step": 2440
865
+ },
866
+ {
867
+ "epoch": 0.16582406471183014,
868
+ "grad_norm": 0.337890625,
869
+ "learning_rate": 0.0007,
870
+ "loss": 2.1059795379638673,
871
+ "step": 2460
872
+ },
873
+ {
874
+ "epoch": 0.1671722278395686,
875
+ "grad_norm": 0.361328125,
876
+ "learning_rate": 0.0007,
877
+ "loss": 2.106010627746582,
878
+ "step": 2480
879
+ },
880
+ {
881
+ "epoch": 0.1683855746545332,
882
+ "eval_loss": 2.100916624069214,
883
+ "eval_runtime": 15.3711,
884
+ "eval_samples_per_second": 619.799,
885
+ "eval_steps_per_second": 0.651,
886
+ "step": 2498
887
+ },
888
+ {
889
+ "epoch": 0.16852039096730703,
890
+ "grad_norm": 0.33984375,
891
+ "learning_rate": 0.0007,
892
+ "loss": 2.087843322753906,
893
+ "step": 2500
894
+ },
895
+ {
896
+ "epoch": 0.16852039096730703,
897
+ "eval_loss": 2.1001250743865967,
898
+ "eval_runtime": 18.4156,
899
+ "eval_samples_per_second": 517.335,
900
+ "eval_steps_per_second": 0.543,
901
+ "step": 2500
902
+ },
903
+ {
904
+ "epoch": 0.1698685540950455,
905
+ "grad_norm": 20840448.0,
906
+ "learning_rate": 3.4200000000000005e-05,
907
+ "loss": 2.092716407775879,
908
+ "step": 2520
909
+ },
910
+ {
911
+ "epoch": 0.17121671722278395,
912
+ "grad_norm": 22937600.0,
913
+ "learning_rate": 7.020000000000001e-05,
914
+ "loss": 2.106246566772461,
915
+ "step": 2540
916
+ },
917
+ {
918
+ "epoch": 0.17256488035052242,
919
+ "grad_norm": 116391936.0,
920
+ "learning_rate": 9e-05,
921
+ "loss": 3.2435539245605467,
922
+ "step": 2560
923
+ },
924
+ {
925
+ "epoch": 0.17391304347826086,
926
+ "grad_norm": 283115520.0,
927
+ "learning_rate": 9e-05,
928
+ "loss": 7.010993957519531,
929
+ "step": 2580
930
+ },
931
+ {
932
+ "epoch": 0.17526120660599934,
933
+ "grad_norm": 583008256.0,
934
+ "learning_rate": 9e-05,
935
+ "loss": 7.576824188232422,
936
+ "step": 2600
937
+ },
938
+ {
939
+ "epoch": 0.17660936973373778,
940
+ "grad_norm": 998244352.0,
941
+ "learning_rate": 9e-05,
942
+ "loss": 7.7992095947265625,
943
+ "step": 2620
944
+ },
945
+ {
946
+ "epoch": 0.17795753286147623,
947
+ "grad_norm": 1501560832.0,
948
+ "learning_rate": 9e-05,
949
+ "loss": 7.901728820800781,
950
+ "step": 2640
951
+ },
952
+ {
953
+ "epoch": 0.1793056959892147,
954
+ "grad_norm": 2055208960.0,
955
+ "learning_rate": 9e-05,
956
+ "loss": 7.956874847412109,
957
+ "step": 2660
958
+ },
959
+ {
960
+ "epoch": 0.18065385911695314,
961
+ "grad_norm": 2650800128.0,
962
+ "learning_rate": 9e-05,
963
+ "loss": 7.995229339599609,
964
+ "step": 2680
965
+ },
966
+ {
967
+ "epoch": 0.1820020222446916,
968
+ "grad_norm": 3288334336.0,
969
+ "learning_rate": 9e-05,
970
+ "loss": 8.019290161132812,
971
+ "step": 2700
972
+ },
973
+ {
974
+ "epoch": 0.18335018537243006,
975
+ "grad_norm": 3942645760.0,
976
+ "learning_rate": 9e-05,
977
+ "loss": 8.043710327148437,
978
+ "step": 2720
979
+ },
980
+ {
981
+ "epoch": 0.18469834850016853,
982
+ "grad_norm": 4630511616.0,
983
+ "learning_rate": 9e-05,
984
+ "loss": 8.043434143066406,
985
+ "step": 2740
986
+ },
987
+ {
988
+ "epoch": 0.18604651162790697,
989
+ "grad_norm": 5301600256.0,
990
+ "learning_rate": 9e-05,
991
+ "loss": 8.049758911132812,
992
+ "step": 2760
993
+ },
994
+ {
995
+ "epoch": 0.18739467475564545,
996
+ "grad_norm": 5972688896.0,
997
+ "learning_rate": 9e-05,
998
+ "loss": 8.061854553222656,
999
+ "step": 2780
1000
+ },
1001
+ {
1002
+ "epoch": 0.1887428378833839,
1003
+ "grad_norm": 6643777536.0,
1004
+ "learning_rate": 9e-05,
1005
+ "loss": 8.073146057128906,
1006
+ "step": 2800
1007
+ },
1008
+ {
1009
+ "epoch": 0.19009100101112233,
1010
+ "grad_norm": 7281311744.0,
1011
+ "learning_rate": 9e-05,
1012
+ "loss": 8.076551818847657,
1013
+ "step": 2820
1014
+ },
1015
+ {
1016
+ "epoch": 0.1914391641388608,
1017
+ "grad_norm": 7918845952.0,
1018
+ "learning_rate": 9e-05,
1019
+ "loss": 8.078580474853515,
1020
+ "step": 2840
1021
+ },
1022
+ {
1023
+ "epoch": 0.19278732726659925,
1024
+ "grad_norm": 8556380160.0,
1025
+ "learning_rate": 9e-05,
1026
+ "loss": 8.073567199707032,
1027
+ "step": 2860
1028
+ },
1029
+ {
1030
+ "epoch": 0.19413549039433772,
1031
+ "grad_norm": 9126805504.0,
1032
+ "learning_rate": 9e-05,
1033
+ "loss": 8.08355941772461,
1034
+ "step": 2880
1035
+ },
1036
+ {
1037
+ "epoch": 0.19548365352207617,
1038
+ "grad_norm": 9730785280.0,
1039
+ "learning_rate": 9e-05,
1040
+ "loss": 8.085595703125,
1041
+ "step": 2900
1042
+ },
1043
+ {
1044
+ "epoch": 0.19683181664981464,
1045
+ "grad_norm": 10267656192.0,
1046
+ "learning_rate": 9e-05,
1047
+ "loss": 8.09107666015625,
1048
+ "step": 2920
1049
+ },
1050
+ {
1051
+ "epoch": 0.19817997977755308,
1052
+ "grad_norm": 10804527104.0,
1053
+ "learning_rate": 9e-05,
1054
+ "loss": 8.093254089355469,
1055
+ "step": 2940
1056
+ },
1057
+ {
1058
+ "epoch": 0.19952814290529153,
1059
+ "grad_norm": 11341398016.0,
1060
+ "learning_rate": 9e-05,
1061
+ "loss": 8.094740295410157,
1062
+ "step": 2960
1063
+ },
1064
+ {
1065
+ "epoch": 0.20087630603303,
1066
+ "grad_norm": 11811160064.0,
1067
+ "learning_rate": 9e-05,
1068
+ "loss": 8.100899505615235,
1069
+ "step": 2980
1070
+ },
1071
+ {
1072
+ "epoch": 0.20222446916076844,
1073
+ "grad_norm": 12213813248.0,
1074
+ "learning_rate": 9e-05,
1075
+ "loss": 8.100372314453125,
1076
+ "step": 3000
1077
+ },
1078
+ {
1079
+ "epoch": 0.20357263228850692,
1080
+ "grad_norm": 12616466432.0,
1081
+ "learning_rate": 9e-05,
1082
+ "loss": 8.101102447509765,
1083
+ "step": 3020
1084
+ },
1085
+ {
1086
+ "epoch": 0.20492079541624536,
1087
+ "grad_norm": 13019119616.0,
1088
+ "learning_rate": 9e-05,
1089
+ "loss": 8.099246215820312,
1090
+ "step": 3040
1091
+ },
1092
+ {
1093
+ "epoch": 0.20626895854398383,
1094
+ "grad_norm": 13421772800.0,
1095
+ "learning_rate": 9e-05,
1096
+ "loss": 8.084943389892578,
1097
+ "step": 3060
1098
+ },
1099
+ {
1100
+ "epoch": 0.20761712167172228,
1101
+ "grad_norm": 13757317120.0,
1102
+ "learning_rate": 9e-05,
1103
+ "loss": 8.084758758544922,
1104
+ "step": 3080
1105
+ },
1106
+ {
1107
+ "epoch": 0.20896528479946072,
1108
+ "grad_norm": 14159970304.0,
1109
+ "learning_rate": 9e-05,
1110
+ "loss": 8.089293670654296,
1111
+ "step": 3100
1112
+ }
1113
+ ],
1114
+ "logging_steps": 20,
1115
+ "max_steps": 3500,
1116
+ "num_input_tokens_seen": 0,
1117
+ "num_train_epochs": 1,
1118
+ "save_steps": 100,
1119
+ "stateful_callbacks": {
1120
+ "TrainerControl": {
1121
+ "args": {
1122
+ "should_epoch_stop": false,
1123
+ "should_evaluate": false,
1124
+ "should_log": false,
1125
+ "should_save": true,
1126
+ "should_training_stop": false
1127
+ },
1128
+ "attributes": {}
1129
+ }
1130
+ },
1131
+ "total_flos": 1.00014797881344e+16,
1132
+ "train_batch_size": 64,
1133
+ "trial_name": null,
1134
+ "trial_params": null
1135
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3100/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759f687547d62f080b691b398df9dd337d9e21ff107770065591d2088b29bf88
3
+ size 4920
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
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": 4
36
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:313d2f96b1d0496de39cb5334e90ed1cf5d66639363394cc24b68d7cf24308ef
3
+ size 33967272
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d8c19a60a3ea5fda86aa609e02b89288ad6b121fe7f07050d27334ee195c421
3
+ size 39573050
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cf9097d4513154245c48236b6ec5137b7ee2a21c9f58f2cba798ea275c6026f
3
+ size 14244
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42032ce2906e08b8158bbc61ccff7a8449b0a1b6c135848ffd257fa664fd4577
3
+ size 1064
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/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-gelu-100L_trash_run/checkpoint-3200/trainer_state.json ADDED
@@ -0,0 +1,1170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.21570610043815303,
6
+ "eval_steps": 2498,
7
+ "global_step": 3200,
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.299579620361328,
17
+ "step": 20
18
+ },
19
+ {
20
+ "epoch": 0.002696326255476913,
21
+ "grad_norm": 1.359375,
22
+ "learning_rate": 5.46e-05,
23
+ "loss": 8.047666931152344,
24
+ "step": 40
25
+ },
26
+ {
27
+ "epoch": 0.004044489383215369,
28
+ "grad_norm": 1.2734375,
29
+ "learning_rate": 8.259999999999999e-05,
30
+ "loss": 7.838191986083984,
31
+ "step": 60
32
+ },
33
+ {
34
+ "epoch": 0.005392652510953826,
35
+ "grad_norm": 1.265625,
36
+ "learning_rate": 0.0001106,
37
+ "loss": 7.57310791015625,
38
+ "step": 80
39
+ },
40
+ {
41
+ "epoch": 0.006740815638692282,
42
+ "grad_norm": 1.1953125,
43
+ "learning_rate": 0.0001386,
44
+ "loss": 7.233531951904297,
45
+ "step": 100
46
+ },
47
+ {
48
+ "epoch": 0.008088978766430738,
49
+ "grad_norm": 1.1328125,
50
+ "learning_rate": 0.00016659999999999998,
51
+ "loss": 6.860941314697266,
52
+ "step": 120
53
+ },
54
+ {
55
+ "epoch": 0.009437141894169195,
56
+ "grad_norm": 0.95703125,
57
+ "learning_rate": 0.00019460000000000001,
58
+ "loss": 6.51712646484375,
59
+ "step": 140
60
+ },
61
+ {
62
+ "epoch": 0.010785305021907651,
63
+ "grad_norm": 0.81640625,
64
+ "learning_rate": 0.0002226,
65
+ "loss": 6.234881591796875,
66
+ "step": 160
67
+ },
68
+ {
69
+ "epoch": 0.012133468149646108,
70
+ "grad_norm": 0.73828125,
71
+ "learning_rate": 0.00025059999999999997,
72
+ "loss": 6.024795913696289,
73
+ "step": 180
74
+ },
75
+ {
76
+ "epoch": 0.013481631277384564,
77
+ "grad_norm": 1.0390625,
78
+ "learning_rate": 0.0002786,
79
+ "loss": 5.837321472167969,
80
+ "step": 200
81
+ },
82
+ {
83
+ "epoch": 0.01482979440512302,
84
+ "grad_norm": 0.63671875,
85
+ "learning_rate": 0.00030659999999999997,
86
+ "loss": 5.647708129882813,
87
+ "step": 220
88
+ },
89
+ {
90
+ "epoch": 0.016177957532861477,
91
+ "grad_norm": 0.8984375,
92
+ "learning_rate": 0.0003346,
93
+ "loss": 5.43688735961914,
94
+ "step": 240
95
+ },
96
+ {
97
+ "epoch": 0.01752612066059993,
98
+ "grad_norm": 0.77734375,
99
+ "learning_rate": 0.00036260000000000003,
100
+ "loss": 5.210599517822265,
101
+ "step": 260
102
+ },
103
+ {
104
+ "epoch": 0.01887428378833839,
105
+ "grad_norm": 0.890625,
106
+ "learning_rate": 0.0003906,
107
+ "loss": 4.956959533691406,
108
+ "step": 280
109
+ },
110
+ {
111
+ "epoch": 0.020222446916076844,
112
+ "grad_norm": 0.84765625,
113
+ "learning_rate": 0.0004186,
114
+ "loss": 4.726874160766601,
115
+ "step": 300
116
+ },
117
+ {
118
+ "epoch": 0.021570610043815303,
119
+ "grad_norm": 0.87890625,
120
+ "learning_rate": 0.0004466,
121
+ "loss": 4.527945327758789,
122
+ "step": 320
123
+ },
124
+ {
125
+ "epoch": 0.022918773171553757,
126
+ "grad_norm": 1.3515625,
127
+ "learning_rate": 0.00047460000000000004,
128
+ "loss": 4.3517810821533205,
129
+ "step": 340
130
+ },
131
+ {
132
+ "epoch": 0.024266936299292215,
133
+ "grad_norm": 0.76171875,
134
+ "learning_rate": 0.0005026,
135
+ "loss": 4.212622451782226,
136
+ "step": 360
137
+ },
138
+ {
139
+ "epoch": 0.02561509942703067,
140
+ "grad_norm": 0.6640625,
141
+ "learning_rate": 0.0005306,
142
+ "loss": 4.076633453369141,
143
+ "step": 380
144
+ },
145
+ {
146
+ "epoch": 0.026963262554769128,
147
+ "grad_norm": 0.498046875,
148
+ "learning_rate": 0.0005586,
149
+ "loss": 3.9346649169921877,
150
+ "step": 400
151
+ },
152
+ {
153
+ "epoch": 0.028311425682507583,
154
+ "grad_norm": 0.85546875,
155
+ "learning_rate": 0.0005866,
156
+ "loss": 3.8310710906982424,
157
+ "step": 420
158
+ },
159
+ {
160
+ "epoch": 0.02965958881024604,
161
+ "grad_norm": 0.59375,
162
+ "learning_rate": 0.0006146,
163
+ "loss": 3.7676155090332033,
164
+ "step": 440
165
+ },
166
+ {
167
+ "epoch": 0.031007751937984496,
168
+ "grad_norm": 0.6171875,
169
+ "learning_rate": 0.0006426,
170
+ "loss": 3.6599285125732424,
171
+ "step": 460
172
+ },
173
+ {
174
+ "epoch": 0.032355915065722954,
175
+ "grad_norm": 0.5546875,
176
+ "learning_rate": 0.0006705999999999999,
177
+ "loss": 3.5886669158935547,
178
+ "step": 480
179
+ },
180
+ {
181
+ "epoch": 0.03370407819346141,
182
+ "grad_norm": 0.546875,
183
+ "learning_rate": 0.0006986,
184
+ "loss": 3.512950134277344,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 0.03505224132119986,
189
+ "grad_norm": 0.53125,
190
+ "learning_rate": 0.0007,
191
+ "loss": 3.444244384765625,
192
+ "step": 520
193
+ },
194
+ {
195
+ "epoch": 0.03640040444893832,
196
+ "grad_norm": 0.5078125,
197
+ "learning_rate": 0.0007,
198
+ "loss": 3.369425964355469,
199
+ "step": 540
200
+ },
201
+ {
202
+ "epoch": 0.03774856757667678,
203
+ "grad_norm": 0.5625,
204
+ "learning_rate": 0.0007,
205
+ "loss": 3.318771743774414,
206
+ "step": 560
207
+ },
208
+ {
209
+ "epoch": 0.03909673070441524,
210
+ "grad_norm": 0.470703125,
211
+ "learning_rate": 0.0007,
212
+ "loss": 3.2595756530761717,
213
+ "step": 580
214
+ },
215
+ {
216
+ "epoch": 0.04044489383215369,
217
+ "grad_norm": 0.53125,
218
+ "learning_rate": 0.0007,
219
+ "loss": 3.2241649627685547,
220
+ "step": 600
221
+ },
222
+ {
223
+ "epoch": 0.04179305695989215,
224
+ "grad_norm": 0.484375,
225
+ "learning_rate": 0.0007,
226
+ "loss": 3.1619989395141603,
227
+ "step": 620
228
+ },
229
+ {
230
+ "epoch": 0.043141220087630605,
231
+ "grad_norm": 0.5625,
232
+ "learning_rate": 0.0007,
233
+ "loss": 3.14036865234375,
234
+ "step": 640
235
+ },
236
+ {
237
+ "epoch": 0.044489383215369056,
238
+ "grad_norm": 0.41796875,
239
+ "learning_rate": 0.0007,
240
+ "loss": 3.083884620666504,
241
+ "step": 660
242
+ },
243
+ {
244
+ "epoch": 0.045837546343107514,
245
+ "grad_norm": 0.451171875,
246
+ "learning_rate": 0.0007,
247
+ "loss": 3.0591588973999024,
248
+ "step": 680
249
+ },
250
+ {
251
+ "epoch": 0.04718570947084597,
252
+ "grad_norm": 0.423828125,
253
+ "learning_rate": 0.0007,
254
+ "loss": 3.015497016906738,
255
+ "step": 700
256
+ },
257
+ {
258
+ "epoch": 0.04853387259858443,
259
+ "grad_norm": 0.41796875,
260
+ "learning_rate": 0.0007,
261
+ "loss": 2.9805700302124025,
262
+ "step": 720
263
+ },
264
+ {
265
+ "epoch": 0.04988203572632288,
266
+ "grad_norm": 0.46875,
267
+ "learning_rate": 0.0007,
268
+ "loss": 2.9426918029785156,
269
+ "step": 740
270
+ },
271
+ {
272
+ "epoch": 0.05123019885406134,
273
+ "grad_norm": 0.46484375,
274
+ "learning_rate": 0.0007,
275
+ "loss": 2.9274158477783203,
276
+ "step": 760
277
+ },
278
+ {
279
+ "epoch": 0.0525783619817998,
280
+ "grad_norm": 0.421875,
281
+ "learning_rate": 0.0007,
282
+ "loss": 2.931459426879883,
283
+ "step": 780
284
+ },
285
+ {
286
+ "epoch": 0.053926525109538256,
287
+ "grad_norm": 0.486328125,
288
+ "learning_rate": 0.0007,
289
+ "loss": 2.877642059326172,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.05527468823727671,
294
+ "grad_norm": 0.40234375,
295
+ "learning_rate": 0.0007,
296
+ "loss": 2.8646697998046875,
297
+ "step": 820
298
+ },
299
+ {
300
+ "epoch": 0.056622851365015166,
301
+ "grad_norm": 0.451171875,
302
+ "learning_rate": 0.0007,
303
+ "loss": 2.8328128814697267,
304
+ "step": 840
305
+ },
306
+ {
307
+ "epoch": 0.057971014492753624,
308
+ "grad_norm": 0.396484375,
309
+ "learning_rate": 0.0007,
310
+ "loss": 2.790274429321289,
311
+ "step": 860
312
+ },
313
+ {
314
+ "epoch": 0.05931917762049208,
315
+ "grad_norm": 0.423828125,
316
+ "learning_rate": 0.0007,
317
+ "loss": 2.7668125152587892,
318
+ "step": 880
319
+ },
320
+ {
321
+ "epoch": 0.06066734074823053,
322
+ "grad_norm": 0.427734375,
323
+ "learning_rate": 0.0007,
324
+ "loss": 2.7516881942749025,
325
+ "step": 900
326
+ },
327
+ {
328
+ "epoch": 0.06201550387596899,
329
+ "grad_norm": 0.42578125,
330
+ "learning_rate": 0.0007,
331
+ "loss": 2.722590446472168,
332
+ "step": 920
333
+ },
334
+ {
335
+ "epoch": 0.06336366700370745,
336
+ "grad_norm": 0.4375,
337
+ "learning_rate": 0.0007,
338
+ "loss": 2.708496856689453,
339
+ "step": 940
340
+ },
341
+ {
342
+ "epoch": 0.06471183013144591,
343
+ "grad_norm": 0.400390625,
344
+ "learning_rate": 0.0007,
345
+ "loss": 2.681654930114746,
346
+ "step": 960
347
+ },
348
+ {
349
+ "epoch": 0.06605999325918437,
350
+ "grad_norm": 0.396484375,
351
+ "learning_rate": 0.0007,
352
+ "loss": 2.6650571823120117,
353
+ "step": 980
354
+ },
355
+ {
356
+ "epoch": 0.06740815638692282,
357
+ "grad_norm": 0.41015625,
358
+ "learning_rate": 0.0007,
359
+ "loss": 2.6637720108032226,
360
+ "step": 1000
361
+ },
362
+ {
363
+ "epoch": 0.06875631951466127,
364
+ "grad_norm": 0.396484375,
365
+ "learning_rate": 0.0007,
366
+ "loss": 2.6311159133911133,
367
+ "step": 1020
368
+ },
369
+ {
370
+ "epoch": 0.07010448264239973,
371
+ "grad_norm": 0.408203125,
372
+ "learning_rate": 0.0007,
373
+ "loss": 2.615707015991211,
374
+ "step": 1040
375
+ },
376
+ {
377
+ "epoch": 0.07145264577013818,
378
+ "grad_norm": 0.404296875,
379
+ "learning_rate": 0.0007,
380
+ "loss": 2.6003103256225586,
381
+ "step": 1060
382
+ },
383
+ {
384
+ "epoch": 0.07280080889787664,
385
+ "grad_norm": 0.41015625,
386
+ "learning_rate": 0.0007,
387
+ "loss": 2.581348991394043,
388
+ "step": 1080
389
+ },
390
+ {
391
+ "epoch": 0.0741489720256151,
392
+ "grad_norm": 0.404296875,
393
+ "learning_rate": 0.0007,
394
+ "loss": 2.574792671203613,
395
+ "step": 1100
396
+ },
397
+ {
398
+ "epoch": 0.07549713515335356,
399
+ "grad_norm": 0.384765625,
400
+ "learning_rate": 0.0007,
401
+ "loss": 2.538400650024414,
402
+ "step": 1120
403
+ },
404
+ {
405
+ "epoch": 0.07684529828109202,
406
+ "grad_norm": 0.41796875,
407
+ "learning_rate": 0.0007,
408
+ "loss": 2.5249162673950196,
409
+ "step": 1140
410
+ },
411
+ {
412
+ "epoch": 0.07819346140883048,
413
+ "grad_norm": 0.404296875,
414
+ "learning_rate": 0.0007,
415
+ "loss": 2.5237232208251954,
416
+ "step": 1160
417
+ },
418
+ {
419
+ "epoch": 0.07954162453656892,
420
+ "grad_norm": 0.396484375,
421
+ "learning_rate": 0.0007,
422
+ "loss": 2.517318344116211,
423
+ "step": 1180
424
+ },
425
+ {
426
+ "epoch": 0.08088978766430738,
427
+ "grad_norm": 0.41796875,
428
+ "learning_rate": 0.0007,
429
+ "loss": 2.507699203491211,
430
+ "step": 1200
431
+ },
432
+ {
433
+ "epoch": 0.08223795079204584,
434
+ "grad_norm": 0.3828125,
435
+ "learning_rate": 0.0007,
436
+ "loss": 2.4932661056518555,
437
+ "step": 1220
438
+ },
439
+ {
440
+ "epoch": 0.0835861139197843,
441
+ "grad_norm": 0.365234375,
442
+ "learning_rate": 0.0007,
443
+ "loss": 2.4781061172485352,
444
+ "step": 1240
445
+ },
446
+ {
447
+ "epoch": 0.08493427704752275,
448
+ "grad_norm": 0.388671875,
449
+ "learning_rate": 0.0007,
450
+ "loss": 2.465612030029297,
451
+ "step": 1260
452
+ },
453
+ {
454
+ "epoch": 0.08628244017526121,
455
+ "grad_norm": 0.41796875,
456
+ "learning_rate": 0.0007,
457
+ "loss": 2.4463497161865235,
458
+ "step": 1280
459
+ },
460
+ {
461
+ "epoch": 0.08763060330299967,
462
+ "grad_norm": 0.390625,
463
+ "learning_rate": 0.0007,
464
+ "loss": 2.4374414443969727,
465
+ "step": 1300
466
+ },
467
+ {
468
+ "epoch": 0.08897876643073811,
469
+ "grad_norm": 0.35546875,
470
+ "learning_rate": 0.0007,
471
+ "loss": 2.4308101654052736,
472
+ "step": 1320
473
+ },
474
+ {
475
+ "epoch": 0.09032692955847657,
476
+ "grad_norm": 0.40234375,
477
+ "learning_rate": 0.0007,
478
+ "loss": 2.416130256652832,
479
+ "step": 1340
480
+ },
481
+ {
482
+ "epoch": 0.09167509268621503,
483
+ "grad_norm": 0.4296875,
484
+ "learning_rate": 0.0007,
485
+ "loss": 2.41524600982666,
486
+ "step": 1360
487
+ },
488
+ {
489
+ "epoch": 0.09302325581395349,
490
+ "grad_norm": 0.359375,
491
+ "learning_rate": 0.0007,
492
+ "loss": 2.392221450805664,
493
+ "step": 1380
494
+ },
495
+ {
496
+ "epoch": 0.09437141894169195,
497
+ "grad_norm": 0.373046875,
498
+ "learning_rate": 0.0007,
499
+ "loss": 2.383866310119629,
500
+ "step": 1400
501
+ },
502
+ {
503
+ "epoch": 0.0957195820694304,
504
+ "grad_norm": 0.396484375,
505
+ "learning_rate": 0.0007,
506
+ "loss": 2.394974708557129,
507
+ "step": 1420
508
+ },
509
+ {
510
+ "epoch": 0.09706774519716886,
511
+ "grad_norm": 0.396484375,
512
+ "learning_rate": 0.0007,
513
+ "loss": 2.3713239669799804,
514
+ "step": 1440
515
+ },
516
+ {
517
+ "epoch": 0.09841590832490732,
518
+ "grad_norm": 0.37109375,
519
+ "learning_rate": 0.0007,
520
+ "loss": 2.3597253799438476,
521
+ "step": 1460
522
+ },
523
+ {
524
+ "epoch": 0.09976407145264576,
525
+ "grad_norm": 0.34765625,
526
+ "learning_rate": 0.0007,
527
+ "loss": 2.3452421188354493,
528
+ "step": 1480
529
+ },
530
+ {
531
+ "epoch": 0.10111223458038422,
532
+ "grad_norm": 0.376953125,
533
+ "learning_rate": 0.0007,
534
+ "loss": 2.3336454391479493,
535
+ "step": 1500
536
+ },
537
+ {
538
+ "epoch": 0.10246039770812268,
539
+ "grad_norm": 0.39453125,
540
+ "learning_rate": 0.0007,
541
+ "loss": 2.3358110427856444,
542
+ "step": 1520
543
+ },
544
+ {
545
+ "epoch": 0.10380856083586114,
546
+ "grad_norm": 0.40625,
547
+ "learning_rate": 0.0007,
548
+ "loss": 2.324386787414551,
549
+ "step": 1540
550
+ },
551
+ {
552
+ "epoch": 0.1051567239635996,
553
+ "grad_norm": 0.376953125,
554
+ "learning_rate": 0.0007,
555
+ "loss": 2.313844108581543,
556
+ "step": 1560
557
+ },
558
+ {
559
+ "epoch": 0.10650488709133805,
560
+ "grad_norm": 0.349609375,
561
+ "learning_rate": 0.0007,
562
+ "loss": 2.3294256210327147,
563
+ "step": 1580
564
+ },
565
+ {
566
+ "epoch": 0.10785305021907651,
567
+ "grad_norm": 0.40234375,
568
+ "learning_rate": 0.0007,
569
+ "loss": 2.297942352294922,
570
+ "step": 1600
571
+ },
572
+ {
573
+ "epoch": 0.10920121334681497,
574
+ "grad_norm": 0.357421875,
575
+ "learning_rate": 0.0007,
576
+ "loss": 2.296767997741699,
577
+ "step": 1620
578
+ },
579
+ {
580
+ "epoch": 0.11054937647455342,
581
+ "grad_norm": 0.37890625,
582
+ "learning_rate": 0.0007,
583
+ "loss": 2.3081844329833983,
584
+ "step": 1640
585
+ },
586
+ {
587
+ "epoch": 0.11189753960229187,
588
+ "grad_norm": 0.35546875,
589
+ "learning_rate": 0.0007,
590
+ "loss": 2.2883626937866213,
591
+ "step": 1660
592
+ },
593
+ {
594
+ "epoch": 0.11324570273003033,
595
+ "grad_norm": 0.375,
596
+ "learning_rate": 0.0007,
597
+ "loss": 2.272566223144531,
598
+ "step": 1680
599
+ },
600
+ {
601
+ "epoch": 0.11459386585776879,
602
+ "grad_norm": 0.373046875,
603
+ "learning_rate": 0.0007,
604
+ "loss": 2.281250762939453,
605
+ "step": 1700
606
+ },
607
+ {
608
+ "epoch": 0.11594202898550725,
609
+ "grad_norm": 0.41796875,
610
+ "learning_rate": 0.0007,
611
+ "loss": 2.257417106628418,
612
+ "step": 1720
613
+ },
614
+ {
615
+ "epoch": 0.1172901921132457,
616
+ "grad_norm": 0.365234375,
617
+ "learning_rate": 0.0007,
618
+ "loss": 2.2473636627197267,
619
+ "step": 1740
620
+ },
621
+ {
622
+ "epoch": 0.11863835524098416,
623
+ "grad_norm": 0.380859375,
624
+ "learning_rate": 0.0007,
625
+ "loss": 2.253912925720215,
626
+ "step": 1760
627
+ },
628
+ {
629
+ "epoch": 0.11998651836872262,
630
+ "grad_norm": 0.3515625,
631
+ "learning_rate": 0.0007,
632
+ "loss": 2.2672780990600585,
633
+ "step": 1780
634
+ },
635
+ {
636
+ "epoch": 0.12133468149646107,
637
+ "grad_norm": 0.3671875,
638
+ "learning_rate": 0.0007,
639
+ "loss": 2.261806106567383,
640
+ "step": 1800
641
+ },
642
+ {
643
+ "epoch": 0.12268284462419952,
644
+ "grad_norm": 0.375,
645
+ "learning_rate": 0.0007,
646
+ "loss": 2.2513105392456056,
647
+ "step": 1820
648
+ },
649
+ {
650
+ "epoch": 0.12403100775193798,
651
+ "grad_norm": 0.369140625,
652
+ "learning_rate": 0.0007,
653
+ "loss": 2.2201774597167967,
654
+ "step": 1840
655
+ },
656
+ {
657
+ "epoch": 0.12537917087967643,
658
+ "grad_norm": 0.35546875,
659
+ "learning_rate": 0.0007,
660
+ "loss": 2.2213924407958983,
661
+ "step": 1860
662
+ },
663
+ {
664
+ "epoch": 0.1267273340074149,
665
+ "grad_norm": 0.361328125,
666
+ "learning_rate": 0.0007,
667
+ "loss": 2.21158447265625,
668
+ "step": 1880
669
+ },
670
+ {
671
+ "epoch": 0.12807549713515334,
672
+ "grad_norm": 0.357421875,
673
+ "learning_rate": 0.0007,
674
+ "loss": 2.218313217163086,
675
+ "step": 1900
676
+ },
677
+ {
678
+ "epoch": 0.12942366026289182,
679
+ "grad_norm": 0.361328125,
680
+ "learning_rate": 0.0007,
681
+ "loss": 2.1961637496948243,
682
+ "step": 1920
683
+ },
684
+ {
685
+ "epoch": 0.13077182339063026,
686
+ "grad_norm": 0.369140625,
687
+ "learning_rate": 0.0007,
688
+ "loss": 2.2057926177978517,
689
+ "step": 1940
690
+ },
691
+ {
692
+ "epoch": 0.13211998651836873,
693
+ "grad_norm": 0.345703125,
694
+ "learning_rate": 0.0007,
695
+ "loss": 2.181648826599121,
696
+ "step": 1960
697
+ },
698
+ {
699
+ "epoch": 0.13346814964610718,
700
+ "grad_norm": 0.380859375,
701
+ "learning_rate": 0.0007,
702
+ "loss": 2.203908920288086,
703
+ "step": 1980
704
+ },
705
+ {
706
+ "epoch": 0.13481631277384565,
707
+ "grad_norm": 0.341796875,
708
+ "learning_rate": 0.0007,
709
+ "loss": 2.1758956909179688,
710
+ "step": 2000
711
+ },
712
+ {
713
+ "epoch": 0.1361644759015841,
714
+ "grad_norm": 0.35546875,
715
+ "learning_rate": 0.0007,
716
+ "loss": 2.185748481750488,
717
+ "step": 2020
718
+ },
719
+ {
720
+ "epoch": 0.13751263902932254,
721
+ "grad_norm": 0.41796875,
722
+ "learning_rate": 0.0007,
723
+ "loss": 2.165380668640137,
724
+ "step": 2040
725
+ },
726
+ {
727
+ "epoch": 0.138860802157061,
728
+ "grad_norm": 0.3515625,
729
+ "learning_rate": 0.0007,
730
+ "loss": 2.1852161407470705,
731
+ "step": 2060
732
+ },
733
+ {
734
+ "epoch": 0.14020896528479945,
735
+ "grad_norm": 0.34765625,
736
+ "learning_rate": 0.0007,
737
+ "loss": 2.160782051086426,
738
+ "step": 2080
739
+ },
740
+ {
741
+ "epoch": 0.14155712841253792,
742
+ "grad_norm": 0.337890625,
743
+ "learning_rate": 0.0007,
744
+ "loss": 2.152023124694824,
745
+ "step": 2100
746
+ },
747
+ {
748
+ "epoch": 0.14290529154027637,
749
+ "grad_norm": 0.359375,
750
+ "learning_rate": 0.0007,
751
+ "loss": 2.160308837890625,
752
+ "step": 2120
753
+ },
754
+ {
755
+ "epoch": 0.14425345466801484,
756
+ "grad_norm": 0.34765625,
757
+ "learning_rate": 0.0007,
758
+ "loss": 2.1609643936157226,
759
+ "step": 2140
760
+ },
761
+ {
762
+ "epoch": 0.14560161779575329,
763
+ "grad_norm": 0.357421875,
764
+ "learning_rate": 0.0007,
765
+ "loss": 2.1475265502929686,
766
+ "step": 2160
767
+ },
768
+ {
769
+ "epoch": 0.14694978092349173,
770
+ "grad_norm": 0.345703125,
771
+ "learning_rate": 0.0007,
772
+ "loss": 2.155929183959961,
773
+ "step": 2180
774
+ },
775
+ {
776
+ "epoch": 0.1482979440512302,
777
+ "grad_norm": 0.341796875,
778
+ "learning_rate": 0.0007,
779
+ "loss": 2.1293373107910156,
780
+ "step": 2200
781
+ },
782
+ {
783
+ "epoch": 0.14964610717896865,
784
+ "grad_norm": 0.35546875,
785
+ "learning_rate": 0.0007,
786
+ "loss": 2.137188720703125,
787
+ "step": 2220
788
+ },
789
+ {
790
+ "epoch": 0.15099427030670712,
791
+ "grad_norm": 0.345703125,
792
+ "learning_rate": 0.0007,
793
+ "loss": 2.1462854385375976,
794
+ "step": 2240
795
+ },
796
+ {
797
+ "epoch": 0.15234243343444556,
798
+ "grad_norm": 0.365234375,
799
+ "learning_rate": 0.0007,
800
+ "loss": 2.1255794525146485,
801
+ "step": 2260
802
+ },
803
+ {
804
+ "epoch": 0.15369059656218403,
805
+ "grad_norm": 0.33984375,
806
+ "learning_rate": 0.0007,
807
+ "loss": 2.1365018844604493,
808
+ "step": 2280
809
+ },
810
+ {
811
+ "epoch": 0.15503875968992248,
812
+ "grad_norm": 0.337890625,
813
+ "learning_rate": 0.0007,
814
+ "loss": 2.1164464950561523,
815
+ "step": 2300
816
+ },
817
+ {
818
+ "epoch": 0.15638692281766095,
819
+ "grad_norm": 0.361328125,
820
+ "learning_rate": 0.0007,
821
+ "loss": 2.127320671081543,
822
+ "step": 2320
823
+ },
824
+ {
825
+ "epoch": 0.1577350859453994,
826
+ "grad_norm": 0.3359375,
827
+ "learning_rate": 0.0007,
828
+ "loss": 2.117641067504883,
829
+ "step": 2340
830
+ },
831
+ {
832
+ "epoch": 0.15908324907313784,
833
+ "grad_norm": 0.357421875,
834
+ "learning_rate": 0.0007,
835
+ "loss": 2.119377326965332,
836
+ "step": 2360
837
+ },
838
+ {
839
+ "epoch": 0.1604314122008763,
840
+ "grad_norm": 0.33203125,
841
+ "learning_rate": 0.0007,
842
+ "loss": 2.121909332275391,
843
+ "step": 2380
844
+ },
845
+ {
846
+ "epoch": 0.16177957532861476,
847
+ "grad_norm": 0.361328125,
848
+ "learning_rate": 0.0007,
849
+ "loss": 2.10235538482666,
850
+ "step": 2400
851
+ },
852
+ {
853
+ "epoch": 0.16312773845635323,
854
+ "grad_norm": 0.353515625,
855
+ "learning_rate": 0.0007,
856
+ "loss": 2.1056692123413088,
857
+ "step": 2420
858
+ },
859
+ {
860
+ "epoch": 0.16447590158409167,
861
+ "grad_norm": 0.3359375,
862
+ "learning_rate": 0.0007,
863
+ "loss": 2.093578910827637,
864
+ "step": 2440
865
+ },
866
+ {
867
+ "epoch": 0.16582406471183014,
868
+ "grad_norm": 0.337890625,
869
+ "learning_rate": 0.0007,
870
+ "loss": 2.1059795379638673,
871
+ "step": 2460
872
+ },
873
+ {
874
+ "epoch": 0.1671722278395686,
875
+ "grad_norm": 0.361328125,
876
+ "learning_rate": 0.0007,
877
+ "loss": 2.106010627746582,
878
+ "step": 2480
879
+ },
880
+ {
881
+ "epoch": 0.1683855746545332,
882
+ "eval_loss": 2.100916624069214,
883
+ "eval_runtime": 15.3711,
884
+ "eval_samples_per_second": 619.799,
885
+ "eval_steps_per_second": 0.651,
886
+ "step": 2498
887
+ },
888
+ {
889
+ "epoch": 0.16852039096730703,
890
+ "grad_norm": 0.33984375,
891
+ "learning_rate": 0.0007,
892
+ "loss": 2.087843322753906,
893
+ "step": 2500
894
+ },
895
+ {
896
+ "epoch": 0.16852039096730703,
897
+ "eval_loss": 2.1001250743865967,
898
+ "eval_runtime": 18.4156,
899
+ "eval_samples_per_second": 517.335,
900
+ "eval_steps_per_second": 0.543,
901
+ "step": 2500
902
+ },
903
+ {
904
+ "epoch": 0.1698685540950455,
905
+ "grad_norm": 20840448.0,
906
+ "learning_rate": 3.4200000000000005e-05,
907
+ "loss": 2.092716407775879,
908
+ "step": 2520
909
+ },
910
+ {
911
+ "epoch": 0.17121671722278395,
912
+ "grad_norm": 22937600.0,
913
+ "learning_rate": 7.020000000000001e-05,
914
+ "loss": 2.106246566772461,
915
+ "step": 2540
916
+ },
917
+ {
918
+ "epoch": 0.17256488035052242,
919
+ "grad_norm": 116391936.0,
920
+ "learning_rate": 9e-05,
921
+ "loss": 3.2435539245605467,
922
+ "step": 2560
923
+ },
924
+ {
925
+ "epoch": 0.17391304347826086,
926
+ "grad_norm": 283115520.0,
927
+ "learning_rate": 9e-05,
928
+ "loss": 7.010993957519531,
929
+ "step": 2580
930
+ },
931
+ {
932
+ "epoch": 0.17526120660599934,
933
+ "grad_norm": 583008256.0,
934
+ "learning_rate": 9e-05,
935
+ "loss": 7.576824188232422,
936
+ "step": 2600
937
+ },
938
+ {
939
+ "epoch": 0.17660936973373778,
940
+ "grad_norm": 998244352.0,
941
+ "learning_rate": 9e-05,
942
+ "loss": 7.7992095947265625,
943
+ "step": 2620
944
+ },
945
+ {
946
+ "epoch": 0.17795753286147623,
947
+ "grad_norm": 1501560832.0,
948
+ "learning_rate": 9e-05,
949
+ "loss": 7.901728820800781,
950
+ "step": 2640
951
+ },
952
+ {
953
+ "epoch": 0.1793056959892147,
954
+ "grad_norm": 2055208960.0,
955
+ "learning_rate": 9e-05,
956
+ "loss": 7.956874847412109,
957
+ "step": 2660
958
+ },
959
+ {
960
+ "epoch": 0.18065385911695314,
961
+ "grad_norm": 2650800128.0,
962
+ "learning_rate": 9e-05,
963
+ "loss": 7.995229339599609,
964
+ "step": 2680
965
+ },
966
+ {
967
+ "epoch": 0.1820020222446916,
968
+ "grad_norm": 3288334336.0,
969
+ "learning_rate": 9e-05,
970
+ "loss": 8.019290161132812,
971
+ "step": 2700
972
+ },
973
+ {
974
+ "epoch": 0.18335018537243006,
975
+ "grad_norm": 3942645760.0,
976
+ "learning_rate": 9e-05,
977
+ "loss": 8.043710327148437,
978
+ "step": 2720
979
+ },
980
+ {
981
+ "epoch": 0.18469834850016853,
982
+ "grad_norm": 4630511616.0,
983
+ "learning_rate": 9e-05,
984
+ "loss": 8.043434143066406,
985
+ "step": 2740
986
+ },
987
+ {
988
+ "epoch": 0.18604651162790697,
989
+ "grad_norm": 5301600256.0,
990
+ "learning_rate": 9e-05,
991
+ "loss": 8.049758911132812,
992
+ "step": 2760
993
+ },
994
+ {
995
+ "epoch": 0.18739467475564545,
996
+ "grad_norm": 5972688896.0,
997
+ "learning_rate": 9e-05,
998
+ "loss": 8.061854553222656,
999
+ "step": 2780
1000
+ },
1001
+ {
1002
+ "epoch": 0.1887428378833839,
1003
+ "grad_norm": 6643777536.0,
1004
+ "learning_rate": 9e-05,
1005
+ "loss": 8.073146057128906,
1006
+ "step": 2800
1007
+ },
1008
+ {
1009
+ "epoch": 0.19009100101112233,
1010
+ "grad_norm": 7281311744.0,
1011
+ "learning_rate": 9e-05,
1012
+ "loss": 8.076551818847657,
1013
+ "step": 2820
1014
+ },
1015
+ {
1016
+ "epoch": 0.1914391641388608,
1017
+ "grad_norm": 7918845952.0,
1018
+ "learning_rate": 9e-05,
1019
+ "loss": 8.078580474853515,
1020
+ "step": 2840
1021
+ },
1022
+ {
1023
+ "epoch": 0.19278732726659925,
1024
+ "grad_norm": 8556380160.0,
1025
+ "learning_rate": 9e-05,
1026
+ "loss": 8.073567199707032,
1027
+ "step": 2860
1028
+ },
1029
+ {
1030
+ "epoch": 0.19413549039433772,
1031
+ "grad_norm": 9126805504.0,
1032
+ "learning_rate": 9e-05,
1033
+ "loss": 8.08355941772461,
1034
+ "step": 2880
1035
+ },
1036
+ {
1037
+ "epoch": 0.19548365352207617,
1038
+ "grad_norm": 9730785280.0,
1039
+ "learning_rate": 9e-05,
1040
+ "loss": 8.085595703125,
1041
+ "step": 2900
1042
+ },
1043
+ {
1044
+ "epoch": 0.19683181664981464,
1045
+ "grad_norm": 10267656192.0,
1046
+ "learning_rate": 9e-05,
1047
+ "loss": 8.09107666015625,
1048
+ "step": 2920
1049
+ },
1050
+ {
1051
+ "epoch": 0.19817997977755308,
1052
+ "grad_norm": 10804527104.0,
1053
+ "learning_rate": 9e-05,
1054
+ "loss": 8.093254089355469,
1055
+ "step": 2940
1056
+ },
1057
+ {
1058
+ "epoch": 0.19952814290529153,
1059
+ "grad_norm": 11341398016.0,
1060
+ "learning_rate": 9e-05,
1061
+ "loss": 8.094740295410157,
1062
+ "step": 2960
1063
+ },
1064
+ {
1065
+ "epoch": 0.20087630603303,
1066
+ "grad_norm": 11811160064.0,
1067
+ "learning_rate": 9e-05,
1068
+ "loss": 8.100899505615235,
1069
+ "step": 2980
1070
+ },
1071
+ {
1072
+ "epoch": 0.20222446916076844,
1073
+ "grad_norm": 12213813248.0,
1074
+ "learning_rate": 9e-05,
1075
+ "loss": 8.100372314453125,
1076
+ "step": 3000
1077
+ },
1078
+ {
1079
+ "epoch": 0.20357263228850692,
1080
+ "grad_norm": 12616466432.0,
1081
+ "learning_rate": 9e-05,
1082
+ "loss": 8.101102447509765,
1083
+ "step": 3020
1084
+ },
1085
+ {
1086
+ "epoch": 0.20492079541624536,
1087
+ "grad_norm": 13019119616.0,
1088
+ "learning_rate": 9e-05,
1089
+ "loss": 8.099246215820312,
1090
+ "step": 3040
1091
+ },
1092
+ {
1093
+ "epoch": 0.20626895854398383,
1094
+ "grad_norm": 13421772800.0,
1095
+ "learning_rate": 9e-05,
1096
+ "loss": 8.084943389892578,
1097
+ "step": 3060
1098
+ },
1099
+ {
1100
+ "epoch": 0.20761712167172228,
1101
+ "grad_norm": 13757317120.0,
1102
+ "learning_rate": 9e-05,
1103
+ "loss": 8.084758758544922,
1104
+ "step": 3080
1105
+ },
1106
+ {
1107
+ "epoch": 0.20896528479946072,
1108
+ "grad_norm": 14159970304.0,
1109
+ "learning_rate": 9e-05,
1110
+ "loss": 8.089293670654296,
1111
+ "step": 3100
1112
+ },
1113
+ {
1114
+ "epoch": 0.2103134479271992,
1115
+ "grad_norm": 14562623488.0,
1116
+ "learning_rate": 9e-05,
1117
+ "loss": 8.095923614501952,
1118
+ "step": 3120
1119
+ },
1120
+ {
1121
+ "epoch": 0.21166161105493764,
1122
+ "grad_norm": 14965276672.0,
1123
+ "learning_rate": 9e-05,
1124
+ "loss": 8.09695816040039,
1125
+ "step": 3140
1126
+ },
1127
+ {
1128
+ "epoch": 0.2130097741826761,
1129
+ "grad_norm": 15367929856.0,
1130
+ "learning_rate": 9e-05,
1131
+ "loss": 8.095108795166016,
1132
+ "step": 3160
1133
+ },
1134
+ {
1135
+ "epoch": 0.21435793731041455,
1136
+ "grad_norm": 15770583040.0,
1137
+ "learning_rate": 9e-05,
1138
+ "loss": 8.094277954101562,
1139
+ "step": 3180
1140
+ },
1141
+ {
1142
+ "epoch": 0.21570610043815303,
1143
+ "grad_norm": 16173236224.0,
1144
+ "learning_rate": 9e-05,
1145
+ "loss": 8.096886444091798,
1146
+ "step": 3200
1147
+ }
1148
+ ],
1149
+ "logging_steps": 20,
1150
+ "max_steps": 3500,
1151
+ "num_input_tokens_seen": 0,
1152
+ "num_train_epochs": 1,
1153
+ "save_steps": 100,
1154
+ "stateful_callbacks": {
1155
+ "TrainerControl": {
1156
+ "args": {
1157
+ "should_epoch_stop": false,
1158
+ "should_evaluate": false,
1159
+ "should_log": false,
1160
+ "should_save": true,
1161
+ "should_training_stop": false
1162
+ },
1163
+ "attributes": {}
1164
+ }
1165
+ },
1166
+ "total_flos": 1.03241081683968e+16,
1167
+ "train_batch_size": 64,
1168
+ "trial_name": null,
1169
+ "trial_params": null
1170
+ }
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3200/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759f687547d62f080b691b398df9dd337d9e21ff107770065591d2088b29bf88
3
+ size 4920
zain/Activation/out/glu-gelu-100L_trash_run/checkpoint-3300/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
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": 4
36
+ }