C10X commited on
Commit
75b1874
·
verified ·
1 Parent(s): bfbef3b

Upload 16 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
added_tokens.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</think>": 151668,
3
+ "</tool_call>": 151658,
4
+ "</tool_response>": 151666,
5
+ "<think>": 151667,
6
+ "<tool_call>": 151657,
7
+ "<tool_response>": 151665,
8
+ "<|box_end|>": 151649,
9
+ "<|box_start|>": 151648,
10
+ "<|endoftext|>": 151643,
11
+ "<|file_sep|>": 151664,
12
+ "<|fim_middle|>": 151660,
13
+ "<|fim_pad|>": 151662,
14
+ "<|fim_prefix|>": 151659,
15
+ "<|fim_suffix|>": 151661,
16
+ "<|im_end|>": 151645,
17
+ "<|im_start|>": 151644,
18
+ "<|image_pad|>": 151655,
19
+ "<|object_ref_end|>": 151647,
20
+ "<|object_ref_start|>": 151646,
21
+ "<|quad_end|>": 151651,
22
+ "<|quad_start|>": 151650,
23
+ "<|repo_name|>": 151663,
24
+ "<|video_pad|>": 151656,
25
+ "<|vision_end|>": 151653,
26
+ "<|vision_pad|>": 151654,
27
+ "<|vision_start|>": 151652
28
+ }
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "CustomQwen3MoeForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "decoder_sparse_step": 1,
8
+ "eos_token_id": 151645,
9
+ "hidden_act": "silu",
10
+ "hidden_size": 256,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 682,
13
+ "max_position_embeddings": 512,
14
+ "max_window_layers": 28,
15
+ "mlp_only_layers": [],
16
+ "model_type": "qwen3_moe",
17
+ "moe_intermediate_size": 128,
18
+ "norm_topk_prob": false,
19
+ "num_attention_heads": 16,
20
+ "num_experts": 8,
21
+ "num_experts_per_tok": 2,
22
+ "num_hidden_layers": 6,
23
+ "num_key_value_heads": 4,
24
+ "output_router_logits": true,
25
+ "pad_token_id": 151643,
26
+ "rms_norm_eps": 1e-06,
27
+ "rope_scaling": null,
28
+ "rope_theta": 1000000.0,
29
+ "router_aux_loss_coef": 0.0,
30
+ "sliding_window": null,
31
+ "tie_word_embeddings": false,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.51.3",
34
+ "use_cache": true,
35
+ "use_sliding_window": false,
36
+ "vocab_size": 151669
37
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "eos_token_id": 151645,
4
+ "pad_token_id": 151643,
5
+ "transformers_version": "4.51.3"
6
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:499879993a561f061c7ddd30183f345959da3ed5e4beafb5aec29ffbed481ada
3
+ size 333511136
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd0745c0158c9979db8a0124ef879e7f5efea873154a5fbc019cf7d9e79f17e8
3
+ size 667144843
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fd9432b50864b1799d071a2391f13a188cc959e985675ab69fd688672db2853
3
+ size 14645
runs/May14_23-16-04_f45cd98ae7cc/events.out.tfevents.1747264565.f45cd98ae7cc.19.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7384b7d4045ca96bbaf8c99fadcdaf5a52fdf1fa271f960c0fbbe86a783bc8b
3
+ size 4184
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b21c5349d5e7d02de630ebc1cb53ade1d9c6079eeb8594d223bb786011a0428b
3
+ size 1383
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27548c755d5ffa5822a681a1ce66ba89bcd92f63551870adf49a24d66b6a554c
3
+ size 1465
special_tokens_map.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|object_ref_start|>",
6
+ "<|object_ref_end|>",
7
+ "<|box_start|>",
8
+ "<|box_end|>",
9
+ "<|quad_start|>",
10
+ "<|quad_end|>",
11
+ "<|vision_start|>",
12
+ "<|vision_end|>",
13
+ "<|vision_pad|>",
14
+ "<|image_pad|>",
15
+ "<|video_pad|>"
16
+ ],
17
+ "eos_token": {
18
+ "content": "<|im_end|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ "pad_token": {
25
+ "content": "<|endoftext|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:352a863cd2761388ccc58f1432467ba6a1037bf12df9069889b142fa246471f6
3
+ size 11422752
tokenizer_config.json ADDED
@@ -0,0 +1,240 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_prefix_space": false,
4
+ "added_tokens_decoder": {
5
+ "151643": {
6
+ "content": "<|endoftext|>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "151644": {
14
+ "content": "<|im_start|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "151645": {
22
+ "content": "<|im_end|>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "151646": {
30
+ "content": "<|object_ref_start|>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ },
37
+ "151647": {
38
+ "content": "<|object_ref_end|>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false,
43
+ "special": true
44
+ },
45
+ "151648": {
46
+ "content": "<|box_start|>",
47
+ "lstrip": false,
48
+ "normalized": false,
49
+ "rstrip": false,
50
+ "single_word": false,
51
+ "special": true
52
+ },
53
+ "151649": {
54
+ "content": "<|box_end|>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false,
59
+ "special": true
60
+ },
61
+ "151650": {
62
+ "content": "<|quad_start|>",
63
+ "lstrip": false,
64
+ "normalized": false,
65
+ "rstrip": false,
66
+ "single_word": false,
67
+ "special": true
68
+ },
69
+ "151651": {
70
+ "content": "<|quad_end|>",
71
+ "lstrip": false,
72
+ "normalized": false,
73
+ "rstrip": false,
74
+ "single_word": false,
75
+ "special": true
76
+ },
77
+ "151652": {
78
+ "content": "<|vision_start|>",
79
+ "lstrip": false,
80
+ "normalized": false,
81
+ "rstrip": false,
82
+ "single_word": false,
83
+ "special": true
84
+ },
85
+ "151653": {
86
+ "content": "<|vision_end|>",
87
+ "lstrip": false,
88
+ "normalized": false,
89
+ "rstrip": false,
90
+ "single_word": false,
91
+ "special": true
92
+ },
93
+ "151654": {
94
+ "content": "<|vision_pad|>",
95
+ "lstrip": false,
96
+ "normalized": false,
97
+ "rstrip": false,
98
+ "single_word": false,
99
+ "special": true
100
+ },
101
+ "151655": {
102
+ "content": "<|image_pad|>",
103
+ "lstrip": false,
104
+ "normalized": false,
105
+ "rstrip": false,
106
+ "single_word": false,
107
+ "special": true
108
+ },
109
+ "151656": {
110
+ "content": "<|video_pad|>",
111
+ "lstrip": false,
112
+ "normalized": false,
113
+ "rstrip": false,
114
+ "single_word": false,
115
+ "special": true
116
+ },
117
+ "151657": {
118
+ "content": "<tool_call>",
119
+ "lstrip": false,
120
+ "normalized": false,
121
+ "rstrip": false,
122
+ "single_word": false,
123
+ "special": false
124
+ },
125
+ "151658": {
126
+ "content": "</tool_call>",
127
+ "lstrip": false,
128
+ "normalized": false,
129
+ "rstrip": false,
130
+ "single_word": false,
131
+ "special": false
132
+ },
133
+ "151659": {
134
+ "content": "<|fim_prefix|>",
135
+ "lstrip": false,
136
+ "normalized": false,
137
+ "rstrip": false,
138
+ "single_word": false,
139
+ "special": false
140
+ },
141
+ "151660": {
142
+ "content": "<|fim_middle|>",
143
+ "lstrip": false,
144
+ "normalized": false,
145
+ "rstrip": false,
146
+ "single_word": false,
147
+ "special": false
148
+ },
149
+ "151661": {
150
+ "content": "<|fim_suffix|>",
151
+ "lstrip": false,
152
+ "normalized": false,
153
+ "rstrip": false,
154
+ "single_word": false,
155
+ "special": false
156
+ },
157
+ "151662": {
158
+ "content": "<|fim_pad|>",
159
+ "lstrip": false,
160
+ "normalized": false,
161
+ "rstrip": false,
162
+ "single_word": false,
163
+ "special": false
164
+ },
165
+ "151663": {
166
+ "content": "<|repo_name|>",
167
+ "lstrip": false,
168
+ "normalized": false,
169
+ "rstrip": false,
170
+ "single_word": false,
171
+ "special": false
172
+ },
173
+ "151664": {
174
+ "content": "<|file_sep|>",
175
+ "lstrip": false,
176
+ "normalized": false,
177
+ "rstrip": false,
178
+ "single_word": false,
179
+ "special": false
180
+ },
181
+ "151665": {
182
+ "content": "<tool_response>",
183
+ "lstrip": false,
184
+ "normalized": false,
185
+ "rstrip": false,
186
+ "single_word": false,
187
+ "special": false
188
+ },
189
+ "151666": {
190
+ "content": "</tool_response>",
191
+ "lstrip": false,
192
+ "normalized": false,
193
+ "rstrip": false,
194
+ "single_word": false,
195
+ "special": false
196
+ },
197
+ "151667": {
198
+ "content": "<think>",
199
+ "lstrip": false,
200
+ "normalized": false,
201
+ "rstrip": false,
202
+ "single_word": false,
203
+ "special": false
204
+ },
205
+ "151668": {
206
+ "content": "</think>",
207
+ "lstrip": false,
208
+ "normalized": false,
209
+ "rstrip": false,
210
+ "single_word": false,
211
+ "special": false
212
+ }
213
+ },
214
+ "additional_special_tokens": [
215
+ "<|im_start|>",
216
+ "<|im_end|>",
217
+ "<|object_ref_start|>",
218
+ "<|object_ref_end|>",
219
+ "<|box_start|>",
220
+ "<|box_end|>",
221
+ "<|quad_start|>",
222
+ "<|quad_end|>",
223
+ "<|vision_start|>",
224
+ "<|vision_end|>",
225
+ "<|vision_pad|>",
226
+ "<|image_pad|>",
227
+ "<|video_pad|>"
228
+ ],
229
+ "bos_token": null,
230
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0].role == 'system' %}\n {{- messages[0].content + '\\n\\n' }}\n {%- endif %}\n {{- \"# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n {%- if messages[0].role == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0].content + '<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}\n{%- for message in messages[::-1] %}\n {%- set index = (messages|length - 1) - loop.index0 %}\n {%- if ns.multi_step_tool and message.role == \"user\" and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}\n {%- set ns.multi_step_tool = false %}\n {%- set ns.last_query_index = index %}\n {%- endif %}\n{%- endfor %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {%- set content = message.content %}\n {%- set reasoning_content = '' %}\n {%- if message.reasoning_content is defined and message.reasoning_content is not none %}\n {%- set reasoning_content = message.reasoning_content %}\n {%- else %}\n {%- if '</think>' in message.content %}\n {%- set content = message.content.split('</think>')[-1].lstrip('\\n') %}\n {%- set reasoning_content = message.content.split('</think>')[0].rstrip('\\n').split('<think>')[-1].lstrip('\\n') %}\n {%- endif %}\n {%- endif %}\n {%- if loop.index0 > ns.last_query_index %}\n {%- if loop.last or (not loop.last and reasoning_content) %}\n {{- '<|im_start|>' + message.role + '\\n<think>\\n' + reasoning_content.strip('\\n') + '\\n</think>\\n\\n' + content.lstrip('\\n') }}\n {%- else %}\n {{- '<|im_start|>' + message.role + '\\n' + content }}\n {%- endif %}\n {%- else %}\n {{- '<|im_start|>' + message.role + '\\n' + content }}\n {%- endif %}\n {%- if message.tool_calls %}\n {%- for tool_call in message.tool_calls %}\n {%- if (loop.first and content) or (not loop.first) %}\n {{- '\\n' }}\n {%- endif %}\n {%- if tool_call.function %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {%- if tool_call.arguments is string %}\n {{- tool_call.arguments }}\n {%- else %}\n {{- tool_call.arguments | tojson }}\n {%- endif %}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {%- endif %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if loop.first or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- message.content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n {%- if enable_thinking is defined and enable_thinking is false %}\n {{- '<think>\\n\\n</think>\\n\\n' }}\n {%- endif %}\n{%- endif %}",
231
+ "clean_up_tokenization_spaces": false,
232
+ "eos_token": "<|im_end|>",
233
+ "errors": "replace",
234
+ "extra_special_tokens": {},
235
+ "model_max_length": 131072,
236
+ "pad_token": "<|endoftext|>",
237
+ "split_special_tokens": false,
238
+ "tokenizer_class": "Qwen2Tokenizer",
239
+ "unk_token": null
240
+ }
trainer_state.json ADDED
@@ -0,0 +1,277 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 2500,
3
+ "best_metric": 3.794926643371582,
4
+ "best_model_checkpoint": "./qwen3moe_tinystories_sft_global_balance/checkpoint-2500",
5
+ "epoch": 0.9997473684210526,
6
+ "eval_steps": 500,
7
+ "global_step": 2968,
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.03368421052631579,
14
+ "grad_norm": 492097.03125,
15
+ "learning_rate": 1.6666666666666667e-05,
16
+ "loss": 11.6957,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 0.06736842105263158,
21
+ "grad_norm": 526197.25,
22
+ "learning_rate": 3.35016835016835e-05,
23
+ "loss": 10.5167,
24
+ "step": 200
25
+ },
26
+ {
27
+ "epoch": 0.10105263157894737,
28
+ "grad_norm": 486177.34375,
29
+ "learning_rate": 4.999993082936328e-05,
30
+ "loss": 8.9635,
31
+ "step": 300
32
+ },
33
+ {
34
+ "epoch": 0.13473684210526315,
35
+ "grad_norm": 394882.59375,
36
+ "learning_rate": 4.982030277845304e-05,
37
+ "loss": 7.2522,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 0.16842105263157894,
42
+ "grad_norm": 279043.53125,
43
+ "learning_rate": 4.9297703006544226e-05,
44
+ "loss": 6.0898,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.16842105263157894,
49
+ "eval_loss": 5.723438262939453,
50
+ "eval_runtime": 135.9602,
51
+ "eval_samples_per_second": 36.775,
52
+ "eval_steps_per_second": 2.302,
53
+ "step": 500
54
+ },
55
+ {
56
+ "epoch": 0.20210526315789473,
57
+ "grad_norm": 258667.03125,
58
+ "learning_rate": 4.843935289787076e-05,
59
+ "loss": 5.4572,
60
+ "step": 600
61
+ },
62
+ {
63
+ "epoch": 0.23578947368421052,
64
+ "grad_norm": 308293.84375,
65
+ "learning_rate": 4.725711329944238e-05,
66
+ "loss": 5.051,
67
+ "step": 700
68
+ },
69
+ {
70
+ "epoch": 0.2694736842105263,
71
+ "grad_norm": 280231.3125,
72
+ "learning_rate": 4.5767320625577836e-05,
73
+ "loss": 4.7862,
74
+ "step": 800
75
+ },
76
+ {
77
+ "epoch": 0.3031578947368421,
78
+ "grad_norm": 383468.1875,
79
+ "learning_rate": 4.399056111818752e-05,
80
+ "loss": 4.5928,
81
+ "step": 900
82
+ },
83
+ {
84
+ "epoch": 0.3368421052631579,
85
+ "grad_norm": 332836.21875,
86
+ "learning_rate": 4.19513863821205e-05,
87
+ "loss": 4.4388,
88
+ "step": 1000
89
+ },
90
+ {
91
+ "epoch": 0.3368421052631579,
92
+ "eval_loss": 4.37507438659668,
93
+ "eval_runtime": 134.9276,
94
+ "eval_samples_per_second": 37.057,
95
+ "eval_steps_per_second": 2.32,
96
+ "step": 1000
97
+ },
98
+ {
99
+ "epoch": 0.3705263157894737,
100
+ "grad_norm": 378902.09375,
101
+ "learning_rate": 3.967797412636315e-05,
102
+ "loss": 4.3341,
103
+ "step": 1100
104
+ },
105
+ {
106
+ "epoch": 0.40421052631578946,
107
+ "grad_norm": 377249.0,
108
+ "learning_rate": 3.7201738799033065e-05,
109
+ "loss": 4.2245,
110
+ "step": 1200
111
+ },
112
+ {
113
+ "epoch": 0.4378947368421053,
114
+ "grad_norm": 357383.0625,
115
+ "learning_rate": 3.4556897496488504e-05,
116
+ "loss": 4.1398,
117
+ "step": 1300
118
+ },
119
+ {
120
+ "epoch": 0.47157894736842104,
121
+ "grad_norm": 475150.9375,
122
+ "learning_rate": 3.177999714490516e-05,
123
+ "loss": 4.1092,
124
+ "step": 1400
125
+ },
126
+ {
127
+ "epoch": 0.5052631578947369,
128
+ "grad_norm": 475546.25,
129
+ "learning_rate": 2.890940948781592e-05,
130
+ "loss": 4.0588,
131
+ "step": 1500
132
+ },
133
+ {
134
+ "epoch": 0.5052631578947369,
135
+ "eval_loss": 4.014543533325195,
136
+ "eval_runtime": 136.1182,
137
+ "eval_samples_per_second": 36.733,
138
+ "eval_steps_per_second": 2.299,
139
+ "step": 1500
140
+ },
141
+ {
142
+ "epoch": 0.5389473684210526,
143
+ "grad_norm": 510377.0625,
144
+ "learning_rate": 2.5984800857973353e-05,
145
+ "loss": 3.9888,
146
+ "step": 1600
147
+ },
148
+ {
149
+ "epoch": 0.5726315789473684,
150
+ "grad_norm": 433969.625,
151
+ "learning_rate": 2.3046584060329007e-05,
152
+ "loss": 3.9544,
153
+ "step": 1700
154
+ },
155
+ {
156
+ "epoch": 0.6063157894736843,
157
+ "grad_norm": 430580.25,
158
+ "learning_rate": 2.0135359940116327e-05,
159
+ "loss": 3.9086,
160
+ "step": 1800
161
+ },
162
+ {
163
+ "epoch": 0.64,
164
+ "grad_norm": 443697.03125,
165
+ "learning_rate": 1.729135635255667e-05,
166
+ "loss": 3.9023,
167
+ "step": 1900
168
+ },
169
+ {
170
+ "epoch": 0.6736842105263158,
171
+ "grad_norm": 382403.0,
172
+ "learning_rate": 1.455387228661314e-05,
173
+ "loss": 3.8641,
174
+ "step": 2000
175
+ },
176
+ {
177
+ "epoch": 0.6736842105263158,
178
+ "eval_loss": 3.8544445037841797,
179
+ "eval_runtime": 134.3764,
180
+ "eval_samples_per_second": 37.209,
181
+ "eval_steps_per_second": 2.329,
182
+ "step": 2000
183
+ },
184
+ {
185
+ "epoch": 0.7073684210526315,
186
+ "grad_norm": 408598.40625,
187
+ "learning_rate": 1.1960734823997168e-05,
188
+ "loss": 3.8511,
189
+ "step": 2100
190
+ },
191
+ {
192
+ "epoch": 0.7410526315789474,
193
+ "grad_norm": 461325.09375,
194
+ "learning_rate": 9.547776437272746e-06,
195
+ "loss": 3.832,
196
+ "step": 2200
197
+ },
198
+ {
199
+ "epoch": 0.7747368421052632,
200
+ "grad_norm": 427811.28125,
201
+ "learning_rate": 7.348339849853858e-06,
202
+ "loss": 3.8217,
203
+ "step": 2300
204
+ },
205
+ {
206
+ "epoch": 0.8084210526315789,
207
+ "grad_norm": 465533.09375,
208
+ "learning_rate": 5.39281729983474e-06,
209
+ "loss": 3.81,
210
+ "step": 2400
211
+ },
212
+ {
213
+ "epoch": 0.8421052631578947,
214
+ "grad_norm": 420344.15625,
215
+ "learning_rate": 3.7082305741943213e-06,
216
+ "loss": 3.7949,
217
+ "step": 2500
218
+ },
219
+ {
220
+ "epoch": 0.8421052631578947,
221
+ "eval_loss": 3.794926643371582,
222
+ "eval_runtime": 133.9027,
223
+ "eval_samples_per_second": 37.341,
224
+ "eval_steps_per_second": 2.338,
225
+ "step": 2500
226
+ },
227
+ {
228
+ "epoch": 0.8757894736842106,
229
+ "grad_norm": 433467.875,
230
+ "learning_rate": 2.3178576165427735e-06,
231
+ "loss": 3.7939,
232
+ "step": 2600
233
+ },
234
+ {
235
+ "epoch": 0.9094736842105263,
236
+ "grad_norm": 398190.46875,
237
+ "learning_rate": 1.2409108680163734e-06,
238
+ "loss": 3.7832,
239
+ "step": 2700
240
+ },
241
+ {
242
+ "epoch": 0.9431578947368421,
243
+ "grad_norm": 369212.90625,
244
+ "learning_rate": 4.922717860680298e-07,
245
+ "loss": 3.7862,
246
+ "step": 2800
247
+ },
248
+ {
249
+ "epoch": 0.9768421052631578,
250
+ "grad_norm": 457141.46875,
251
+ "learning_rate": 8.228520962394182e-08,
252
+ "loss": 3.8013,
253
+ "step": 2900
254
+ }
255
+ ],
256
+ "logging_steps": 100,
257
+ "max_steps": 2968,
258
+ "num_input_tokens_seen": 0,
259
+ "num_train_epochs": 1,
260
+ "save_steps": 500,
261
+ "stateful_callbacks": {
262
+ "TrainerControl": {
263
+ "args": {
264
+ "should_epoch_stop": false,
265
+ "should_evaluate": false,
266
+ "should_log": false,
267
+ "should_save": true,
268
+ "should_training_stop": true
269
+ },
270
+ "attributes": {}
271
+ }
272
+ },
273
+ "total_flos": 9101710322884608.0,
274
+ "train_batch_size": 8,
275
+ "trial_name": null,
276
+ "trial_params": null
277
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef65ea9fe944bba238b832d817df04a15277135e09a4c3334aa5f6b1c5945388
3
+ size 5777
vocab.json ADDED
The diff for this file is too large to render. See raw diff