chansung commited on
Commit
15a8776
·
verified ·
1 Parent(s): 53aebc5

Model save

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
README.md ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen2.5-1.5B-Instruct
3
+ library_name: transformers
4
+ model_name: Qwen2.5-1.5B-CRL-Open-R1-Code-GRPO-exp1
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - grpo
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for Qwen2.5-1.5B-CRL-Open-R1-Code-GRPO-exp1
13
+
14
+ This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="chansung/Qwen2.5-1.5B-CRL-Open-R1-Code-GRPO-exp1", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/chansung18/huggingface/runs/ekt9nbsx)
31
+
32
+
33
+ This model was trained with GRPO, a method introduced in [DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models](https://huggingface.co/papers/2402.03300).
34
+
35
+ ### Framework versions
36
+
37
+ - TRL: 0.16.0
38
+ - Transformers: 4.50.0
39
+ - Pytorch: 2.5.1
40
+ - Datasets: 3.4.1
41
+ - Tokenizers: 0.21.1
42
+
43
+ ## Citations
44
+
45
+ Cite GRPO as:
46
+
47
+ ```bibtex
48
+ @article{zhihong2024deepseekmath,
49
+ title = {{DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models}},
50
+ author = {Zhihong Shao and Peiyi Wang and Qihao Zhu and Runxin Xu and Junxiao Song and Mingchuan Zhang and Y. K. Li and Y. Wu and Daya Guo},
51
+ year = 2024,
52
+ eprint = {arXiv:2402.03300},
53
+ }
54
+
55
+ ```
56
+
57
+ Cite TRL as:
58
+
59
+ ```bibtex
60
+ @misc{vonwerra2022trl,
61
+ title = {{TRL: Transformer Reinforcement Learning}},
62
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
63
+ year = 2020,
64
+ journal = {GitHub repository},
65
+ publisher = {GitHub},
66
+ howpublished = {\url{https://github.com/huggingface/trl}}
67
+ }
68
+ ```
added_tokens.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</tool_call>": 151658,
3
+ "<tool_call>": 151657,
4
+ "<|box_end|>": 151649,
5
+ "<|box_start|>": 151648,
6
+ "<|endoftext|>": 151643,
7
+ "<|file_sep|>": 151664,
8
+ "<|fim_middle|>": 151660,
9
+ "<|fim_pad|>": 151662,
10
+ "<|fim_prefix|>": 151659,
11
+ "<|fim_suffix|>": 151661,
12
+ "<|im_end|>": 151645,
13
+ "<|im_start|>": 151644,
14
+ "<|image_pad|>": 151655,
15
+ "<|object_ref_end|>": 151647,
16
+ "<|object_ref_start|>": 151646,
17
+ "<|quad_end|>": 151651,
18
+ "<|quad_start|>": 151650,
19
+ "<|repo_name|>": 151663,
20
+ "<|video_pad|>": 151656,
21
+ "<|vision_end|>": 151653,
22
+ "<|vision_pad|>": 151654,
23
+ "<|vision_start|>": 151652
24
+ }
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_flos": 0.0,
3
+ "train_loss": 0.0,
4
+ "train_runtime": 1.2498,
5
+ "train_samples": 949,
6
+ "train_samples_per_second": 1518.669,
7
+ "train_steps_per_second": 46.408
8
+ }
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "bos_token_id": 151643,
7
+ "eos_token_id": 151645,
8
+ "hidden_act": "silu",
9
+ "hidden_size": 1536,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 8960,
12
+ "max_position_embeddings": 32768,
13
+ "max_window_layers": 21,
14
+ "model_type": "qwen2",
15
+ "num_attention_heads": 12,
16
+ "num_hidden_layers": 28,
17
+ "num_key_value_heads": 2,
18
+ "rms_norm_eps": 1e-06,
19
+ "rope_scaling": null,
20
+ "rope_theta": 1000000.0,
21
+ "sliding_window": 32768,
22
+ "tie_word_embeddings": true,
23
+ "torch_dtype": "bfloat16",
24
+ "transformers_version": "4.50.0",
25
+ "use_cache": false,
26
+ "use_sliding_window": false,
27
+ "vocab_size": 151936
28
+ }
generation_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 151645,
6
+ 151643
7
+ ],
8
+ "pad_token_id": 151643,
9
+ "repetition_penalty": 1.1,
10
+ "temperature": 0.7,
11
+ "top_k": 20,
12
+ "top_p": 0.8,
13
+ "transformers_version": "4.50.0"
14
+ }
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:8990088dabd855e218079b7a7ca8b5d77c5b5404aab1109c6f2766d0df5cfb3e
3
+ size 3554214752
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:9c5ae00e602b8860cbd784ba82a8aa14e8feecec692e7076590d014d7b7fdafa
3
+ size 11421896
tokenizer_config.json ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ },
182
+ "additional_special_tokens": [
183
+ "<|im_start|>",
184
+ "<|im_end|>",
185
+ "<|object_ref_start|>",
186
+ "<|object_ref_end|>",
187
+ "<|box_start|>",
188
+ "<|box_end|>",
189
+ "<|quad_start|>",
190
+ "<|quad_end|>",
191
+ "<|vision_start|>",
192
+ "<|vision_end|>",
193
+ "<|vision_pad|>",
194
+ "<|image_pad|>",
195
+ "<|video_pad|>"
196
+ ],
197
+ "bos_token": null,
198
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0]['role'] == 'system' %}\n {{- messages[0]['content'] }}\n {%- else %}\n {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}\n {%- endif %}\n {{- \"\\n\\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 {%- else %}\n {{- '<|im_start|>system\\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role }}\n {%- if message.content %}\n {{- '\\n' + message.content }}\n {%- endif %}\n {%- for tool_call in message.tool_calls %}\n {%- if tool_call.function is defined %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '\\n<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if (loop.index0 == 0) 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{%- endif %}\n",
199
+ "clean_up_tokenization_spaces": false,
200
+ "eos_token": "<|im_end|>",
201
+ "errors": "replace",
202
+ "extra_special_tokens": {},
203
+ "model_max_length": 131072,
204
+ "pad_token": "<|endoftext|>",
205
+ "split_special_tokens": false,
206
+ "tokenizer_class": "Qwen2Tokenizer",
207
+ "unk_token": null
208
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_flos": 0.0,
3
+ "train_loss": 0.0,
4
+ "train_runtime": 1.2498,
5
+ "train_samples": 949,
6
+ "train_samples_per_second": 1518.669,
7
+ "train_steps_per_second": 46.408
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,1543 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.6890756302521008,
6
+ "eval_steps": 500,
7
+ "global_step": 100,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "clip_ratio": 0.0,
14
+ "completion_length": 523.3984375,
15
+ "epoch": 0.01680672268907563,
16
+ "grad_norm": 0.4084254801273346,
17
+ "kl": 0.0,
18
+ "learning_rate": 3.3333333333333335e-07,
19
+ "loss": 0.0346,
20
+ "num_tokens": 149643.0,
21
+ "reward": 0.05572916800156236,
22
+ "reward_std": 0.1236814484000206,
23
+ "rewards/curriculum_aware_reward_fn": 0.024479168001562357,
24
+ "rewards/format_reward": 0.03125,
25
+ "step": 1
26
+ },
27
+ {
28
+ "clip_ratio": 0.0,
29
+ "completion_length": 446.5390625,
30
+ "epoch": 0.03361344537815126,
31
+ "grad_norm": 0.5236720442771912,
32
+ "kl": 0.0,
33
+ "learning_rate": 6.666666666666667e-07,
34
+ "loss": 0.015,
35
+ "num_tokens": 282184.0,
36
+ "reward": 0.17552083544433117,
37
+ "reward_std": 0.2788553349673748,
38
+ "rewards/curriculum_aware_reward_fn": 0.12864583544433117,
39
+ "rewards/format_reward": 0.046875,
40
+ "step": 2
41
+ },
42
+ {
43
+ "clip_ratio": 0.0,
44
+ "completion_length": 404.5625,
45
+ "epoch": 0.05042016806722689,
46
+ "grad_norm": 0.5860257744789124,
47
+ "kl": 0.00027751922607421875,
48
+ "learning_rate": 1.0000000000000002e-06,
49
+ "loss": 0.0033,
50
+ "num_tokens": 411536.0,
51
+ "reward": 0.1171875,
52
+ "reward_std": 0.2090039700269699,
53
+ "rewards/curriculum_aware_reward_fn": 0.078125,
54
+ "rewards/format_reward": 0.0390625,
55
+ "step": 3
56
+ },
57
+ {
58
+ "clip_ratio": 0.0,
59
+ "completion_length": 437.8359375,
60
+ "epoch": 0.06722689075630252,
61
+ "grad_norm": 0.5572423338890076,
62
+ "kl": 0.00028705596923828125,
63
+ "learning_rate": 1.3333333333333334e-06,
64
+ "loss": 0.0288,
65
+ "num_tokens": 543811.0,
66
+ "reward": 0.1276041716337204,
67
+ "reward_std": 0.17299909517169,
68
+ "rewards/curriculum_aware_reward_fn": 0.1041666716337204,
69
+ "rewards/format_reward": 0.0234375,
70
+ "step": 4
71
+ },
72
+ {
73
+ "clip_ratio": 0.0,
74
+ "completion_length": 484.453125,
75
+ "epoch": 0.08403361344537816,
76
+ "grad_norm": 0.5424997806549072,
77
+ "kl": 0.0003066062927246094,
78
+ "learning_rate": 1.6666666666666667e-06,
79
+ "loss": 0.0016,
80
+ "num_tokens": 674637.0,
81
+ "reward": 0.15625,
82
+ "reward_std": 0.20851250365376472,
83
+ "rewards/curriculum_aware_reward_fn": 0.109375,
84
+ "rewards/format_reward": 0.046875,
85
+ "step": 5
86
+ },
87
+ {
88
+ "clip_ratio": 0.0,
89
+ "completion_length": 468.078125,
90
+ "epoch": 0.10084033613445378,
91
+ "grad_norm": 0.47155389189720154,
92
+ "kl": 0.0003123283386230469,
93
+ "learning_rate": 2.0000000000000003e-06,
94
+ "loss": -0.0177,
95
+ "num_tokens": 810527.0,
96
+ "reward": 0.1276041716337204,
97
+ "reward_std": 0.18760672956705093,
98
+ "rewards/curriculum_aware_reward_fn": 0.0651041716337204,
99
+ "rewards/format_reward": 0.0625,
100
+ "step": 6
101
+ },
102
+ {
103
+ "clip_ratio": 0.0,
104
+ "completion_length": 500.3125,
105
+ "epoch": 0.11764705882352941,
106
+ "grad_norm": 0.5090876817703247,
107
+ "kl": 0.0004954338073730469,
108
+ "learning_rate": 2.3333333333333336e-06,
109
+ "loss": 0.0566,
110
+ "num_tokens": 947063.0,
111
+ "reward": 0.1354166679084301,
112
+ "reward_std": 0.1753537617623806,
113
+ "rewards/curriculum_aware_reward_fn": 0.07291666697710752,
114
+ "rewards/format_reward": 0.0625,
115
+ "step": 7
116
+ },
117
+ {
118
+ "clip_ratio": 0.0,
119
+ "completion_length": 483.84375,
120
+ "epoch": 0.13445378151260504,
121
+ "grad_norm": 0.6272737979888916,
122
+ "kl": 0.0008707046508789062,
123
+ "learning_rate": 2.666666666666667e-06,
124
+ "loss": 0.0704,
125
+ "num_tokens": 1093987.0,
126
+ "reward": 0.16250000149011612,
127
+ "reward_std": 0.3029462620615959,
128
+ "rewards/curriculum_aware_reward_fn": 0.037500000558793545,
129
+ "rewards/format_reward": 0.125,
130
+ "step": 8
131
+ },
132
+ {
133
+ "clip_ratio": 0.0,
134
+ "completion_length": 423.1484375,
135
+ "epoch": 0.15126050420168066,
136
+ "grad_norm": 0.7402248978614807,
137
+ "kl": 0.00279998779296875,
138
+ "learning_rate": 3e-06,
139
+ "loss": 0.0728,
140
+ "num_tokens": 1218118.0,
141
+ "reward": 0.3968750089406967,
142
+ "reward_std": 0.45669952034950256,
143
+ "rewards/curriculum_aware_reward_fn": 0.10781250474974513,
144
+ "rewards/format_reward": 0.2890625,
145
+ "step": 9
146
+ },
147
+ {
148
+ "clip_ratio": 0.0,
149
+ "completion_length": 496.734375,
150
+ "epoch": 0.16806722689075632,
151
+ "grad_norm": 0.7178624272346497,
152
+ "kl": 0.00432586669921875,
153
+ "learning_rate": 3.3333333333333333e-06,
154
+ "loss": -0.0207,
155
+ "num_tokens": 1362188.0,
156
+ "reward": 0.5677083432674408,
157
+ "reward_std": 0.5023705065250397,
158
+ "rewards/curriculum_aware_reward_fn": 0.013020833721384406,
159
+ "rewards/format_reward": 0.5546875,
160
+ "step": 10
161
+ },
162
+ {
163
+ "clip_ratio": 0.0,
164
+ "completion_length": 500.3984375,
165
+ "epoch": 0.18487394957983194,
166
+ "grad_norm": 0.6241538524627686,
167
+ "kl": 0.0084075927734375,
168
+ "learning_rate": 3.6666666666666666e-06,
169
+ "loss": 0.0165,
170
+ "num_tokens": 1496743.0,
171
+ "reward": 0.714062511920929,
172
+ "reward_std": 0.344537615776062,
173
+ "rewards/curriculum_aware_reward_fn": 0.010937500279396772,
174
+ "rewards/format_reward": 0.703125,
175
+ "step": 11
176
+ },
177
+ {
178
+ "clip_ratio": 0.0,
179
+ "completion_length": 517.5625,
180
+ "epoch": 0.20168067226890757,
181
+ "grad_norm": 0.6269044280052185,
182
+ "kl": 0.00717926025390625,
183
+ "learning_rate": 4.000000000000001e-06,
184
+ "loss": 0.0195,
185
+ "num_tokens": 1650679.0,
186
+ "reward": 0.7604166716337204,
187
+ "reward_std": 0.2794100269675255,
188
+ "rewards/curriculum_aware_reward_fn": 0.010416666977107525,
189
+ "rewards/format_reward": 0.75,
190
+ "step": 12
191
+ },
192
+ {
193
+ "clip_ratio": 0.0,
194
+ "completion_length": 489.9921875,
195
+ "epoch": 0.2184873949579832,
196
+ "grad_norm": 0.6628085374832153,
197
+ "kl": 0.011993408203125,
198
+ "learning_rate": 4.333333333333334e-06,
199
+ "loss": 0.0178,
200
+ "num_tokens": 1791190.0,
201
+ "reward": 0.9375000298023224,
202
+ "reward_std": 0.23163868859410286,
203
+ "rewards/curriculum_aware_reward_fn": 0.031250000931322575,
204
+ "rewards/format_reward": 0.90625,
205
+ "step": 13
206
+ },
207
+ {
208
+ "clip_ratio": 0.0,
209
+ "completion_length": 418.90625,
210
+ "epoch": 0.23529411764705882,
211
+ "grad_norm": 0.7439326643943787,
212
+ "kl": 0.0212249755859375,
213
+ "learning_rate": 4.666666666666667e-06,
214
+ "loss": -0.0283,
215
+ "num_tokens": 1912322.0,
216
+ "reward": 0.8645833432674408,
217
+ "reward_std": 0.2743529714643955,
218
+ "rewards/curriculum_aware_reward_fn": 0.02864583395421505,
219
+ "rewards/format_reward": 0.8359375,
220
+ "step": 14
221
+ },
222
+ {
223
+ "clip_ratio": 0.0,
224
+ "completion_length": 440.5546875,
225
+ "epoch": 0.25210084033613445,
226
+ "grad_norm": 0.477730393409729,
227
+ "kl": 0.0169830322265625,
228
+ "learning_rate": 5e-06,
229
+ "loss": 0.0314,
230
+ "num_tokens": 2041473.0,
231
+ "reward": 1.0234375298023224,
232
+ "reward_std": 0.159461235627532,
233
+ "rewards/curriculum_aware_reward_fn": 0.05468750325962901,
234
+ "rewards/format_reward": 0.96875,
235
+ "step": 15
236
+ },
237
+ {
238
+ "clip_ratio": 0.0,
239
+ "completion_length": 497.265625,
240
+ "epoch": 0.2689075630252101,
241
+ "grad_norm": 0.3044699728488922,
242
+ "kl": 0.021484375,
243
+ "learning_rate": 4.999952797253148e-06,
244
+ "loss": 0.0386,
245
+ "num_tokens": 2178899.0,
246
+ "reward": 0.9427083432674408,
247
+ "reward_std": 0.10222155228257179,
248
+ "rewards/curriculum_aware_reward_fn": 0.06770833395421505,
249
+ "rewards/format_reward": 0.875,
250
+ "step": 16
251
+ },
252
+ {
253
+ "clip_ratio": 0.0,
254
+ "completion_length": 503.7421875,
255
+ "epoch": 0.2857142857142857,
256
+ "grad_norm": 0.4504559338092804,
257
+ "kl": 0.033233642578125,
258
+ "learning_rate": 4.9998111909931225e-06,
259
+ "loss": 0.0346,
260
+ "num_tokens": 2311042.0,
261
+ "reward": 1.0927083492279053,
262
+ "reward_std": 0.13363875821232796,
263
+ "rewards/curriculum_aware_reward_fn": 0.09270833618938923,
264
+ "rewards/format_reward": 1.0,
265
+ "step": 17
266
+ },
267
+ {
268
+ "clip_ratio": 0.0,
269
+ "completion_length": 582.390625,
270
+ "epoch": 0.3025210084033613,
271
+ "grad_norm": 0.3522663712501526,
272
+ "kl": 0.033416748046875,
273
+ "learning_rate": 4.999575187161439e-06,
274
+ "loss": 0.0049,
275
+ "num_tokens": 2477404.0,
276
+ "reward": 0.997395858168602,
277
+ "reward_std": 0.0840194127522409,
278
+ "rewards/curriculum_aware_reward_fn": 0.05989583651535213,
279
+ "rewards/format_reward": 0.9375,
280
+ "step": 18
281
+ },
282
+ {
283
+ "clip_ratio": 0.0,
284
+ "completion_length": 603.4140625,
285
+ "epoch": 0.31932773109243695,
286
+ "grad_norm": 0.2832561731338501,
287
+ "kl": 0.0328369140625,
288
+ "learning_rate": 4.9992447956603455e-06,
289
+ "loss": 0.0152,
290
+ "num_tokens": 2642801.0,
291
+ "reward": 0.895833358168602,
292
+ "reward_std": 0.05173455877229571,
293
+ "rewards/curriculum_aware_reward_fn": 0.020833333721384406,
294
+ "rewards/format_reward": 0.875,
295
+ "step": 19
296
+ },
297
+ {
298
+ "clip_ratio": 0.0,
299
+ "completion_length": 501.46875,
300
+ "epoch": 0.33613445378151263,
301
+ "grad_norm": 0.4254682660102844,
302
+ "kl": 0.0416259765625,
303
+ "learning_rate": 4.998820030352409e-06,
304
+ "loss": 0.0009,
305
+ "num_tokens": 2784445.0,
306
+ "reward": 1.0010417103767395,
307
+ "reward_std": 0.13327472284436226,
308
+ "rewards/curriculum_aware_reward_fn": 0.07135416753590107,
309
+ "rewards/format_reward": 0.9296875,
310
+ "step": 20
311
+ },
312
+ {
313
+ "clip_ratio": 0.0,
314
+ "completion_length": 598.765625,
315
+ "epoch": 0.35294117647058826,
316
+ "grad_norm": 0.3762480914592743,
317
+ "kl": 0.033721923828125,
318
+ "learning_rate": 4.998300909059929e-06,
319
+ "loss": -0.0163,
320
+ "num_tokens": 2951415.0,
321
+ "reward": 0.9947916865348816,
322
+ "reward_std": 0.11004853993654251,
323
+ "rewards/curriculum_aware_reward_fn": 0.07291666604578495,
324
+ "rewards/format_reward": 0.921875,
325
+ "step": 21
326
+ },
327
+ {
328
+ "clip_ratio": 0.0,
329
+ "completion_length": 489.6484375,
330
+ "epoch": 0.3697478991596639,
331
+ "grad_norm": 0.6521299481391907,
332
+ "kl": 0.04229736328125,
333
+ "learning_rate": 4.997687453564198e-06,
334
+ "loss": 0.0083,
335
+ "num_tokens": 3090354.0,
336
+ "reward": 0.9609375298023224,
337
+ "reward_std": 0.25490327179431915,
338
+ "rewards/curriculum_aware_reward_fn": 0.1562500037252903,
339
+ "rewards/format_reward": 0.8046875,
340
+ "step": 22
341
+ },
342
+ {
343
+ "clip_ratio": 0.0,
344
+ "completion_length": 446.625,
345
+ "epoch": 0.3865546218487395,
346
+ "grad_norm": 0.4581259489059448,
347
+ "kl": 0.02838134765625,
348
+ "learning_rate": 4.9969796896045775e-06,
349
+ "loss": 0.0239,
350
+ "num_tokens": 3234002.0,
351
+ "reward": 1.1093750298023224,
352
+ "reward_std": 0.15402578841894865,
353
+ "rewards/curriculum_aware_reward_fn": 0.17187500139698386,
354
+ "rewards/format_reward": 0.9375,
355
+ "step": 23
356
+ },
357
+ {
358
+ "clip_ratio": 0.0,
359
+ "completion_length": 438.296875,
360
+ "epoch": 0.40336134453781514,
361
+ "grad_norm": 0.469014436006546,
362
+ "kl": 0.02874755859375,
363
+ "learning_rate": 4.996177646877426e-06,
364
+ "loss": 0.0065,
365
+ "num_tokens": 3368280.0,
366
+ "reward": 1.0302083790302277,
367
+ "reward_std": 0.12476669438183308,
368
+ "rewards/curriculum_aware_reward_fn": 0.045833335258066654,
369
+ "rewards/format_reward": 0.984375,
370
+ "step": 24
371
+ },
372
+ {
373
+ "clip_ratio": 0.0,
374
+ "completion_length": 433.9921875,
375
+ "epoch": 0.42016806722689076,
376
+ "grad_norm": 0.5325790643692017,
377
+ "kl": 0.028350830078125,
378
+ "learning_rate": 4.995281359034851e-06,
379
+ "loss": -0.0046,
380
+ "num_tokens": 3495607.0,
381
+ "reward": 1.0026041865348816,
382
+ "reward_std": 0.2044544592499733,
383
+ "rewards/curriculum_aware_reward_fn": 0.08072916977107525,
384
+ "rewards/format_reward": 0.921875,
385
+ "step": 25
386
+ },
387
+ {
388
+ "clip_ratio": 0.0,
389
+ "completion_length": 380.015625,
390
+ "epoch": 0.4369747899159664,
391
+ "grad_norm": 0.5200158357620239,
392
+ "kl": 0.03680419921875,
393
+ "learning_rate": 4.994290863683296e-06,
394
+ "loss": 0.0419,
395
+ "num_tokens": 3609809.0,
396
+ "reward": 1.1458334028720856,
397
+ "reward_std": 0.20207119127735496,
398
+ "rewards/curriculum_aware_reward_fn": 0.15364583837799728,
399
+ "rewards/format_reward": 0.9921875,
400
+ "step": 26
401
+ },
402
+ {
403
+ "clip_ratio": 0.0,
404
+ "completion_length": 413.7265625,
405
+ "epoch": 0.453781512605042,
406
+ "grad_norm": 0.41265320777893066,
407
+ "kl": 0.033966064453125,
408
+ "learning_rate": 4.99320620238196e-06,
409
+ "loss": -0.0204,
410
+ "num_tokens": 3744550.0,
411
+ "reward": 1.0963541865348816,
412
+ "reward_std": 0.11664257757365704,
413
+ "rewards/curriculum_aware_reward_fn": 0.09635416604578495,
414
+ "rewards/format_reward": 1.0,
415
+ "step": 27
416
+ },
417
+ {
418
+ "clip_ratio": 0.0,
419
+ "completion_length": 461.4375,
420
+ "epoch": 0.47058823529411764,
421
+ "grad_norm": 0.3836808502674103,
422
+ "kl": 0.032623291015625,
423
+ "learning_rate": 4.99202742064106e-06,
424
+ "loss": 0.035,
425
+ "num_tokens": 3875238.0,
426
+ "reward": 1.0182291865348816,
427
+ "reward_std": 0.10335793904960155,
428
+ "rewards/curriculum_aware_reward_fn": 0.0807291679084301,
429
+ "rewards/format_reward": 0.9375,
430
+ "step": 28
431
+ },
432
+ {
433
+ "clip_ratio": 0.0,
434
+ "completion_length": 506.953125,
435
+ "epoch": 0.48739495798319327,
436
+ "grad_norm": 0.3787353038787842,
437
+ "kl": 0.029510498046875,
438
+ "learning_rate": 4.990754567919917e-06,
439
+ "loss": 0.0725,
440
+ "num_tokens": 4024312.0,
441
+ "reward": 0.9531250298023224,
442
+ "reward_std": 0.08838835544884205,
443
+ "rewards/curriculum_aware_reward_fn": 0.08593750279396772,
444
+ "rewards/format_reward": 0.8671875,
445
+ "step": 29
446
+ },
447
+ {
448
+ "clip_ratio": 0.0,
449
+ "completion_length": 474.1875,
450
+ "epoch": 0.5042016806722689,
451
+ "grad_norm": 0.37932828068733215,
452
+ "kl": 0.03497314453125,
453
+ "learning_rate": 4.989387697624881e-06,
454
+ "loss": 0.0078,
455
+ "num_tokens": 4161712.0,
456
+ "reward": 1.1328125298023224,
457
+ "reward_std": 0.14552949741482735,
458
+ "rewards/curriculum_aware_reward_fn": 0.1328125037252903,
459
+ "rewards/format_reward": 1.0,
460
+ "step": 30
461
+ },
462
+ {
463
+ "clip_ratio": 0.0,
464
+ "completion_length": 447.6796875,
465
+ "epoch": 0.5210084033613446,
466
+ "grad_norm": 0.396627813577652,
467
+ "kl": 0.0396728515625,
468
+ "learning_rate": 4.987926867107095e-06,
469
+ "loss": 0.004,
470
+ "num_tokens": 4304935.0,
471
+ "reward": 0.9671875089406967,
472
+ "reward_std": 0.12499183788895607,
473
+ "rewards/curriculum_aware_reward_fn": 0.09218750381842256,
474
+ "rewards/format_reward": 0.875,
475
+ "step": 31
476
+ },
477
+ {
478
+ "clip_ratio": 0.0,
479
+ "completion_length": 574.703125,
480
+ "epoch": 0.5378151260504201,
481
+ "grad_norm": 0.39759182929992676,
482
+ "kl": 0.030426025390625,
483
+ "learning_rate": 4.986372137660078e-06,
484
+ "loss": 0.0566,
485
+ "num_tokens": 4464105.0,
486
+ "reward": 0.8953125327825546,
487
+ "reward_std": 0.147516256198287,
488
+ "rewards/curriculum_aware_reward_fn": 0.0906250006519258,
489
+ "rewards/format_reward": 0.8046875,
490
+ "step": 32
491
+ },
492
+ {
493
+ "clip_ratio": 0.0,
494
+ "completion_length": 434.640625,
495
+ "epoch": 0.5546218487394958,
496
+ "grad_norm": 0.3933623433113098,
497
+ "kl": 0.034210205078125,
498
+ "learning_rate": 4.984723574517165e-06,
499
+ "loss": 0.0163,
500
+ "num_tokens": 4602139.0,
501
+ "reward": 1.1234374940395355,
502
+ "reward_std": 0.15428178012371063,
503
+ "rewards/curriculum_aware_reward_fn": 0.18593750335276127,
504
+ "rewards/format_reward": 0.9375,
505
+ "step": 33
506
+ },
507
+ {
508
+ "clip_ratio": 0.0,
509
+ "completion_length": 445.1640625,
510
+ "epoch": 0.5714285714285714,
511
+ "grad_norm": 0.3263511061668396,
512
+ "kl": 0.04083251953125,
513
+ "learning_rate": 4.9829812468487655e-06,
514
+ "loss": 0.0145,
515
+ "num_tokens": 4736544.0,
516
+ "reward": 0.9843750298023224,
517
+ "reward_std": 0.09849035926163197,
518
+ "rewards/curriculum_aware_reward_fn": 0.05468750139698386,
519
+ "rewards/format_reward": 0.9296875,
520
+ "step": 34
521
+ },
522
+ {
523
+ "clip_ratio": 0.0,
524
+ "completion_length": 477.53125,
525
+ "epoch": 0.5882352941176471,
526
+ "grad_norm": 0.336434543132782,
527
+ "kl": 0.0318603515625,
528
+ "learning_rate": 4.981145227759457e-06,
529
+ "loss": 0.0017,
530
+ "num_tokens": 4881308.0,
531
+ "reward": 0.966145858168602,
532
+ "reward_std": 0.07084779627621174,
533
+ "rewards/curriculum_aware_reward_fn": 0.09114583441987634,
534
+ "rewards/format_reward": 0.875,
535
+ "step": 35
536
+ },
537
+ {
538
+ "clip_ratio": 0.0,
539
+ "completion_length": 469.4609375,
540
+ "epoch": 0.6050420168067226,
541
+ "grad_norm": 0.3600573241710663,
542
+ "kl": 0.0386962890625,
543
+ "learning_rate": 4.979215594284924e-06,
544
+ "loss": 0.008,
545
+ "num_tokens": 5017415.0,
546
+ "reward": 1.0843750536441803,
547
+ "reward_std": 0.11371596809476614,
548
+ "rewards/curriculum_aware_reward_fn": 0.08437500288709998,
549
+ "rewards/format_reward": 1.0,
550
+ "step": 36
551
+ },
552
+ {
553
+ "clip_ratio": 0.0,
554
+ "completion_length": 422.7890625,
555
+ "epoch": 0.6218487394957983,
556
+ "grad_norm": 0.38463500142097473,
557
+ "kl": 0.03997802734375,
558
+ "learning_rate": 4.977192427388722e-06,
559
+ "loss": 0.0195,
560
+ "num_tokens": 5141804.0,
561
+ "reward": 1.1130208671092987,
562
+ "reward_std": 0.1456994889304042,
563
+ "rewards/curriculum_aware_reward_fn": 0.12083333730697632,
564
+ "rewards/format_reward": 0.9921875,
565
+ "step": 37
566
+ },
567
+ {
568
+ "clip_ratio": 0.0,
569
+ "completion_length": 415.4765625,
570
+ "epoch": 0.6386554621848739,
571
+ "grad_norm": 0.5343978404998779,
572
+ "kl": 0.04315185546875,
573
+ "learning_rate": 4.9750758119588824e-06,
574
+ "loss": -0.0,
575
+ "num_tokens": 5262529.0,
576
+ "reward": 1.136979192495346,
577
+ "reward_std": 0.29098181426525116,
578
+ "rewards/curriculum_aware_reward_fn": 0.16822916036471725,
579
+ "rewards/format_reward": 0.96875,
580
+ "step": 38
581
+ },
582
+ {
583
+ "clip_ratio": 0.0,
584
+ "completion_length": 442.2421875,
585
+ "epoch": 0.6554621848739496,
586
+ "grad_norm": 0.33322009444236755,
587
+ "kl": 0.047119140625,
588
+ "learning_rate": 4.972865836804349e-06,
589
+ "loss": 0.0088,
590
+ "num_tokens": 5399424.0,
591
+ "reward": 1.03125,
592
+ "reward_std": 0.13204573467373848,
593
+ "rewards/curriculum_aware_reward_fn": 0.109375,
594
+ "rewards/format_reward": 0.921875,
595
+ "step": 39
596
+ },
597
+ {
598
+ "clip_ratio": 0.0,
599
+ "completion_length": 482.59375,
600
+ "epoch": 0.6722689075630253,
601
+ "grad_norm": 0.35028988122940063,
602
+ "kl": 0.0504150390625,
603
+ "learning_rate": 4.970562594651254e-06,
604
+ "loss": 0.0066,
605
+ "num_tokens": 5539492.0,
606
+ "reward": 1.09375,
607
+ "reward_std": 0.1325825173407793,
608
+ "rewards/curriculum_aware_reward_fn": 0.125,
609
+ "rewards/format_reward": 0.96875,
610
+ "step": 40
611
+ },
612
+ {
613
+ "clip_ratio": 0.0,
614
+ "completion_length": 423.125,
615
+ "epoch": 0.6890756302521008,
616
+ "grad_norm": 0.3613918125629425,
617
+ "kl": 0.06549072265625,
618
+ "learning_rate": 4.968166182139026e-06,
619
+ "loss": 0.0516,
620
+ "num_tokens": 5667012.0,
621
+ "reward": 1.122395858168602,
622
+ "reward_std": 0.09522313997149467,
623
+ "rewards/curriculum_aware_reward_fn": 0.13020833837799728,
624
+ "rewards/format_reward": 0.9921875,
625
+ "step": 41
626
+ },
627
+ {
628
+ "clip_ratio": 0.0,
629
+ "completion_length": 391.953125,
630
+ "epoch": 0.7058823529411765,
631
+ "grad_norm": 0.6798313856124878,
632
+ "kl": 0.080322265625,
633
+ "learning_rate": 4.9656766998163306e-06,
634
+ "loss": 0.0429,
635
+ "num_tokens": 5786006.0,
636
+ "reward": 1.1645833551883698,
637
+ "reward_std": 0.23287939466536045,
638
+ "rewards/curriculum_aware_reward_fn": 0.19583334028720856,
639
+ "rewards/format_reward": 0.96875,
640
+ "step": 42
641
+ },
642
+ {
643
+ "clip_ratio": 0.0,
644
+ "completion_length": 406.59375,
645
+ "epoch": 0.7226890756302521,
646
+ "grad_norm": 0.4006165862083435,
647
+ "kl": 0.06298828125,
648
+ "learning_rate": 4.963094252136865e-06,
649
+ "loss": 0.0119,
650
+ "num_tokens": 5910762.0,
651
+ "reward": 1.0989583432674408,
652
+ "reward_std": 0.1059559416025877,
653
+ "rewards/curriculum_aware_reward_fn": 0.16927083767950535,
654
+ "rewards/format_reward": 0.9296875,
655
+ "step": 43
656
+ },
657
+ {
658
+ "clip_ratio": 0.0,
659
+ "completion_length": 411.28125,
660
+ "epoch": 0.7394957983193278,
661
+ "grad_norm": 0.3308926522731781,
662
+ "kl": 0.068359375,
663
+ "learning_rate": 4.960418947454958e-06,
664
+ "loss": 0.0204,
665
+ "num_tokens": 6042942.0,
666
+ "reward": 0.9947916865348816,
667
+ "reward_std": 0.10640286095440388,
668
+ "rewards/curriculum_aware_reward_fn": 0.0572916679084301,
669
+ "rewards/format_reward": 0.9375,
670
+ "step": 44
671
+ },
672
+ {
673
+ "clip_ratio": 0.0,
674
+ "completion_length": 380.9140625,
675
+ "epoch": 0.7563025210084033,
676
+ "grad_norm": 0.478098064661026,
677
+ "kl": 0.0755615234375,
678
+ "learning_rate": 4.957650898021038e-06,
679
+ "loss": 0.0624,
680
+ "num_tokens": 6162299.0,
681
+ "reward": 1.158854216337204,
682
+ "reward_std": 0.20395735278725624,
683
+ "rewards/curriculum_aware_reward_fn": 0.15885416977107525,
684
+ "rewards/format_reward": 1.0,
685
+ "step": 45
686
+ },
687
+ {
688
+ "clip_ratio": 0.0,
689
+ "completion_length": 333.5390625,
690
+ "epoch": 0.773109243697479,
691
+ "grad_norm": 0.47225892543792725,
692
+ "kl": 0.0811767578125,
693
+ "learning_rate": 4.954790219976915e-06,
694
+ "loss": 0.0167,
695
+ "num_tokens": 6272592.0,
696
+ "reward": 1.170312523841858,
697
+ "reward_std": 0.1300742938183248,
698
+ "rewards/curriculum_aware_reward_fn": 0.17031250218860805,
699
+ "rewards/format_reward": 1.0,
700
+ "step": 46
701
+ },
702
+ {
703
+ "clip_ratio": 0.0,
704
+ "completion_length": 366.34375,
705
+ "epoch": 0.7899159663865546,
706
+ "grad_norm": 0.4816167652606964,
707
+ "kl": 0.0848388671875,
708
+ "learning_rate": 4.95183703335091e-06,
709
+ "loss": 0.0103,
710
+ "num_tokens": 6397436.0,
711
+ "reward": 1.1953125298023224,
712
+ "reward_std": 0.12861409783363342,
713
+ "rewards/curriculum_aware_reward_fn": 0.1953125074505806,
714
+ "rewards/format_reward": 1.0,
715
+ "step": 47
716
+ },
717
+ {
718
+ "clip_ratio": 0.0,
719
+ "completion_length": 338.65625,
720
+ "epoch": 0.8067226890756303,
721
+ "grad_norm": 0.47690296173095703,
722
+ "kl": 0.07958984375,
723
+ "learning_rate": 4.948791462052819e-06,
724
+ "loss": -0.0039,
725
+ "num_tokens": 6519896.0,
726
+ "reward": 1.0442708730697632,
727
+ "reward_std": 0.11976211331784725,
728
+ "rewards/curriculum_aware_reward_fn": 0.05208333441987634,
729
+ "rewards/format_reward": 0.9921875,
730
+ "step": 48
731
+ },
732
+ {
733
+ "clip_ratio": 0.0,
734
+ "completion_length": 325.78125,
735
+ "epoch": 0.8235294117647058,
736
+ "grad_norm": 0.46987131237983704,
737
+ "kl": 0.08935546875,
738
+ "learning_rate": 4.945653633868716e-06,
739
+ "loss": 0.023,
740
+ "num_tokens": 6637036.0,
741
+ "reward": 1.1223958730697632,
742
+ "reward_std": 0.18378917127847672,
743
+ "rewards/curriculum_aware_reward_fn": 0.14583333488553762,
744
+ "rewards/format_reward": 0.9765625,
745
+ "step": 49
746
+ },
747
+ {
748
+ "clip_ratio": 0.0,
749
+ "completion_length": 302.6171875,
750
+ "epoch": 0.8403361344537815,
751
+ "grad_norm": 0.5635585784912109,
752
+ "kl": 0.088623046875,
753
+ "learning_rate": 4.942423680455584e-06,
754
+ "loss": 0.0157,
755
+ "num_tokens": 6749411.0,
756
+ "reward": 1.1494792103767395,
757
+ "reward_std": 0.19005249440670013,
758
+ "rewards/curriculum_aware_reward_fn": 0.1494791703298688,
759
+ "rewards/format_reward": 1.0,
760
+ "step": 50
761
+ },
762
+ {
763
+ "clip_ratio": 0.0,
764
+ "completion_length": 366.4140625,
765
+ "epoch": 0.8571428571428571,
766
+ "grad_norm": 0.5150569677352905,
767
+ "kl": 0.0848388671875,
768
+ "learning_rate": 4.939101737335802e-06,
769
+ "loss": -0.0145,
770
+ "num_tokens": 6879104.0,
771
+ "reward": 0.9713541865348816,
772
+ "reward_std": 0.12468592822551727,
773
+ "rewards/curriculum_aware_reward_fn": 0.041666666977107525,
774
+ "rewards/format_reward": 0.9296875,
775
+ "step": 51
776
+ },
777
+ {
778
+ "clip_ratio": 0.0,
779
+ "completion_length": 288.9609375,
780
+ "epoch": 0.8739495798319328,
781
+ "grad_norm": 0.6744732856750488,
782
+ "kl": 0.088134765625,
783
+ "learning_rate": 4.935687943891447e-06,
784
+ "loss": 0.0413,
785
+ "num_tokens": 6981627.0,
786
+ "reward": 1.171354204416275,
787
+ "reward_std": 0.26688926108181477,
788
+ "rewards/curriculum_aware_reward_fn": 0.1791666648350656,
789
+ "rewards/format_reward": 0.9921875,
790
+ "step": 52
791
+ },
792
+ {
793
+ "clip_ratio": 0.0,
794
+ "completion_length": 286.609375,
795
+ "epoch": 0.8907563025210085,
796
+ "grad_norm": 0.6298589706420898,
797
+ "kl": 0.0899658203125,
798
+ "learning_rate": 4.932182443358458e-06,
799
+ "loss": 0.002,
800
+ "num_tokens": 7088449.0,
801
+ "reward": 1.1250000298023224,
802
+ "reward_std": 0.14211241621524096,
803
+ "rewards/curriculum_aware_reward_fn": 0.125000003259629,
804
+ "rewards/format_reward": 1.0,
805
+ "step": 53
806
+ },
807
+ {
808
+ "clip_ratio": 0.0,
809
+ "completion_length": 302.4921875,
810
+ "epoch": 0.907563025210084,
811
+ "grad_norm": 0.6446663737297058,
812
+ "kl": 0.0955810546875,
813
+ "learning_rate": 4.928585382820616e-06,
814
+ "loss": 0.0001,
815
+ "num_tokens": 7197360.0,
816
+ "reward": 1.1005208790302277,
817
+ "reward_std": 0.08887648163363338,
818
+ "rewards/curriculum_aware_reward_fn": 0.10052083828486502,
819
+ "rewards/format_reward": 1.0,
820
+ "step": 54
821
+ },
822
+ {
823
+ "clip_ratio": 0.0,
824
+ "completion_length": 399.0625,
825
+ "epoch": 0.9243697478991597,
826
+ "grad_norm": 0.729965090751648,
827
+ "kl": 0.1077880859375,
828
+ "learning_rate": 4.924896913203376e-06,
829
+ "loss": 0.0496,
830
+ "num_tokens": 7331528.0,
831
+ "reward": 0.9932291805744171,
832
+ "reward_std": 0.15547171607613564,
833
+ "rewards/curriculum_aware_reward_fn": 0.1807291698642075,
834
+ "rewards/format_reward": 0.8125,
835
+ "step": 55
836
+ },
837
+ {
838
+ "clip_ratio": 0.0,
839
+ "completion_length": 494.09375,
840
+ "epoch": 0.9411764705882353,
841
+ "grad_norm": 0.41514354944229126,
842
+ "kl": 0.0888671875,
843
+ "learning_rate": 4.921117189267535e-06,
844
+ "loss": -0.0255,
845
+ "num_tokens": 7482548.0,
846
+ "reward": 0.8671875149011612,
847
+ "reward_std": 0.09729943191632628,
848
+ "rewards/curriculum_aware_reward_fn": 0.05468750069849193,
849
+ "rewards/format_reward": 0.8125,
850
+ "step": 56
851
+ },
852
+ {
853
+ "clip_ratio": 0.0,
854
+ "completion_length": 511.90625,
855
+ "epoch": 0.957983193277311,
856
+ "grad_norm": 0.6172528266906738,
857
+ "kl": 0.1680908203125,
858
+ "learning_rate": 4.917246369602742e-06,
859
+ "loss": 0.0368,
860
+ "num_tokens": 7621256.0,
861
+ "reward": 1.050520896911621,
862
+ "reward_std": 0.1373576819896698,
863
+ "rewards/curriculum_aware_reward_fn": 0.17552083916962147,
864
+ "rewards/format_reward": 0.875,
865
+ "step": 57
866
+ },
867
+ {
868
+ "clip_ratio": 0.0,
869
+ "completion_length": 415.8046875,
870
+ "epoch": 0.9747899159663865,
871
+ "grad_norm": 0.5895751118659973,
872
+ "kl": 0.2454833984375,
873
+ "learning_rate": 4.9132846166208355e-06,
874
+ "loss": 0.0654,
875
+ "num_tokens": 7740207.0,
876
+ "reward": 1.1197916865348816,
877
+ "reward_std": 0.09127403190359473,
878
+ "rewards/curriculum_aware_reward_fn": 0.12760416674427688,
879
+ "rewards/format_reward": 0.9921875,
880
+ "step": 58
881
+ },
882
+ {
883
+ "clip_ratio": 0.0,
884
+ "completion_length": 578.2857208251953,
885
+ "epoch": 0.9915966386554622,
886
+ "grad_norm": 0.4072950482368469,
887
+ "kl": 0.12353515625,
888
+ "learning_rate": 4.9092320965490365e-06,
889
+ "loss": 0.0363,
890
+ "num_tokens": 7885325.0,
891
+ "reward": 1.0598958432674408,
892
+ "reward_std": 0.11012758687138557,
893
+ "rewards/curriculum_aware_reward_fn": 0.0598958358168602,
894
+ "rewards/format_reward": 1.0,
895
+ "step": 59
896
+ },
897
+ {
898
+ "clip_ratio": 0.0,
899
+ "completion_length": 564.78125,
900
+ "epoch": 1.0168067226890756,
901
+ "grad_norm": 0.4054146707057953,
902
+ "kl": 0.1219482421875,
903
+ "learning_rate": 4.905088979422971e-06,
904
+ "loss": 0.0515,
905
+ "num_tokens": 8038297.0,
906
+ "reward": 1.0937500596046448,
907
+ "reward_std": 0.1636663186363876,
908
+ "rewards/curriculum_aware_reward_fn": 0.1015625053551048,
909
+ "rewards/format_reward": 0.9921875,
910
+ "step": 60
911
+ },
912
+ {
913
+ "clip_ratio": 0.0,
914
+ "completion_length": 578.2890625,
915
+ "epoch": 1.0336134453781514,
916
+ "grad_norm": 0.3728667199611664,
917
+ "kl": 0.126708984375,
918
+ "learning_rate": 4.900855439079536e-06,
919
+ "loss": 0.0134,
920
+ "num_tokens": 8185046.0,
921
+ "reward": 1.1265625357627869,
922
+ "reward_std": 0.14053087309002876,
923
+ "rewards/curriculum_aware_reward_fn": 0.12656250409781933,
924
+ "rewards/format_reward": 1.0,
925
+ "step": 61
926
+ },
927
+ {
928
+ "clip_ratio": 0.0,
929
+ "completion_length": 722.3125,
930
+ "epoch": 1.050420168067227,
931
+ "grad_norm": 0.32597795128822327,
932
+ "kl": 0.1097412109375,
933
+ "learning_rate": 4.8965316531496055e-06,
934
+ "loss": 0.0184,
935
+ "num_tokens": 8366894.0,
936
+ "reward": 0.9677083492279053,
937
+ "reward_std": 0.10382660711184144,
938
+ "rewards/curriculum_aware_reward_fn": 0.04583333386108279,
939
+ "rewards/format_reward": 0.921875,
940
+ "step": 62
941
+ },
942
+ {
943
+ "clip_ratio": 0.0,
944
+ "completion_length": 663.3671875,
945
+ "epoch": 1.0672268907563025,
946
+ "grad_norm": 0.4411354064941406,
947
+ "kl": 0.1256103515625,
948
+ "learning_rate": 4.892117803050578e-06,
949
+ "loss": 0.0637,
950
+ "num_tokens": 8514637.0,
951
+ "reward": 1.0671875327825546,
952
+ "reward_std": 0.21371026523411274,
953
+ "rewards/curriculum_aware_reward_fn": 0.09843750204890966,
954
+ "rewards/format_reward": 0.96875,
955
+ "step": 63
956
+ },
957
+ {
958
+ "clip_ratio": 0.0,
959
+ "completion_length": 644.1328125,
960
+ "epoch": 1.084033613445378,
961
+ "grad_norm": 0.36323124170303345,
962
+ "kl": 0.1290283203125,
963
+ "learning_rate": 4.887614073978761e-06,
964
+ "loss": -0.0258,
965
+ "num_tokens": 8683182.0,
966
+ "reward": 0.9921875,
967
+ "reward_std": 0.12415501847863197,
968
+ "rewards/curriculum_aware_reward_fn": 0.0859375,
969
+ "rewards/format_reward": 0.90625,
970
+ "step": 64
971
+ },
972
+ {
973
+ "clip_ratio": 0.0,
974
+ "completion_length": 659.5703125,
975
+ "epoch": 1.1008403361344539,
976
+ "grad_norm": 0.5263646245002747,
977
+ "kl": 0.1494140625,
978
+ "learning_rate": 4.883020654901609e-06,
979
+ "loss": -0.03,
980
+ "num_tokens": 8846031.0,
981
+ "reward": 0.9375000149011612,
982
+ "reward_std": 0.29325952008366585,
983
+ "rewards/curriculum_aware_reward_fn": 0.08593750093132257,
984
+ "rewards/format_reward": 0.8515625,
985
+ "step": 65
986
+ },
987
+ {
988
+ "clip_ratio": 0.0,
989
+ "completion_length": 542.421875,
990
+ "epoch": 1.1176470588235294,
991
+ "grad_norm": 0.5841398239135742,
992
+ "kl": 0.1328125,
993
+ "learning_rate": 4.878337738549785e-06,
994
+ "loss": 0.0185,
995
+ "num_tokens": 8994685.0,
996
+ "reward": 0.9244791865348816,
997
+ "reward_std": 0.350746251642704,
998
+ "rewards/curriculum_aware_reward_fn": 0.0651041679084301,
999
+ "rewards/format_reward": 0.859375,
1000
+ "step": 66
1001
+ },
1002
+ {
1003
+ "clip_ratio": 0.0,
1004
+ "completion_length": 582.9453125,
1005
+ "epoch": 1.134453781512605,
1006
+ "grad_norm": 0.5570164322853088,
1007
+ "kl": 0.122314453125,
1008
+ "learning_rate": 4.873565521409082e-06,
1009
+ "loss": 0.0521,
1010
+ "num_tokens": 9155622.0,
1011
+ "reward": 0.9843750298023224,
1012
+ "reward_std": 0.3105625621974468,
1013
+ "rewards/curriculum_aware_reward_fn": 0.140625,
1014
+ "rewards/format_reward": 0.84375,
1015
+ "step": 67
1016
+ },
1017
+ {
1018
+ "clip_ratio": 0.0,
1019
+ "completion_length": 485.3359375,
1020
+ "epoch": 1.1512605042016806,
1021
+ "grad_norm": 0.5985521078109741,
1022
+ "kl": 0.1263427734375,
1023
+ "learning_rate": 4.868704203712173e-06,
1024
+ "loss": 0.0059,
1025
+ "num_tokens": 9288609.0,
1026
+ "reward": 1.052083358168602,
1027
+ "reward_std": 0.3865399658679962,
1028
+ "rewards/curriculum_aware_reward_fn": 0.18489583488553762,
1029
+ "rewards/format_reward": 0.8671875,
1030
+ "step": 68
1031
+ },
1032
+ {
1033
+ "clip_ratio": 0.0,
1034
+ "completion_length": 535.578125,
1035
+ "epoch": 1.1680672268907564,
1036
+ "grad_norm": 0.564681887626648,
1037
+ "kl": 0.1126708984375,
1038
+ "learning_rate": 4.86375398943021e-06,
1039
+ "loss": -0.0485,
1040
+ "num_tokens": 9437915.0,
1041
+ "reward": 0.9140625298023224,
1042
+ "reward_std": 0.28478266298770905,
1043
+ "rewards/curriculum_aware_reward_fn": 0.07031250256113708,
1044
+ "rewards/format_reward": 0.84375,
1045
+ "step": 69
1046
+ },
1047
+ {
1048
+ "clip_ratio": 0.0,
1049
+ "completion_length": 496.0703125,
1050
+ "epoch": 1.184873949579832,
1051
+ "grad_norm": 0.5399004220962524,
1052
+ "kl": 0.099853515625,
1053
+ "learning_rate": 4.858715086264274e-06,
1054
+ "loss": 0.0,
1055
+ "num_tokens": 9580828.0,
1056
+ "reward": 0.934895858168602,
1057
+ "reward_std": 0.19700524397194386,
1058
+ "rewards/curriculum_aware_reward_fn": 0.04427083441987634,
1059
+ "rewards/format_reward": 0.890625,
1060
+ "step": 70
1061
+ },
1062
+ {
1063
+ "clip_ratio": 0.0,
1064
+ "completion_length": 469.078125,
1065
+ "epoch": 1.2016806722689075,
1066
+ "grad_norm": 0.552808940410614,
1067
+ "kl": 0.104736328125,
1068
+ "learning_rate": 4.853587705636646e-06,
1069
+ "loss": -0.0129,
1070
+ "num_tokens": 9710926.0,
1071
+ "reward": 1.0234375447034836,
1072
+ "reward_std": 0.2662259414792061,
1073
+ "rewards/curriculum_aware_reward_fn": 0.07031250186264515,
1074
+ "rewards/format_reward": 0.953125,
1075
+ "step": 71
1076
+ },
1077
+ {
1078
+ "clip_ratio": 0.0,
1079
+ "completion_length": 539.765625,
1080
+ "epoch": 1.2184873949579833,
1081
+ "grad_norm": 0.4241795837879181,
1082
+ "kl": 0.093505859375,
1083
+ "learning_rate": 4.84837206268195e-06,
1084
+ "loss": -0.0161,
1085
+ "num_tokens": 9867328.0,
1086
+ "reward": 0.796875,
1087
+ "reward_std": 0.12126770988106728,
1088
+ "rewards/curriculum_aware_reward_fn": 0.015625,
1089
+ "rewards/format_reward": 0.78125,
1090
+ "step": 72
1091
+ },
1092
+ {
1093
+ "clip_ratio": 0.0,
1094
+ "completion_length": 502.6328125,
1095
+ "epoch": 1.2352941176470589,
1096
+ "grad_norm": 0.4336708188056946,
1097
+ "kl": 0.0975341796875,
1098
+ "learning_rate": 4.8430683762381195e-06,
1099
+ "loss": 0.0086,
1100
+ "num_tokens": 10007809.0,
1101
+ "reward": 1.0416666865348816,
1102
+ "reward_std": 0.1647402998059988,
1103
+ "rewards/curriculum_aware_reward_fn": 0.07291666697710752,
1104
+ "rewards/format_reward": 0.96875,
1105
+ "step": 73
1106
+ },
1107
+ {
1108
+ "clip_ratio": 0.0,
1109
+ "completion_length": 523.640625,
1110
+ "epoch": 1.2521008403361344,
1111
+ "grad_norm": 0.46911758184432983,
1112
+ "kl": 0.1041259765625,
1113
+ "learning_rate": 4.837676868837213e-06,
1114
+ "loss": 0.0313,
1115
+ "num_tokens": 10147291.0,
1116
+ "reward": 1.0302083790302277,
1117
+ "reward_std": 0.18167300708591938,
1118
+ "rewards/curriculum_aware_reward_fn": 0.06145833572372794,
1119
+ "rewards/format_reward": 0.96875,
1120
+ "step": 74
1121
+ },
1122
+ {
1123
+ "clip_ratio": 0.0,
1124
+ "completion_length": 628.6171875,
1125
+ "epoch": 1.26890756302521,
1126
+ "grad_norm": 0.5553966760635376,
1127
+ "kl": 0.091552734375,
1128
+ "learning_rate": 4.832197766696085e-06,
1129
+ "loss": 0.0771,
1130
+ "num_tokens": 10304234.0,
1131
+ "reward": 1.0401041805744171,
1132
+ "reward_std": 0.2994466572999954,
1133
+ "rewards/curriculum_aware_reward_fn": 0.14166666939854622,
1134
+ "rewards/format_reward": 0.8984375,
1135
+ "step": 75
1136
+ },
1137
+ {
1138
+ "clip_ratio": 0.0,
1139
+ "completion_length": 541.03125,
1140
+ "epoch": 1.2857142857142856,
1141
+ "grad_norm": 0.4850277602672577,
1142
+ "kl": 0.092529296875,
1143
+ "learning_rate": 4.826631299706887e-06,
1144
+ "loss": -0.0076,
1145
+ "num_tokens": 10455718.0,
1146
+ "reward": 1.043229192495346,
1147
+ "reward_std": 0.18734892271459103,
1148
+ "rewards/curriculum_aware_reward_fn": 0.12135417200624943,
1149
+ "rewards/format_reward": 0.921875,
1150
+ "step": 76
1151
+ },
1152
+ {
1153
+ "clip_ratio": 0.0,
1154
+ "completion_length": 513.34375,
1155
+ "epoch": 1.3025210084033614,
1156
+ "grad_norm": 0.5054645538330078,
1157
+ "kl": 0.098876953125,
1158
+ "learning_rate": 4.820977701427424e-06,
1159
+ "loss": 0.0342,
1160
+ "num_tokens": 10586514.0,
1161
+ "reward": 1.0703125149011612,
1162
+ "reward_std": 0.18218252062797546,
1163
+ "rewards/curriculum_aware_reward_fn": 0.10156250256113708,
1164
+ "rewards/format_reward": 0.96875,
1165
+ "step": 77
1166
+ },
1167
+ {
1168
+ "clip_ratio": 0.0,
1169
+ "completion_length": 554.0390625,
1170
+ "epoch": 1.319327731092437,
1171
+ "grad_norm": 0.5157454013824463,
1172
+ "kl": 0.0882568359375,
1173
+ "learning_rate": 4.81523720907136e-06,
1174
+ "loss": 0.05,
1175
+ "num_tokens": 10726735.0,
1176
+ "reward": 1.1385416984558105,
1177
+ "reward_std": 0.23471001349389553,
1178
+ "rewards/curriculum_aware_reward_fn": 0.17760416865348816,
1179
+ "rewards/format_reward": 0.9609375,
1180
+ "step": 78
1181
+ },
1182
+ {
1183
+ "clip_ratio": 0.0,
1184
+ "completion_length": 580.015625,
1185
+ "epoch": 1.3361344537815127,
1186
+ "grad_norm": 0.41959676146507263,
1187
+ "kl": 0.0958251953125,
1188
+ "learning_rate": 4.809410063498254e-06,
1189
+ "loss": 0.0059,
1190
+ "num_tokens": 10884185.0,
1191
+ "reward": 0.8411458283662796,
1192
+ "reward_std": 0.22234245762228966,
1193
+ "rewards/curriculum_aware_reward_fn": 0.06770833395421505,
1194
+ "rewards/format_reward": 0.7734375,
1195
+ "step": 79
1196
+ },
1197
+ {
1198
+ "clip_ratio": 0.0,
1199
+ "completion_length": 534.84375,
1200
+ "epoch": 1.3529411764705883,
1201
+ "grad_norm": 0.6390196681022644,
1202
+ "kl": 0.093017578125,
1203
+ "learning_rate": 4.8034965092034656e-06,
1204
+ "loss": 0.0963,
1205
+ "num_tokens": 11024149.0,
1206
+ "reward": 1.0494791865348816,
1207
+ "reward_std": 0.2942212373018265,
1208
+ "rewards/curriculum_aware_reward_fn": 0.1432291679084301,
1209
+ "rewards/format_reward": 0.90625,
1210
+ "step": 80
1211
+ },
1212
+ {
1213
+ "clip_ratio": 0.0,
1214
+ "completion_length": 525.671875,
1215
+ "epoch": 1.3697478991596639,
1216
+ "grad_norm": 0.4957650303840637,
1217
+ "kl": 0.105224609375,
1218
+ "learning_rate": 4.797496794307889e-06,
1219
+ "loss": -0.0236,
1220
+ "num_tokens": 11167547.0,
1221
+ "reward": 0.9828125238418579,
1222
+ "reward_std": 0.14313106751069427,
1223
+ "rewards/curriculum_aware_reward_fn": 0.06093750288709998,
1224
+ "rewards/format_reward": 0.921875,
1225
+ "step": 81
1226
+ },
1227
+ {
1228
+ "clip_ratio": 0.0,
1229
+ "completion_length": 488.7890625,
1230
+ "epoch": 1.3865546218487395,
1231
+ "grad_norm": 0.5502341389656067,
1232
+ "kl": 0.104248046875,
1233
+ "learning_rate": 4.791411170547545e-06,
1234
+ "loss": 0.072,
1235
+ "num_tokens": 11300104.0,
1236
+ "reward": 1.143229216337204,
1237
+ "reward_std": 0.2514217048883438,
1238
+ "rewards/curriculum_aware_reward_fn": 0.1822916679084301,
1239
+ "rewards/format_reward": 0.9609375,
1240
+ "step": 82
1241
+ },
1242
+ {
1243
+ "clip_ratio": 0.0,
1244
+ "completion_length": 508.1015625,
1245
+ "epoch": 1.403361344537815,
1246
+ "grad_norm": 0.489242285490036,
1247
+ "kl": 0.0986328125,
1248
+ "learning_rate": 4.785239893263017e-06,
1249
+ "loss": 0.1064,
1250
+ "num_tokens": 11446565.0,
1251
+ "reward": 0.9505208730697632,
1252
+ "reward_std": 0.18909456953406334,
1253
+ "rewards/curriculum_aware_reward_fn": 0.1145833358168602,
1254
+ "rewards/format_reward": 0.8359375,
1255
+ "step": 83
1256
+ },
1257
+ {
1258
+ "clip_ratio": 0.0,
1259
+ "completion_length": 448.359375,
1260
+ "epoch": 1.4201680672268908,
1261
+ "grad_norm": 0.46208083629608154,
1262
+ "kl": 0.11083984375,
1263
+ "learning_rate": 4.778983221388742e-06,
1264
+ "loss": 0.0139,
1265
+ "num_tokens": 11580547.0,
1266
+ "reward": 1.0276041626930237,
1267
+ "reward_std": 0.12672015465795994,
1268
+ "rewards/curriculum_aware_reward_fn": 0.09791666641831398,
1269
+ "rewards/format_reward": 0.9296875,
1270
+ "step": 84
1271
+ },
1272
+ {
1273
+ "clip_ratio": 0.0,
1274
+ "completion_length": 457.3515625,
1275
+ "epoch": 1.4369747899159664,
1276
+ "grad_norm": 0.44321703910827637,
1277
+ "kl": 0.1085205078125,
1278
+ "learning_rate": 4.77264141744214e-06,
1279
+ "loss": 0.0184,
1280
+ "num_tokens": 11712984.0,
1281
+ "reward": 1.160416692495346,
1282
+ "reward_std": 0.15310384705662727,
1283
+ "rewards/curriculum_aware_reward_fn": 0.16041666828095913,
1284
+ "rewards/format_reward": 1.0,
1285
+ "step": 85
1286
+ },
1287
+ {
1288
+ "clip_ratio": 0.0,
1289
+ "completion_length": 384.3984375,
1290
+ "epoch": 1.453781512605042,
1291
+ "grad_norm": 0.4427275061607361,
1292
+ "kl": 0.123779296875,
1293
+ "learning_rate": 4.766214747512603e-06,
1294
+ "loss": 0.0233,
1295
+ "num_tokens": 11829315.0,
1296
+ "reward": 1.1666666865348816,
1297
+ "reward_std": 0.14362479094415903,
1298
+ "rewards/curriculum_aware_reward_fn": 0.1744791674427688,
1299
+ "rewards/format_reward": 0.9921875,
1300
+ "step": 86
1301
+ },
1302
+ {
1303
+ "clip_ratio": 0.0,
1304
+ "completion_length": 389.265625,
1305
+ "epoch": 1.4705882352941178,
1306
+ "grad_norm": 0.5121983885765076,
1307
+ "kl": 0.12841796875,
1308
+ "learning_rate": 4.759703481250331e-06,
1309
+ "loss": 0.0229,
1310
+ "num_tokens": 11949533.0,
1311
+ "reward": 1.1343750357627869,
1312
+ "reward_std": 0.16322745941579342,
1313
+ "rewards/curriculum_aware_reward_fn": 0.1500000013038516,
1314
+ "rewards/format_reward": 0.984375,
1315
+ "step": 87
1316
+ },
1317
+ {
1318
+ "clip_ratio": 0.0,
1319
+ "completion_length": 459.921875,
1320
+ "epoch": 1.4873949579831933,
1321
+ "grad_norm": 0.6308655738830566,
1322
+ "kl": 0.11328125,
1323
+ "learning_rate": 4.753107891855015e-06,
1324
+ "loss": 0.02,
1325
+ "num_tokens": 12092563.0,
1326
+ "reward": 0.8854166716337204,
1327
+ "reward_std": 0.22940894588828087,
1328
+ "rewards/curriculum_aware_reward_fn": 0.11197916697710752,
1329
+ "rewards/format_reward": 0.7734375,
1330
+ "step": 88
1331
+ },
1332
+ {
1333
+ "clip_ratio": 0.0,
1334
+ "completion_length": 325.625,
1335
+ "epoch": 1.504201680672269,
1336
+ "grad_norm": 0.6008861660957336,
1337
+ "kl": 0.14013671875,
1338
+ "learning_rate": 4.746428256064375e-06,
1339
+ "loss": 0.0021,
1340
+ "num_tokens": 12195803.0,
1341
+ "reward": 1.261979192495346,
1342
+ "reward_std": 0.17762075550854206,
1343
+ "rewards/curriculum_aware_reward_fn": 0.2697916701436043,
1344
+ "rewards/format_reward": 0.9921875,
1345
+ "step": 89
1346
+ },
1347
+ {
1348
+ "clip_ratio": 0.0,
1349
+ "completion_length": 348.2421875,
1350
+ "epoch": 1.5210084033613445,
1351
+ "grad_norm": 0.5485793948173523,
1352
+ "kl": 0.13720703125,
1353
+ "learning_rate": 4.7396648541425534e-06,
1354
+ "loss": 0.0053,
1355
+ "num_tokens": 12314866.0,
1356
+ "reward": 1.2442708611488342,
1357
+ "reward_std": 0.21353545226156712,
1358
+ "rewards/curriculum_aware_reward_fn": 0.24427084252238274,
1359
+ "rewards/format_reward": 1.0,
1360
+ "step": 90
1361
+ },
1362
+ {
1363
+ "clip_ratio": 0.0,
1364
+ "completion_length": 519.265625,
1365
+ "epoch": 1.53781512605042,
1366
+ "grad_norm": 0.4276430010795593,
1367
+ "kl": 0.11962890625,
1368
+ "learning_rate": 4.732817969868348e-06,
1369
+ "loss": 0.0315,
1370
+ "num_tokens": 12463604.0,
1371
+ "reward": 0.895833358168602,
1372
+ "reward_std": 0.09582467563450336,
1373
+ "rewards/curriculum_aware_reward_fn": 0.0833333358168602,
1374
+ "rewards/format_reward": 0.8125,
1375
+ "step": 91
1376
+ },
1377
+ {
1378
+ "clip_ratio": 0.0,
1379
+ "completion_length": 334.859375,
1380
+ "epoch": 1.5546218487394958,
1381
+ "grad_norm": 0.5961228013038635,
1382
+ "kl": 0.1279296875,
1383
+ "learning_rate": 4.7258878905233095e-06,
1384
+ "loss": 0.0697,
1385
+ "num_tokens": 12583698.0,
1386
+ "reward": 1.2239583432674408,
1387
+ "reward_std": 0.18167817778885365,
1388
+ "rewards/curriculum_aware_reward_fn": 0.2317708358168602,
1389
+ "rewards/format_reward": 0.9921875,
1390
+ "step": 92
1391
+ },
1392
+ {
1393
+ "clip_ratio": 0.0,
1394
+ "completion_length": 363.859375,
1395
+ "epoch": 1.5714285714285714,
1396
+ "grad_norm": 0.7697988152503967,
1397
+ "kl": 0.1495361328125,
1398
+ "learning_rate": 4.718874906879688e-06,
1399
+ "loss": 0.1313,
1400
+ "num_tokens": 12705640.0,
1401
+ "reward": 1.123437523841858,
1402
+ "reward_std": 0.29410218447446823,
1403
+ "rewards/curriculum_aware_reward_fn": 0.17812500521540642,
1404
+ "rewards/format_reward": 0.9453125,
1405
+ "step": 93
1406
+ },
1407
+ {
1408
+ "clip_ratio": 0.0,
1409
+ "completion_length": 297.1328125,
1410
+ "epoch": 1.5882352941176472,
1411
+ "grad_norm": 0.5211566686630249,
1412
+ "kl": 0.133056640625,
1413
+ "learning_rate": 4.711779313188231e-06,
1414
+ "loss": 0.04,
1415
+ "num_tokens": 12824305.0,
1416
+ "reward": 1.1119791865348816,
1417
+ "reward_std": 0.18469560518860817,
1418
+ "rewards/curriculum_aware_reward_fn": 0.1119791679084301,
1419
+ "rewards/format_reward": 1.0,
1420
+ "step": 94
1421
+ },
1422
+ {
1423
+ "clip_ratio": 0.0,
1424
+ "completion_length": 253.859375,
1425
+ "epoch": 1.6050420168067228,
1426
+ "grad_norm": 0.6686471104621887,
1427
+ "kl": 0.1435546875,
1428
+ "learning_rate": 4.70460140716584e-06,
1429
+ "loss": -0.0138,
1430
+ "num_tokens": 12918559.0,
1431
+ "reward": 1.1968750357627869,
1432
+ "reward_std": 0.1724256370216608,
1433
+ "rewards/curriculum_aware_reward_fn": 0.20468750596046448,
1434
+ "rewards/format_reward": 0.9921875,
1435
+ "step": 95
1436
+ },
1437
+ {
1438
+ "clip_ratio": 0.0,
1439
+ "completion_length": 575.90625,
1440
+ "epoch": 1.6218487394957983,
1441
+ "grad_norm": 0.3436340093612671,
1442
+ "kl": 0.160888671875,
1443
+ "learning_rate": 4.697341489983076e-06,
1444
+ "loss": 0.0223,
1445
+ "num_tokens": 13076699.0,
1446
+ "reward": 0.8567708432674408,
1447
+ "reward_std": 0.08664888702332973,
1448
+ "rewards/curriculum_aware_reward_fn": 0.10677083395421505,
1449
+ "rewards/format_reward": 0.75,
1450
+ "step": 96
1451
+ },
1452
+ {
1453
+ "clip_ratio": 0.0,
1454
+ "completion_length": 301.9140625,
1455
+ "epoch": 1.638655462184874,
1456
+ "grad_norm": 1.0377366542816162,
1457
+ "kl": 0.162353515625,
1458
+ "learning_rate": 4.6899998662515215e-06,
1459
+ "loss": 0.0906,
1460
+ "num_tokens": 13192776.0,
1461
+ "reward": 1.1197916716337204,
1462
+ "reward_std": 0.12685893662273884,
1463
+ "rewards/curriculum_aware_reward_fn": 0.13541666674427688,
1464
+ "rewards/format_reward": 0.984375,
1465
+ "step": 97
1466
+ },
1467
+ {
1468
+ "clip_ratio": 0.0,
1469
+ "completion_length": 331.28125,
1470
+ "epoch": 1.6554621848739495,
1471
+ "grad_norm": 0.477889746427536,
1472
+ "kl": 0.151123046875,
1473
+ "learning_rate": 4.682576844011007e-06,
1474
+ "loss": 0.0147,
1475
+ "num_tokens": 13309884.0,
1476
+ "reward": 1.2942708730697632,
1477
+ "reward_std": 0.10486710164695978,
1478
+ "rewards/curriculum_aware_reward_fn": 0.3567708432674408,
1479
+ "rewards/format_reward": 0.9375,
1480
+ "step": 98
1481
+ },
1482
+ {
1483
+ "clip_ratio": 0.0,
1484
+ "completion_length": 418.96875,
1485
+ "epoch": 1.6722689075630253,
1486
+ "grad_norm": 1.2051795721054077,
1487
+ "kl": 0.1630859375,
1488
+ "learning_rate": 4.675072734716678e-06,
1489
+ "loss": 0.0986,
1490
+ "num_tokens": 13443272.0,
1491
+ "reward": 1.0937500298023224,
1492
+ "reward_std": 0.1639669369906187,
1493
+ "rewards/curriculum_aware_reward_fn": 0.1796875111758709,
1494
+ "rewards/format_reward": 0.9140625,
1495
+ "step": 99
1496
+ },
1497
+ {
1498
+ "clip_ratio": 0.0,
1499
+ "completion_length": 257.484375,
1500
+ "epoch": 1.6890756302521008,
1501
+ "grad_norm": 0.6367622017860413,
1502
+ "kl": 0.157958984375,
1503
+ "learning_rate": 4.667487853225931e-06,
1504
+ "loss": 0.0274,
1505
+ "num_tokens": 13546814.0,
1506
+ "reward": 1.3072916865348816,
1507
+ "reward_std": 0.1563644390553236,
1508
+ "rewards/curriculum_aware_reward_fn": 0.3307291641831398,
1509
+ "rewards/format_reward": 0.9765625,
1510
+ "step": 100
1511
+ },
1512
+ {
1513
+ "epoch": 1.6890756302521008,
1514
+ "step": 100,
1515
+ "total_flos": 0.0,
1516
+ "train_loss": 0.0,
1517
+ "train_runtime": 1.2498,
1518
+ "train_samples_per_second": 1518.669,
1519
+ "train_steps_per_second": 46.408
1520
+ }
1521
+ ],
1522
+ "logging_steps": 1,
1523
+ "max_steps": 58,
1524
+ "num_input_tokens_seen": 0,
1525
+ "num_train_epochs": 2,
1526
+ "save_steps": 50,
1527
+ "stateful_callbacks": {
1528
+ "TrainerControl": {
1529
+ "args": {
1530
+ "should_epoch_stop": false,
1531
+ "should_evaluate": false,
1532
+ "should_log": false,
1533
+ "should_save": true,
1534
+ "should_training_stop": false
1535
+ },
1536
+ "attributes": {}
1537
+ }
1538
+ },
1539
+ "total_flos": 0.0,
1540
+ "train_batch_size": 8,
1541
+ "trial_name": null,
1542
+ "trial_params": null
1543
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df1fd71891db881a097397969327d52f21ed217f14db9c60387e5a9f621777bd
3
+ size 8440
vocab.json ADDED
The diff for this file is too large to render. See raw diff