aawwssdddwdwd commited on
Commit
177c659
·
verified ·
1 Parent(s): 50c668d

End of training

Browse files
README.md CHANGED
@@ -38,7 +38,7 @@ This model was trained with GRPO, a method introduced in [DeepSeekMath: Pushing
38
  ### Framework versions
39
 
40
  - TRL: 0.15.2
41
- - Transformers: 4.50.3
42
  - Pytorch: 2.5.1
43
  - Datasets: 3.5.0
44
  - Tokenizers: 0.21.1
 
38
  ### Framework versions
39
 
40
  - TRL: 0.15.2
41
+ - Transformers: 4.51.3
42
  - Pytorch: 2.5.1
43
  - Datasets: 3.5.0
44
  - Tokenizers: 0.21.1
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 2.1765008568763733e-06,
4
- "train_runtime": 389.9144,
5
- "train_samples": 160,
6
- "train_samples_per_second": 0.821,
7
- "train_steps_per_second": 0.051
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 4.3076954852949715e-06,
4
+ "train_runtime": 535.5837,
5
+ "train_samples": 104,
6
+ "train_samples_per_second": 0.597,
7
+ "train_steps_per_second": 0.037
8
  }
config.json CHANGED
@@ -21,7 +21,7 @@
21
  "sliding_window": 32768,
22
  "tie_word_embeddings": true,
23
  "torch_dtype": "float32",
24
- "transformers_version": "4.50.3",
25
  "use_cache": true,
26
  "use_sliding_window": false,
27
  "vocab_size": 151936
 
21
  "sliding_window": 32768,
22
  "tie_word_embeddings": true,
23
  "torch_dtype": "float32",
24
+ "transformers_version": "4.51.3",
25
  "use_cache": true,
26
  "use_sliding_window": false,
27
  "vocab_size": 151936
generation_config.json CHANGED
@@ -10,5 +10,5 @@
10
  "temperature": 0.7,
11
  "top_k": 20,
12
  "top_p": 0.8,
13
- "transformers_version": "4.50.3"
14
  }
 
10
  "temperature": 0.7,
11
  "top_k": 20,
12
  "top_p": 0.8,
13
+ "transformers_version": "4.51.3"
14
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce55caf9b32891b1be461781f4025f8943aa70196c8974e5d2b9ac4cc7a61a97
3
  size 1976163472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b628e031b7b0f2dceb08abad5826a1cd769a637820ca38d014570674461496b
3
  size 1976163472
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 2.1765008568763733e-06,
4
- "train_runtime": 389.9144,
5
- "train_samples": 160,
6
- "train_samples_per_second": 0.821,
7
- "train_steps_per_second": 0.051
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 4.3076954852949715e-06,
4
+ "train_runtime": 535.5837,
5
+ "train_samples": 104,
6
+ "train_samples_per_second": 0.597,
7
+ "train_steps_per_second": 0.037
8
  }
trainer_state.json CHANGED
@@ -2,7 +2,7 @@
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": 20,
8
  "is_hyper_param_search": false,
@@ -10,209 +10,209 @@
10
  "is_world_process_zero": true,
