Training in progress, step 4000, checkpoint
Browse files- checkpoint-4000/config.json +40 -0
- checkpoint-4000/generation_config.json +12 -0
- checkpoint-4000/model.safetensors +3 -0
- checkpoint-4000/optimizer.pt +3 -0
- checkpoint-4000/rng_state.pth +3 -0
- checkpoint-4000/scheduler.pt +3 -0
- checkpoint-4000/trainer_state.json +313 -0
- checkpoint-4000/training_args.bin +3 -0
checkpoint-4000/config.json
ADDED
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "Grogros/dmWM-llama-3.2-1B-Instruct-OWTWM-DistillationWM-Al4-wmToken-d4-a0.1-v2-meta-OWT",
|
3 |
+
"architectures": [
|
4 |
+
"LlamaForCausalLM"
|
5 |
+
],
|
6 |
+
"attention_bias": false,
|
7 |
+
"attention_dropout": 0.0,
|
8 |
+
"bos_token_id": 128000,
|
9 |
+
"eos_token_id": [
|
10 |
+
128001,
|
11 |
+
128008,
|
12 |
+
128009
|
13 |
+
],
|
14 |
+
"head_dim": 64,
|
15 |
+
"hidden_act": "silu",
|
16 |
+
"hidden_size": 2048,
|
17 |
+
"initializer_range": 0.02,
|
18 |
+
"intermediate_size": 8192,
|
19 |
+
"max_position_embeddings": 131072,
|
20 |
+
"mlp_bias": false,
|
21 |
+
"model_type": "llama",
|
22 |
+
"num_attention_heads": 32,
|
23 |
+
"num_hidden_layers": 16,
|
24 |
+
"num_key_value_heads": 8,
|
25 |
+
"pretraining_tp": 1,
|
26 |
+
"rms_norm_eps": 1e-05,
|
27 |
+
"rope_scaling": {
|
28 |
+
"factor": 32.0,
|
29 |
+
"high_freq_factor": 4.0,
|
30 |
+
"low_freq_factor": 1.0,
|
31 |
+
"original_max_position_embeddings": 8192,
|
32 |
+
"rope_type": "llama3"
|
33 |
+
},
|
34 |
+
"rope_theta": 500000.0,
|
35 |
+
"tie_word_embeddings": true,
|
36 |
+
"torch_dtype": "float32",
|
37 |
+
"transformers_version": "4.46.3",
|
38 |
+
"use_cache": true,
|
39 |
+
"vocab_size": 128257
|
40 |
+
}
|
checkpoint-4000/generation_config.json
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token_id": 128000,
|
3 |
+
"do_sample": true,
|
4 |
+
"eos_token_id": [
|
5 |
+
128001,
|
6 |
+
128008,
|
7 |
+
128009
|
8 |
+
],
|
9 |
+
"temperature": 0.6,
|
10 |
+
"top_p": 0.9,
|
11 |
+
"transformers_version": "4.46.3"
|
12 |
+
}
|
checkpoint-4000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:44dbea65727d4aecda8154871c13d8202093be43c52f2a3ad36d2f1a740bb2ce
|
3 |
+
size 4943282520
|
checkpoint-4000/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d1447a7faee858fdce2f8371d565d431d09427f621ddd21d0e4a24a02f02f6eb
|
3 |
+
size 3724602
|
checkpoint-4000/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f96d9df5ac20d0d02e7be698a6a5b6f28f5af246066a71da5ac8e379b2953797
|
3 |
+
size 14244
|
checkpoint-4000/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fd9d32ba7e86ceef28e79b0a75c6df1b71c1ab36ea11926de64b4cba8dda47dd
|
3 |
+
size 1064
|
checkpoint-4000/trainer_state.json
ADDED
@@ -0,0 +1,313 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 0.12254714235382423,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 4000,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.003063678558845606,
|
13 |
+
"grad_norm": 2.5270798206329346,
|
14 |
+
"learning_rate": 1.0000000000000002e-06,
|
15 |
+
"loss": 3.1564,
|
16 |
+
"step": 100
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.006127357117691212,
|
20 |
+
"grad_norm": 2.604092597961426,
|
21 |
+
"learning_rate": 2.0000000000000003e-06,
|
22 |
+
"loss": 3.1166,
|
23 |
+
"step": 200
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.009191035676536818,
|
27 |
+
"grad_norm": 2.5005924701690674,
|
28 |
+
"learning_rate": 3e-06,
|
29 |
+
"loss": 3.0905,
|
30 |
+
"step": 300
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.012254714235382424,
|
34 |
+
"grad_norm": 2.7525177001953125,
|
35 |
+
"learning_rate": 4.000000000000001e-06,
|
36 |
+
"loss": 3.0652,
|
37 |
+
"step": 400
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.015318392794228029,
|
41 |
+
"grad_norm": 2.5251283645629883,
|
42 |
+
"learning_rate": 5e-06,
|
43 |
+
"loss": 3.0351,
|
44 |
+
"step": 500
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.018382071353073636,
|
48 |
+
"grad_norm": 2.6771860122680664,
|
49 |
+
"learning_rate": 6e-06,
|
50 |
+
"loss": 3.0028,
|
51 |
+
"step": 600
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.02144574991191924,
|
55 |
+
"grad_norm": 2.4799458980560303,
|
56 |
+
"learning_rate": 7e-06,
|
57 |
+
"loss": 3.0007,
|
58 |
+
"step": 700
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.024509428470764847,
|
62 |
+
"grad_norm": 2.8402369022369385,
|
63 |
+
"learning_rate": 8.000000000000001e-06,
|
64 |
+
"loss": 2.9753,
|
65 |
+
"step": 800
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.027573107029610452,
|
69 |
+
"grad_norm": 2.446842908859253,
|
70 |
+
"learning_rate": 9e-06,
|
71 |
+
"loss": 2.9737,
|
72 |
+
"step": 900
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.030636785588456058,
|
76 |
+
"grad_norm": 2.3989391326904297,
|
77 |
+
"learning_rate": 1e-05,
|
78 |
+
"loss": 2.9618,
|
79 |
+
"step": 1000
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.03370046414730166,
|
83 |
+
"grad_norm": 2.469160795211792,
|
84 |
+
"learning_rate": 9.99695413509548e-06,
|
85 |
+
"loss": 2.928,
|
86 |
+
"step": 1100
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.03676414270614727,
|
90 |
+
"grad_norm": 2.4107494354248047,
|
91 |
+
"learning_rate": 9.987820251299121e-06,
|
92 |
+
"loss": 2.9214,
|
93 |
+
"step": 1200
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.039827821264992874,
|
97 |
+
"grad_norm": 2.540278911590576,
|
98 |
+
"learning_rate": 9.972609476841368e-06,
|
99 |
+
"loss": 2.9198,
|
100 |
+
"step": 1300
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.04289149982383848,
|
104 |
+
"grad_norm": 2.4398670196533203,
|
105 |
+
"learning_rate": 9.951340343707852e-06,
|
106 |
+
"loss": 2.9182,
|
107 |
+
"step": 1400
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.04595517838268409,
|
111 |
+
"grad_norm": 2.49687123298645,
|
112 |
+
"learning_rate": 9.924038765061042e-06,
|
113 |
+
"loss": 2.8994,
|
114 |
+
"step": 1500
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.049018856941529694,
|
118 |
+
"grad_norm": 2.4655182361602783,
|
119 |
+
"learning_rate": 9.890738003669029e-06,
|
120 |
+
"loss": 2.8805,
|
121 |
+
"step": 1600
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.0520825355003753,
|
125 |
+
"grad_norm": 2.4170656204223633,
|
126 |
+
"learning_rate": 9.851478631379982e-06,
|
127 |
+
"loss": 2.8829,
|
128 |
+
"step": 1700
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.055146214059220905,
|
132 |
+
"grad_norm": 2.362334966659546,
|
133 |
+
"learning_rate": 9.806308479691595e-06,
|
134 |
+
"loss": 2.8891,
|
135 |
+
"step": 1800
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.058209892618066514,
|
139 |
+
"grad_norm": 2.426173210144043,
|
140 |
+
"learning_rate": 9.755282581475769e-06,
|
141 |
+
"loss": 2.875,
|
142 |
+
"step": 1900
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.061273571176912116,
|
146 |
+
"grad_norm": 2.6380629539489746,
|
147 |
+
"learning_rate": 9.698463103929542e-06,
|
148 |
+
"loss": 2.879,
|
149 |
+
"step": 2000
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.06433724973575772,
|
153 |
+
"grad_norm": 2.5123095512390137,
|
154 |
+
"learning_rate": 9.635919272833938e-06,
|
155 |
+
"loss": 2.8622,
|
156 |
+
"step": 2100
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.06740092829460333,
|
160 |
+
"grad_norm": 2.3789472579956055,
|
161 |
+
"learning_rate": 9.567727288213005e-06,
|
162 |
+
"loss": 2.8669,
|
163 |
+
"step": 2200
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.07046460685344894,
|
167 |
+
"grad_norm": 2.387221336364746,
|
168 |
+
"learning_rate": 9.493970231495836e-06,
|
169 |
+
"loss": 2.8641,
|
170 |
+
"step": 2300
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.07352828541229454,
|
174 |
+
"grad_norm": 2.4947221279144287,
|
175 |
+
"learning_rate": 9.414737964294636e-06,
|
176 |
+
"loss": 2.8525,
|
177 |
+
"step": 2400
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.07659196397114015,
|
181 |
+
"grad_norm": 2.390381097793579,
|
182 |
+
"learning_rate": 9.330127018922195e-06,
|
183 |
+
"loss": 2.8497,
|
184 |
+
"step": 2500
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.07965564252998575,
|
188 |
+
"grad_norm": 2.451345682144165,
|
189 |
+
"learning_rate": 9.24024048078213e-06,
|
190 |
+
"loss": 2.8482,
|
191 |
+
"step": 2600
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.08271932108883136,
|
195 |
+
"grad_norm": 2.317512035369873,
|
196 |
+
"learning_rate": 9.145187862775208e-06,
|
197 |
+
"loss": 2.8486,
|
198 |
+
"step": 2700
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.08578299964767697,
|
202 |
+
"grad_norm": 2.620295524597168,
|
203 |
+
"learning_rate": 9.045084971874738e-06,
|
204 |
+
"loss": 2.8474,
|
205 |
+
"step": 2800
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.08884667820652258,
|
209 |
+
"grad_norm": 2.3616020679473877,
|
210 |
+
"learning_rate": 8.94005376803361e-06,
|
211 |
+
"loss": 2.8283,
|
212 |
+
"step": 2900
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.09191035676536818,
|
216 |
+
"grad_norm": 2.517153263092041,
|
217 |
+
"learning_rate": 8.83022221559489e-06,
|
218 |
+
"loss": 2.8451,
|
219 |
+
"step": 3000
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.09497403532421378,
|
223 |
+
"grad_norm": 2.6935179233551025,
|
224 |
+
"learning_rate": 8.715724127386971e-06,
|
225 |
+
"loss": 2.8472,
|
226 |
+
"step": 3100
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.09803771388305939,
|
230 |
+
"grad_norm": 2.4080545902252197,
|
231 |
+
"learning_rate": 8.596699001693257e-06,
|
232 |
+
"loss": 2.8428,
|
233 |
+
"step": 3200
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.101101392441905,
|
237 |
+
"grad_norm": 2.369750499725342,
|
238 |
+
"learning_rate": 8.473291852294986e-06,
|
239 |
+
"loss": 2.829,
|
240 |
+
"step": 3300
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.1041650710007506,
|
244 |
+
"grad_norm": 2.3385934829711914,
|
245 |
+
"learning_rate": 8.345653031794292e-06,
|
246 |
+
"loss": 2.8349,
|
247 |
+
"step": 3400
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.1072287495595962,
|
251 |
+
"grad_norm": 2.2659566402435303,
|
252 |
+
"learning_rate": 8.213938048432697e-06,
|
253 |
+
"loss": 2.8306,
|
254 |
+
"step": 3500
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.11029242811844181,
|
258 |
+
"grad_norm": 2.508164882659912,
|
259 |
+
"learning_rate": 8.078307376628292e-06,
|
260 |
+
"loss": 2.8225,
|
261 |
+
"step": 3600
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.11335610667728742,
|
265 |
+
"grad_norm": 2.302351713180542,
|
266 |
+
"learning_rate": 7.938926261462366e-06,
|
267 |
+
"loss": 2.808,
|
268 |
+
"step": 3700
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.11641978523613303,
|
272 |
+
"grad_norm": 2.3972647190093994,
|
273 |
+
"learning_rate": 7.795964517353734e-06,
|
274 |
+
"loss": 2.8214,
|
275 |
+
"step": 3800
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.11948346379497864,
|
279 |
+
"grad_norm": 2.4006614685058594,
|
280 |
+
"learning_rate": 7.649596321166024e-06,
|
281 |
+
"loss": 2.8198,
|
282 |
+
"step": 3900
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.12254714235382423,
|
286 |
+
"grad_norm": 2.271172285079956,
|
287 |
+
"learning_rate": 7.500000000000001e-06,
|
288 |
+
"loss": 2.8186,
|
289 |
+
"step": 4000
|
290 |
+
}
|
291 |
+
],
|
292 |
+
"logging_steps": 100,
|
293 |
+
"max_steps": 10000,
|
294 |
+
"num_input_tokens_seen": 0,
|
295 |
+
"num_train_epochs": 1,
|
296 |
+
"save_steps": 500,
|
297 |
+
"stateful_callbacks": {
|
298 |
+
"TrainerControl": {
|
299 |
+
"args": {
|
300 |
+
"should_epoch_stop": false,
|
301 |
+
"should_evaluate": false,
|
302 |
+
"should_log": false,
|
303 |
+
"should_save": true,
|
304 |
+
"should_training_stop": false
|
305 |
+
},
|
306 |
+
"attributes": {}
|
307 |
+
}
|
308 |
+
},
|
309 |
+
"total_flos": 3.82656621576192e+17,
|
310 |
+
"train_batch_size": 16,
|
311 |
+
"trial_name": null,
|
312 |
+
"trial_params": null
|
313 |
+
}
|
checkpoint-4000/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:343fd12cd57f4f5c4c4d092162b0058e69041ee73092ab79cda1f02d0d79adab
|
3 |
+
size 5368
|