derko83 commited on
Commit
6a86618
·
verified ·
1 Parent(s): 6fc6295

Upload folder using huggingface_hub

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,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3ForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "eos_token_id": 151643,
9
+ "head_dim": 128,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 1024,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "max_position_embeddings": 32768,
15
+ "max_window_layers": 28,
16
+ "model_type": "qwen3",
17
+ "num_attention_heads": 16,
18
+ "num_hidden_layers": 28,
19
+ "num_key_value_heads": 8,
20
+ "rms_norm_eps": 1e-06,
21
+ "rope_scaling": null,
22
+ "rope_theta": 1000000,
23
+ "sliding_window": null,
24
+ "tie_word_embeddings": true,
25
+ "torch_dtype": "float32",
26
+ "transformers_version": "4.51.3",
27
+ "use_cache": true,
28
+ "use_sliding_window": false,
29
+ "vocab_size": 151936
30
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "eos_token_id": 151643,
4
+ "max_new_tokens": 2048,
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:30692ceb235c4429caf85385bb5458694b351ae55100bbfb3f50a74566d5f9f1
3
+ size 2384234968
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd62ddddee6c23a6650a91d1b2d97b6552130af70cc0c03b742eeb84fd96ec01
3
+ size 4768663315
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:250560ab3d528161ab3659b120def6e4a9ab4b457e3399603bbcfa40db3efc90
3
+ size 14645
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60dc60940088350df2476daec612777256353a222718d12ad1d77c0a8edad709
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": "<|endoftext|>",
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:aeb13307a71acd8fe81861d94ad54ab689df773318809eed3cbe794b4492dae4
3
+ size 11422654
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": "<|endoftext|>",
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,469 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 1493,
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.033489618218352314,
14
+ "grad_norm": 90.51985931396484,
15
+ "learning_rate": 4.835900870730074e-06,
16
+ "logits/chosen": NaN,
17
+ "logits/rejected": NaN,
18
+ "logps/chosen": -179.42369079589844,
19
+ "logps/rejected": -227.34011840820312,
20
+ "loss": 0.6272,
21
+ "rewards/accuracies": 0.4987500011920929,
22
+ "rewards/chosen": -0.41917884349823,
23
+ "rewards/margins": 0.4206826388835907,
24
+ "rewards/rejected": -0.8398614525794983,
25
+ "step": 50
26
+ },
27
+ {
28
+ "epoch": 0.06697923643670463,
29
+ "grad_norm": 122.67408752441406,
30
+ "learning_rate": 4.668452779638312e-06,
31
+ "logits/chosen": NaN,
32
+ "logits/rejected": NaN,
33
+ "logps/chosen": -189.38446044921875,
34
+ "logps/rejected": -243.21636962890625,
35
+ "loss": 0.6008,
36
+ "rewards/accuracies": 0.5625,
37
+ "rewards/chosen": -0.9909499287605286,
38
+ "rewards/margins": 0.8664093613624573,
39
+ "rewards/rejected": -1.8573591709136963,
40
+ "step": 100
41
+ },
42
+ {
43
+ "epoch": 0.10046885465505694,
44
+ "grad_norm": 69.09490203857422,
45
+ "learning_rate": 4.501004688546551e-06,
46
+ "logits/chosen": NaN,
47
+ "logits/rejected": NaN,
48
+ "logps/chosen": -172.51051330566406,
49
+ "logps/rejected": -237.64398193359375,
50
+ "loss": 0.5445,
51
+ "rewards/accuracies": 0.5912500023841858,
52
+ "rewards/chosen": -0.773537814617157,
53
+ "rewards/margins": 1.0922305583953857,
54
+ "rewards/rejected": -1.8657684326171875,
55
+ "step": 150
56
+ },
57
+ {
58
+ "epoch": 0.13395847287340926,
59
+ "grad_norm": 92.90730285644531,
60
+ "learning_rate": 4.333556597454789e-06,
61
+ "logits/chosen": NaN,
62
+ "logits/rejected": NaN,
63
+ "logps/chosen": -185.27850341796875,
64
+ "logps/rejected": -240.15431213378906,
65
+ "loss": 0.5958,
66
+ "rewards/accuracies": 0.581250011920929,
67
+ "rewards/chosen": -0.8680741190910339,
68
+ "rewards/margins": 1.0129902362823486,
69
+ "rewards/rejected": -1.8810642957687378,
70
+ "step": 200
71
+ },
72
+ {
73
+ "epoch": 0.16744809109176156,
74
+ "grad_norm": 41.30106735229492,
75
+ "learning_rate": 4.166108506363028e-06,
76
+ "logits/chosen": NaN,
77
+ "logits/rejected": NaN,
78
+ "logps/chosen": -185.0808563232422,
79
+ "logps/rejected": -243.77664184570312,
80
+ "loss": 0.5232,
81
+ "rewards/accuracies": 0.5950000286102295,
82
+ "rewards/chosen": -0.6241927742958069,
83
+ "rewards/margins": 1.5013166666030884,
84
+ "rewards/rejected": -2.125509262084961,
85
+ "step": 250
86
+ },
87
+ {
88
+ "epoch": 0.20093770931011387,
89
+ "grad_norm": 88.7403793334961,
90
+ "learning_rate": 3.998660415271266e-06,
91
+ "logits/chosen": NaN,
92
+ "logits/rejected": NaN,
93
+ "logps/chosen": -184.0787353515625,
94
+ "logps/rejected": -243.6924285888672,
95
+ "loss": 0.5419,
96
+ "rewards/accuracies": 0.5950000286102295,
97
+ "rewards/chosen": -0.8100302219390869,
98
+ "rewards/margins": 1.521090030670166,
99
+ "rewards/rejected": -2.331120491027832,
100
+ "step": 300
101
+ },
102
+ {
103
+ "epoch": 0.23442732752846618,
104
+ "grad_norm": 70.95861053466797,
105
+ "learning_rate": 3.831212324179505e-06,
106
+ "logits/chosen": NaN,
107
+ "logits/rejected": -1.532848834991455,
108
+ "logps/chosen": -182.43124389648438,
109
+ "logps/rejected": -248.5233917236328,
110
+ "loss": 0.5533,
111
+ "rewards/accuracies": 0.5824999809265137,
112
+ "rewards/chosen": -1.069503903388977,
113
+ "rewards/margins": 1.6424156427383423,
114
+ "rewards/rejected": -2.7119195461273193,
115
+ "step": 350
116
+ },
117
+ {
118
+ "epoch": 0.2679169457468185,
119
+ "grad_norm": 71.3283462524414,
120
+ "learning_rate": 3.663764233087743e-06,
121
+ "logits/chosen": NaN,
122
+ "logits/rejected": NaN,
123
+ "logps/chosen": -180.90724182128906,
124
+ "logps/rejected": -253.06655883789062,
125
+ "loss": 0.5131,
126
+ "rewards/accuracies": 0.6449999809265137,
127
+ "rewards/chosen": -0.9932506084442139,
128
+ "rewards/margins": 2.001958131790161,
129
+ "rewards/rejected": -2.995208740234375,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 0.3014065639651708,
134
+ "grad_norm": 52.41379928588867,
135
+ "learning_rate": 3.496316141995982e-06,
136
+ "logits/chosen": NaN,
137
+ "logits/rejected": NaN,
138
+ "logps/chosen": -187.8169708251953,
139
+ "logps/rejected": -250.7092742919922,
140
+ "loss": 0.574,
141
+ "rewards/accuracies": 0.6087499856948853,
142
+ "rewards/chosen": -0.9525413513183594,
143
+ "rewards/margins": 1.8562077283859253,
144
+ "rewards/rejected": -2.808749198913574,
145
+ "step": 450
146
+ },
147
+ {
148
+ "epoch": 0.33489618218352313,
149
+ "grad_norm": 86.05554962158203,
150
+ "learning_rate": 3.32886805090422e-06,
151
+ "logits/chosen": NaN,
152
+ "logits/rejected": NaN,
153
+ "logps/chosen": -184.93011474609375,
154
+ "logps/rejected": -245.39427185058594,
155
+ "loss": 0.545,
156
+ "rewards/accuracies": 0.6200000047683716,
157
+ "rewards/chosen": -0.9822418093681335,
158
+ "rewards/margins": 1.7062770128250122,
159
+ "rewards/rejected": -2.68851900100708,
160
+ "step": 500
161
+ },
162
+ {
163
+ "epoch": 0.3683858004018754,
164
+ "grad_norm": 24.4642391204834,
165
+ "learning_rate": 3.1614199598124583e-06,
166
+ "logits/chosen": NaN,
167
+ "logits/rejected": NaN,
168
+ "logps/chosen": -177.97482299804688,
169
+ "logps/rejected": -253.92169189453125,
170
+ "loss": 0.5249,
171
+ "rewards/accuracies": 0.6150000095367432,
172
+ "rewards/chosen": -0.7585690021514893,
173
+ "rewards/margins": 2.060235023498535,
174
+ "rewards/rejected": -2.8188037872314453,
175
+ "step": 550
176
+ },
177
+ {
178
+ "epoch": 0.40187541862022774,
179
+ "grad_norm": 28.010169982910156,
180
+ "learning_rate": 2.993971868720697e-06,
181
+ "logits/chosen": NaN,
182
+ "logits/rejected": NaN,
183
+ "logps/chosen": -187.2054443359375,
184
+ "logps/rejected": -262.4362487792969,
185
+ "loss": 0.531,
186
+ "rewards/accuracies": 0.6212499737739563,
187
+ "rewards/chosen": -1.0288543701171875,
188
+ "rewards/margins": 2.0049030780792236,
189
+ "rewards/rejected": -3.033757209777832,
190
+ "step": 600
191
+ },
192
+ {
193
+ "epoch": 0.43536503683858,
194
+ "grad_norm": 88.00039672851562,
195
+ "learning_rate": 2.8265237776289352e-06,
196
+ "logits/chosen": NaN,
197
+ "logits/rejected": NaN,
198
+ "logps/chosen": -199.0946044921875,
199
+ "logps/rejected": -257.4405212402344,
200
+ "loss": 0.506,
201
+ "rewards/accuracies": 0.6175000071525574,
202
+ "rewards/chosen": -0.9385867118835449,
203
+ "rewards/margins": 1.9474469423294067,
204
+ "rewards/rejected": -2.8860340118408203,
205
+ "step": 650
206
+ },
207
+ {
208
+ "epoch": 0.46885465505693236,
209
+ "grad_norm": 76.95013427734375,
210
+ "learning_rate": 2.6590756865371737e-06,
211
+ "logits/chosen": NaN,
212
+ "logits/rejected": NaN,
213
+ "logps/chosen": -188.61651611328125,
214
+ "logps/rejected": -255.59820556640625,
215
+ "loss": 0.4841,
216
+ "rewards/accuracies": 0.6287500262260437,
217
+ "rewards/chosen": -0.9187755584716797,
218
+ "rewards/margins": 2.190307378768921,
219
+ "rewards/rejected": -3.1090831756591797,
220
+ "step": 700
221
+ },
222
+ {
223
+ "epoch": 0.5023442732752846,
224
+ "grad_norm": 37.079219818115234,
225
+ "learning_rate": 2.491627595445412e-06,
226
+ "logits/chosen": NaN,
227
+ "logits/rejected": NaN,
228
+ "logps/chosen": -182.82937622070312,
229
+ "logps/rejected": -253.3831787109375,
230
+ "loss": 0.5061,
231
+ "rewards/accuracies": 0.6324999928474426,
232
+ "rewards/chosen": -0.8986356854438782,
233
+ "rewards/margins": 2.1806037425994873,
234
+ "rewards/rejected": -3.0792391300201416,
235
+ "step": 750
236
+ },
237
+ {
238
+ "epoch": 0.535833891493637,
239
+ "grad_norm": 31.89027976989746,
240
+ "learning_rate": 2.3241795043536505e-06,
241
+ "logits/chosen": NaN,
242
+ "logits/rejected": NaN,
243
+ "logps/chosen": -172.42593383789062,
244
+ "logps/rejected": -251.63729858398438,
245
+ "loss": 0.4358,
246
+ "rewards/accuracies": 0.6625000238418579,
247
+ "rewards/chosen": -0.7772516012191772,
248
+ "rewards/margins": 2.509305953979492,
249
+ "rewards/rejected": -3.286557674407959,
250
+ "step": 800
251
+ },
252
+ {
253
+ "epoch": 0.5693235097119893,
254
+ "grad_norm": 49.636253356933594,
255
+ "learning_rate": 2.156731413261889e-06,
256
+ "logits/chosen": NaN,
257
+ "logits/rejected": NaN,
258
+ "logps/chosen": -177.4981231689453,
259
+ "logps/rejected": -252.06234741210938,
260
+ "loss": 0.4752,
261
+ "rewards/accuracies": 0.6262500286102295,
262
+ "rewards/chosen": -0.8873915672302246,
263
+ "rewards/margins": 2.302338123321533,
264
+ "rewards/rejected": -3.1897289752960205,
265
+ "step": 850
266
+ },
267
+ {
268
+ "epoch": 0.6028131279303416,
269
+ "grad_norm": 42.22234344482422,
270
+ "learning_rate": 1.9892833221701274e-06,
271
+ "logits/chosen": NaN,
272
+ "logits/rejected": NaN,
273
+ "logps/chosen": -185.84585571289062,
274
+ "logps/rejected": -251.1869354248047,
275
+ "loss": 0.5284,
276
+ "rewards/accuracies": 0.6200000047683716,
277
+ "rewards/chosen": -0.9590955972671509,
278
+ "rewards/margins": 2.267340660095215,
279
+ "rewards/rejected": -3.2264363765716553,
280
+ "step": 900
281
+ },
282
+ {
283
+ "epoch": 0.6363027461486939,
284
+ "grad_norm": 60.56090545654297,
285
+ "learning_rate": 1.8218352310783657e-06,
286
+ "logits/chosen": NaN,
287
+ "logits/rejected": NaN,
288
+ "logps/chosen": -182.73912048339844,
289
+ "logps/rejected": -245.673828125,
290
+ "loss": 0.4587,
291
+ "rewards/accuracies": 0.6625000238418579,
292
+ "rewards/chosen": -0.6209310293197632,
293
+ "rewards/margins": 2.47007155418396,
294
+ "rewards/rejected": -3.0910024642944336,
295
+ "step": 950
296
+ },
297
+ {
298
+ "epoch": 0.6697923643670463,
299
+ "grad_norm": 47.82743453979492,
300
+ "learning_rate": 1.6543871399866043e-06,
301
+ "logits/chosen": NaN,
302
+ "logits/rejected": NaN,
303
+ "logps/chosen": -170.28599548339844,
304
+ "logps/rejected": -247.58837890625,
305
+ "loss": 0.4471,
306
+ "rewards/accuracies": 0.6424999833106995,
307
+ "rewards/chosen": -0.5806804895401001,
308
+ "rewards/margins": 2.4890031814575195,
309
+ "rewards/rejected": -3.06968355178833,
310
+ "step": 1000
311
+ },
312
+ {
313
+ "epoch": 0.7032819825853985,
314
+ "grad_norm": 33.236053466796875,
315
+ "learning_rate": 1.4869390488948425e-06,
316
+ "logits/chosen": NaN,
317
+ "logits/rejected": NaN,
318
+ "logps/chosen": -173.26913452148438,
319
+ "logps/rejected": -257.9516906738281,
320
+ "loss": 0.4204,
321
+ "rewards/accuracies": 0.6825000047683716,
322
+ "rewards/chosen": -0.4792047142982483,
323
+ "rewards/margins": 2.651007890701294,
324
+ "rewards/rejected": -3.1302123069763184,
325
+ "step": 1050
326
+ },
327
+ {
328
+ "epoch": 0.7367716008037508,
329
+ "grad_norm": 39.73881530761719,
330
+ "learning_rate": 1.3194909578030812e-06,
331
+ "logits/chosen": NaN,
332
+ "logits/rejected": NaN,
333
+ "logps/chosen": -161.0741424560547,
334
+ "logps/rejected": -243.2300262451172,
335
+ "loss": 0.4058,
336
+ "rewards/accuracies": 0.6625000238418579,
337
+ "rewards/chosen": -0.5020321607589722,
338
+ "rewards/margins": 2.7466940879821777,
339
+ "rewards/rejected": -3.2487261295318604,
340
+ "step": 1100
341
+ },
342
+ {
343
+ "epoch": 0.7702612190221031,
344
+ "grad_norm": 37.8878173828125,
345
+ "learning_rate": 1.1520428667113196e-06,
346
+ "logits/chosen": NaN,
347
+ "logits/rejected": NaN,
348
+ "logps/chosen": -177.0689697265625,
349
+ "logps/rejected": -248.60035705566406,
350
+ "loss": 0.4246,
351
+ "rewards/accuracies": 0.6575000286102295,
352
+ "rewards/chosen": -0.6515741944313049,
353
+ "rewards/margins": 2.5524582862854004,
354
+ "rewards/rejected": -3.2040326595306396,
355
+ "step": 1150
356
+ },
357
+ {
358
+ "epoch": 0.8037508372404555,
359
+ "grad_norm": 54.17752456665039,
360
+ "learning_rate": 9.84594775619558e-07,
361
+ "logits/chosen": NaN,
362
+ "logits/rejected": NaN,
363
+ "logps/chosen": -184.8086395263672,
364
+ "logps/rejected": -270.40771484375,
365
+ "loss": 0.4499,
366
+ "rewards/accuracies": 0.65625,
367
+ "rewards/chosen": -0.5556185841560364,
368
+ "rewards/margins": 2.5394182205200195,
369
+ "rewards/rejected": -3.0950369834899902,
370
+ "step": 1200
371
+ },
372
+ {
373
+ "epoch": 0.8372404554588078,
374
+ "grad_norm": 37.586639404296875,
375
+ "learning_rate": 8.171466845277964e-07,
376
+ "logits/chosen": NaN,
377
+ "logits/rejected": NaN,
378
+ "logps/chosen": -182.44801330566406,
379
+ "logps/rejected": -255.1699676513672,
380
+ "loss": 0.4307,
381
+ "rewards/accuracies": 0.6537500023841858,
382
+ "rewards/chosen": -0.6889155507087708,
383
+ "rewards/margins": 2.603567123413086,
384
+ "rewards/rejected": -3.292482852935791,
385
+ "step": 1250
386
+ },
387
+ {
388
+ "epoch": 0.87073007367716,
389
+ "grad_norm": 52.10836410522461,
390
+ "learning_rate": 6.496985934360349e-07,
391
+ "logits/chosen": NaN,
392
+ "logits/rejected": NaN,
393
+ "logps/chosen": -179.65557861328125,
394
+ "logps/rejected": -245.46560668945312,
395
+ "loss": 0.4125,
396
+ "rewards/accuracies": 0.6637499928474426,
397
+ "rewards/chosen": -0.5343858003616333,
398
+ "rewards/margins": 2.6487746238708496,
399
+ "rewards/rejected": -3.1831603050231934,
400
+ "step": 1300
401
+ },
402
+ {
403
+ "epoch": 0.9042196918955124,
404
+ "grad_norm": 30.635120391845703,
405
+ "learning_rate": 4.822505023442733e-07,
406
+ "logits/chosen": NaN,
407
+ "logits/rejected": NaN,
408
+ "logps/chosen": -175.81723022460938,
409
+ "logps/rejected": -254.23828125,
410
+ "loss": 0.4304,
411
+ "rewards/accuracies": 0.6524999737739563,
412
+ "rewards/chosen": -0.5225290060043335,
413
+ "rewards/margins": 2.57625675201416,
414
+ "rewards/rejected": -3.098785638809204,
415
+ "step": 1350
416
+ },
417
+ {
418
+ "epoch": 0.9377093101138647,
419
+ "grad_norm": 49.38210678100586,
420
+ "learning_rate": 3.1480241125251174e-07,
421
+ "logits/chosen": NaN,
422
+ "logits/rejected": NaN,
423
+ "logps/chosen": -177.96661376953125,
424
+ "logps/rejected": -240.63694763183594,
425
+ "loss": 0.4411,
426
+ "rewards/accuracies": 0.65625,
427
+ "rewards/chosen": -0.4394516050815582,
428
+ "rewards/margins": 2.5781943798065186,
429
+ "rewards/rejected": -3.017645835876465,
430
+ "step": 1400
431
+ },
432
+ {
433
+ "epoch": 0.971198928332217,
434
+ "grad_norm": 57.8912467956543,
435
+ "learning_rate": 1.4735432016075018e-07,
436
+ "logits/chosen": NaN,
437
+ "logits/rejected": NaN,
438
+ "logps/chosen": -176.60218811035156,
439
+ "logps/rejected": -250.90786743164062,
440
+ "loss": 0.4317,
441
+ "rewards/accuracies": 0.6474999785423279,
442
+ "rewards/chosen": -0.514062225818634,
443
+ "rewards/margins": 2.4711577892303467,
444
+ "rewards/rejected": -2.985220432281494,
445
+ "step": 1450
446
+ }
447
+ ],
448
+ "logging_steps": 50,
449
+ "max_steps": 1493,
450
+ "num_input_tokens_seen": 0,
451
+ "num_train_epochs": 1,
452
+ "save_steps": 200,
453
+ "stateful_callbacks": {
454
+ "TrainerControl": {
455
+ "args": {
456
+ "should_epoch_stop": false,
457
+ "should_evaluate": false,
458
+ "should_log": false,
459
+ "should_save": true,
460
+ "should_training_stop": true
461
+ },
462
+ "attributes": {}
463
+ }
464
+ },
465
+ "total_flos": 0.0,
466
+ "train_batch_size": 2,
467
+ "trial_name": null,
468
+ "trial_params": null
469
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83253cf573c71383f710bb1441ffc24338aa5407acd305912e4964de7e52bda0
3
+ size 6545
vocab.json ADDED
The diff for this file is too large to render. See raw diff