11
  "log_history": [
12
  {
13
- "completion_length": 358.8125,
14
- "epoch": 0.1,
15
- "grad_norm": 5.128148078918457,
16
  "kl": 0.0,
17
- "learning_rate": 4.965903258506806e-07,
18
- "loss": 0.0,
19
- "reward": 1.920902309473604,
20
- "reward_std": 1.5184391534421593,
21
- "rewards/concensus_correctness_reward_func": 0.3513750033453107,
22
- "rewards/consensus_reward_func": 0.4375,
23
  "rewards/cumulative_reward_2": 0.0,
24
- "rewards/final_correctness_reward_func": 0.25,
25
- "rewards/question_recreation_reward_func": 0.41521482449024916,
26
  "rewards/soft_format_reward_func": 0.0,
27
  "rewards/strict_format_reward_func": 0.0,
28
- "rewards/xmlcount_reward_func": 0.466812496073544,
29
  "step": 2
30
  },
31
  {
32
- "completion_length": 326.90625,
33
- "epoch": 0.2,
34
- "grad_norm": 4.2698469161987305,
35
- "kl": 0.0012383201719785575,
36
- "learning_rate": 4.698684378016222e-07,
37
  "loss": 0.0,
38
- "reward": 1.7122610099613667,
39
- "reward_std": 1.1609715959639288,
40
- "rewards/concensus_correctness_reward_func": 0.23168749827891588,
41
- "rewards/consensus_reward_func": 0.625,
42
  "rewards/cumulative_reward_2": 0.0,
43
- "rewards/final_correctness_reward_func": 0.0,
44
- "rewards/question_recreation_reward_func": 0.5111047318205237,
45
  "rewards/soft_format_reward_func": 0.0,
46
  "rewards/strict_format_reward_func": 0.0,
47
- "rewards/xmlcount_reward_func": 0.3444687547162175,
48
  "step": 4
49
  },
50
  {
51
- "completion_length": 418.59375,
52
- "epoch": 0.3,
53
- "grad_norm": 5.097483158111572,
54
- "kl": 0.0015109705473150825,
55
- "learning_rate": 4.193203929064353e-07,
56
  "loss": 0.0,
57
- "reward": 1.2479718840913847,
58
- "reward_std": 0.6527503429970238,
59
- "rewards/concensus_correctness_reward_func": 0.07975000143051147,
60
- "rewards/consensus_reward_func": 0.25,
61
  "rewards/cumulative_reward_2": 0.0,
62
- "rewards/final_correctness_reward_func": 0.0625,
63
- "rewards/question_recreation_reward_func": 0.4569093860918656,
64
  "rewards/soft_format_reward_func": 0.0,
65
  "rewards/strict_format_reward_func": 0.0,
66
- "rewards/xmlcount_reward_func": 0.3988124942407012,
67
  "step": 6
68
  },
69
  {
70
- "completion_length": 396.53125,
71
- "epoch": 0.4,
72
- "grad_norm": 4.257869243621826,
73
- "kl": 0.0019715054659172893,
74
- "learning_rate": 3.5042385616324236e-07,
75
  "loss": 0.0,
76
- "reward": 1.635931245982647,
77
- "reward_std": 1.020388813689351,
78
- "rewards/concensus_correctness_reward_func": 0.14943750109523535,
79
- "rewards/consensus_reward_func": 0.4375,
80
  "rewards/cumulative_reward_2": 0.0,
81
- "rewards/final_correctness_reward_func": 0.1875,
82
- "rewards/question_recreation_reward_func": 0.48218128900043666,
83
  "rewards/soft_format_reward_func": 0.0,
84
- "rewards/strict_format_reward_func": 0.0,
85
- "rewards/xmlcount_reward_func": 0.3793124984949827,
86
  "step": 8
87
  },
88
  {
89
- "completion_length": 419.875,
90
- "epoch": 0.5,
91
- "grad_norm": 3.371570110321045,
92
- "kl": 0.00194889300109935,
93
- "learning_rate": 2.706448363680831e-07,
94
  "loss": 0.0,
95
- "reward": 2.2834888957440853,
96
- "reward_std": 1.2663272079080343,
97
- "rewards/concensus_correctness_reward_func": 0.40874999947845936,
98
- "rewards/consensus_reward_func": 0.375,
99
  "rewards/cumulative_reward_2": 0.0,
100
- "rewards/final_correctness_reward_func": 0.25,
101
- "rewards/question_recreation_reward_func": 0.5105513697490096,
102
  "rewards/soft_format_reward_func": 0.0,
103
  "rewards/strict_format_reward_func": 0.0,
104
- "rewards/xmlcount_reward_func": 0.7391874864697456,
105
  "step": 10
106
  },
107
  {
108
- "completion_length": 422.03125,
109
- "epoch": 0.6,
110
- "grad_norm": 4.007848739624023,
111
- "kl": 0.0029282994655659422,
112
- "learning_rate": 1.886286282148002e-07,
113
  "loss": 0.0,
114
- "reward": 1.628133011981845,
115
- "reward_std": 1.141701115295291,
116
- "rewards/concensus_correctness_reward_func": 0.2252500019967556,
117
- "rewards/consensus_reward_func": 0.25,
118
  "rewards/cumulative_reward_2": 0.0,
119
  "rewards/final_correctness_reward_func": 0.125,
120
- "rewards/question_recreation_reward_func": 0.5204767659306526,
121
  "rewards/soft_format_reward_func": 0.0,
122
  "rewards/strict_format_reward_func": 0.0,
123
- "rewards/xmlcount_reward_func": 0.507406250697386,
124
  "step": 12
125
  },
126
  {
127
- "completion_length": 453.9375,
128
- "epoch": 0.7,
129
- "grad_norm": 3.688523054122925,
130
- "kl": 0.0023045147099765018,
131
- "learning_rate": 1.1326296046939333e-07,
132
  "loss": 0.0,
133
- "reward": 1.8677344880998135,
134
- "reward_std": 1.1840693070553243,
135
- "rewards/concensus_correctness_reward_func": 0.3175625018775463,
136
- "rewards/consensus_reward_func": 0.25,
137
  "rewards/cumulative_reward_2": 0.0,
138
- "rewards/final_correctness_reward_func": 0.1875,
139
- "rewards/question_recreation_reward_func": 0.6593594700098038,
140
  "rewards/soft_format_reward_func": 0.0,
141
  "rewards/strict_format_reward_func": 0.0,
142
- "rewards/xmlcount_reward_func": 0.45331250689923763,
143
  "step": 14
144
  },
145
  {
146
- "completion_length": 345.3125,
147
- "epoch": 0.8,
148
- "grad_norm": 4.259862899780273,
149
- "kl": 0.002908603215473704,
150
- "learning_rate": 5.271487265090163e-08,
151
  "loss": 0.0,
152
- "reward": 2.4872301314026117,
153
- "reward_std": 2.5430639277910814,
154
- "rewards/concensus_correctness_reward_func": 0.8553749984130263,
155
- "rewards/consensus_reward_func": 0.4375,
156
  "rewards/cumulative_reward_2": 0.0,
157
- "rewards/final_correctness_reward_func": 0.1875,
158
- "rewards/question_recreation_reward_func": 0.5011364271049388,
159
- "rewards/soft_format_reward_func": 0.015625,
160
  "rewards/strict_format_reward_func": 0.015625,
161
- "rewards/xmlcount_reward_func": 0.47446875739842653,
162
  "step": 16
163
  },
164
  {
165
- "completion_length": 343.9375,
166
- "epoch": 0.9,
167
- "grad_norm": 4.737185955047607,
168
- "kl": 0.00293535369928577,
169
- "learning_rate": 1.3545689574841341e-08,
170
  "loss": 0.0,
171
- "reward": 1.9544069655239582,
172
- "reward_std": 0.9897497426718473,
173
- "rewards/concensus_correctness_reward_func": 0.20006249845027924,
174
- "rewards/consensus_reward_func": 0.3125,
175
  "rewards/cumulative_reward_2": 0.0,
176
- "rewards/final_correctness_reward_func": 0.25,
177
- "rewards/question_recreation_reward_func": 0.43893826520070434,
178
  "rewards/soft_format_reward_func": 0.0,
179
  "rewards/strict_format_reward_func": 0.0,
180
- "rewards/xmlcount_reward_func": 0.7529062535613775,
181
  "step": 18
182
  },
183
  {
184
- "completion_length": 290.25,
185
- "epoch": 1.0,
186
- "grad_norm": 4.982654094696045,
187
- "kl": 0.0030415507935686037,
188
- "learning_rate": 0.0,
189
  "loss": 0.0,
190
- "reward": 2.1933444906026125,
191
- "reward_std": 1.347527216479648,
192
- "rewards/concensus_correctness_reward_func": 0.4152499958872795,
193
- "rewards/consensus_reward_func": 0.375,
194
  "rewards/cumulative_reward_2": 0.0,
195
- "rewards/final_correctness_reward_func": 0.1875,
196
- "rewards/question_recreation_reward_func": 0.5650007203221321,
197
  "rewards/soft_format_reward_func": 0.0,
198
  "rewards/strict_format_reward_func": 0.0,
199
- "rewards/xmlcount_reward_func": 0.6505937650799751,
200
  "step": 20
201
  },
202
  {
203
- "epoch": 1.0,
204
  "step": 20,
205
  "total_flos": 0.0,
206
- "train_loss": 2.1765008568763733e-06,
207
- "train_runtime": 389.9144,
208
- "train_samples_per_second": 0.821,
209
- "train_steps_per_second": 0.051
210
  }
211
  ],
212
  "logging_steps": 2,
213
  "max_steps": 20,
214
  "num_input_tokens_seen": 0,
215
- "num_train_epochs": 1,
216
  "save_steps": 25,
217
  "stateful_callbacks": {
218
  "TrainerControl": {
 
2
  "best_global_step": null,
3
  "best_metric": null,
4
  "best_model_checkpoint": null,
5
+ "epoch": 1.5384615384615383,
6
  "eval_steps": 500,
7
  "global_step": 20,
8
  "is_hyper_param_search": false,
 
10
  "is_world_process_zero": true,
11
  "log_history": [
12
  {
13
+ "completion_length": 335.125,
14
+ "epoch": 0.15384615384615385,
15
+ "grad_norm": 113.76084899902344,
16
  "kl": 0.0,
17
+ "learning_rate": 5e-07,
18
+ "loss": -0.0,
19
+ "reward": 0.5326231555081904,
20
+ "reward_std": 0.39712623693048954,
21
+ "rewards/concensus_correctness_reward_func": 0.0,
22
+ "rewards/consensus_reward_func": 0.0,
23
  "rewards/cumulative_reward_2": 0.0,
24
+ "rewards/final_correctness_reward_func": 0.0625,
25
+ "rewards/question_recreation_reward_func": 0.28109190798568306,
26
  "rewards/soft_format_reward_func": 0.0,
27
  "rewards/strict_format_reward_func": 0.0,
28
+ "rewards/xmlcount_reward_func": 0.18903125426732004,
29
  "step": 2
30
  },
31
  {
32
+ "completion_length": 222.0625,
33
+ "epoch": 0.3076923076923077,
34
+ "grad_norm": 167.0595245361328,
35
+ "kl": 0.0030416610734391725,
36
+ "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
+ "reward": 0.2555721157696098,
39
+ "reward_std": 0.3549694743414875,
40
+ "rewards/concensus_correctness_reward_func": 0.0,
41
+ "rewards/consensus_reward_func": 0.0,
42
  "rewards/cumulative_reward_2": 0.0,
43
+ "rewards/final_correctness_reward_func": 0.0625,
44
+ "rewards/question_recreation_reward_func": 0.1653533645439893,
45
  "rewards/soft_format_reward_func": 0.0,
46
  "rewards/strict_format_reward_func": 0.0,
47
+ "rewards/xmlcount_reward_func": 0.02771875075995922,
48
  "step": 4
49
  },
50
  {
51
+ "completion_length": 354.625,
52
+ "epoch": 0.46153846153846156,
53
+ "grad_norm": 6.337334156036377,
54
+ "kl": 0.0010862785820791032,
55
+ "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
+ "reward": 0.11460426822304726,
58
+ "reward_std": 0.7663493577274494,
59
+ "rewards/concensus_correctness_reward_func": 0.0,
60
+ "rewards/consensus_reward_func": 0.0625,
61
  "rewards/cumulative_reward_2": 0.0,
62
+ "rewards/final_correctness_reward_func": 0.0,
63
+ "rewards/question_recreation_reward_func": 0.19647929281927645,
64
  "rewards/soft_format_reward_func": 0.0,
65
  "rewards/strict_format_reward_func": 0.0,
66
+ "rewards/xmlcount_reward_func": -0.14437501080101356,
67
  "step": 6
68
  },
69
  {
70
+ "completion_length": 286.375,
71
+ "epoch": 0.6153846153846154,
72
+ "grad_norm": 14.78044319152832,
73
+ "kl": 0.0018513284521759488,
74
+ "learning_rate": 3.867370395306068e-07,
75
  "loss": 0.0,
76
+ "reward": 0.5740392594598234,
77
+ "reward_std": 0.5501163068693131,
78
+ "rewards/concensus_correctness_reward_func": 0.0,
79
+ "rewards/consensus_reward_func": 0.0625,
80
  "rewards/cumulative_reward_2": 0.0,
81
+ "rewards/final_correctness_reward_func": 0.125,
82
+ "rewards/question_recreation_reward_func": 0.2515392726054415,
83
  "rewards/soft_format_reward_func": 0.0,
84
+ "rewards/strict_format_reward_func": 0.015625,
85
+ "rewards/xmlcount_reward_func": 0.11937499977648258,
86
  "step": 8
87
  },
88
  {
89
+ "completion_length": 322.875,
90
+ "epoch": 0.7692307692307693,
91
+ "grad_norm": 23.10611343383789,
92
+ "kl": 0.0021936903613095637,
93
+ "learning_rate": 3.1137137178519977e-07,
94
  "loss": 0.0,
95
+ "reward": 0.21336799208074808,
96
+ "reward_std": 0.44811401196056977,
97
+ "rewards/concensus_correctness_reward_func": 0.0,
98
+ "rewards/consensus_reward_func": 0.0,
99
  "rewards/cumulative_reward_2": 0.0,
100
+ "rewards/final_correctness_reward_func": 0.0,
101
+ "rewards/question_recreation_reward_func": 0.258524238714017,
102
  "rewards/soft_format_reward_func": 0.0,
103
  "rewards/strict_format_reward_func": 0.0,
104
+ "rewards/xmlcount_reward_func": -0.045156255131587386,
105
  "step": 10
106
  },
107
  {
108
+ "completion_length": 305.875,
109
+ "epoch": 0.9230769230769231,
110
+ "grad_norm": 13.471948623657227,
111
+ "kl": 0.0015927579770504963,
112
+ "learning_rate": 2.2935516363191693e-07,
113
  "loss": 0.0,
114
+ "reward": 0.25780566956382245,
115
+ "reward_std": 0.5996597472112626,
116
+ "rewards/concensus_correctness_reward_func": 0.0,
117
+ "rewards/consensus_reward_func": 0.0,
118
  "rewards/cumulative_reward_2": 0.0,
119
  "rewards/final_correctness_reward_func": 0.125,
120
+ "rewards/question_recreation_reward_func": 0.19636816903948784,
121
  "rewards/soft_format_reward_func": 0.0,
122
  "rewards/strict_format_reward_func": 0.0,
123
+ "rewards/xmlcount_reward_func": -0.06356249563395977,
124
  "step": 12
125
  },
126
  {
127
+ "completion_length": 356.90625,
128
+ "epoch": 1.0769230769230769,
129
+ "grad_norm": 7.529191970825195,
130
+ "kl": 0.005199393173825229,
131
+ "learning_rate": 1.4957614383675767e-07,
132
  "loss": 0.0,
133
+ "reward": 0.5008027306757867,
134
+ "reward_std": 0.6540710940025747,
135
+ "rewards/concensus_correctness_reward_func": 0.0,
136
+ "rewards/consensus_reward_func": 0.125,
137
  "rewards/cumulative_reward_2": 0.0,
138
+ "rewards/final_correctness_reward_func": 0.0625,
139
+ "rewards/question_recreation_reward_func": 0.2908964797388762,
140
  "rewards/soft_format_reward_func": 0.0,
141
  "rewards/strict_format_reward_func": 0.0,
142
+ "rewards/xmlcount_reward_func": 0.02240624837577343,
143
  "step": 14
144
  },
145
  {
146
+ "completion_length": 282.46875,
147
+ "epoch": 1.2307692307692308,
148
+ "grad_norm": 11.132549285888672,
149
+ "kl": 0.0023966655353433453,
150
+ "learning_rate": 8.067960709356478e-08,
151
  "loss": 0.0,
152
+ "reward": 0.4687036501709372,
153
+ "reward_std": 0.495733046438545,
154
+ "rewards/concensus_correctness_reward_func": 0.0,
155
+ "rewards/consensus_reward_func": 0.0,
156
  "rewards/cumulative_reward_2": 0.0,
157
+ "rewards/final_correctness_reward_func": 0.0625,
158
+ "rewards/question_recreation_reward_func": 0.26454739971086383,
159
+ "rewards/soft_format_reward_func": 0.0,
160
  "rewards/strict_format_reward_func": 0.015625,
161
+ "rewards/xmlcount_reward_func": 0.12603125057648867,
162
  "step": 16
163
  },
164
  {
165
+ "completion_length": 169.96875,
166
+ "epoch": 1.3846153846153846,
167
+ "grad_norm": 34.29732131958008,
168
+ "kl": 0.0047469171695411205,
169
+ "learning_rate": 3.013156219837776e-08,
170
  "loss": 0.0,
171
+ "reward": 0.3130655135028064,
172
+ "reward_std": 0.3251972365542315,
173
+ "rewards/concensus_correctness_reward_func": 0.0,
174
+ "rewards/consensus_reward_func": 0.0,
175
  "rewards/cumulative_reward_2": 0.0,
176
+ "rewards/final_correctness_reward_func": 0.0625,
177
+ "rewards/question_recreation_reward_func": 0.17259676140383817,
178
  "rewards/soft_format_reward_func": 0.0,
179
  "rewards/strict_format_reward_func": 0.0,
180
+ "rewards/xmlcount_reward_func": 0.0779687506146729,
181
  "step": 18
182
  },
183
  {
184
+ "completion_length": 305.9375,
185
+ "epoch": 1.5384615384615383,
186
+ "grad_norm": 66.23303985595703,
187
+ "kl": 0.021018446917878464,
188
+ "learning_rate": 3.4096741493194193e-09,
189
  "loss": 0.0,
190
+ "reward": 0.21660699727362953,
191
+ "reward_std": 0.2998809102573432,
192
+ "rewards/concensus_correctness_reward_func": 0.0,
193
+ "rewards/consensus_reward_func": 0.0,
194
  "rewards/cumulative_reward_2": 0.0,
195
+ "rewards/final_correctness_reward_func": 0.0,
196
+ "rewards/question_recreation_reward_func": 0.14573199834558181,
197
  "rewards/soft_format_reward_func": 0.0,
198
  "rewards/strict_format_reward_func": 0.0,
199
+ "rewards/xmlcount_reward_func": 0.07087500323541462,
200
  "step": 20
201
  },
202
  {
203
+ "epoch": 1.5384615384615383,
204
  "step": 20,
205
  "total_flos": 0.0,
206
+ "train_loss": 4.3076954852949715e-06,
207
+ "train_runtime": 535.5837,
208
+ "train_samples_per_second": 0.597,
209
+ "train_steps_per_second": 0.037
210
  }
211
  ],
212
  "logging_steps": 2,
213
  "max_steps": 20,
214
  "num_input_tokens_seen": 0,
215
+ "num_train_epochs": 2,
216
  "save_steps": 25,
217
  "stateful_callbacks": {
218
  "TrainerControl": {
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5c4591d867a8fa7315cfbd122d0fc772738fe97893db9e19c17c8ef3d5ac990d
3
- size 6008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0c5a3fafa9d09ed82a2894e2a76e52e3e3c219a83ca4c86615bf3934872d12a
3
+ size 5944