sergisimi commited on
Commit
a030675
·
verified ·
1 Parent(s): 8ba3f7b

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.51.0
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.5179237127304076e-06,
4
- "train_runtime": 1425.2641,
5
- "train_samples": 40,
6
- "train_samples_per_second": 0.225,
7
- "train_steps_per_second": 0.014
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 4.036165773868561e-06,
4
+ "train_runtime": 203.1494,
5
+ "train_samples": 28,
6
+ "train_samples_per_second": 1.575,
7
+ "train_steps_per_second": 0.098
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.51.0",
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.51.0"
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:ef6f8fd882018c2029ed9817c1fe3c188df890b2bfcb39f1ef9a84f855449f8c
3
  size 1976163472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a39febe3f60d310eff0a5798a7b17b97a8078c436d4acaf77cbeacaf30e6ed9
3
  size 1976163472
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 2.5179237127304076e-06,
4
- "train_runtime": 1425.2641,
5
- "train_samples": 40,
6
- "train_samples_per_second": 0.225,
7
- "train_steps_per_second": 0.014
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 4.036165773868561e-06,
4
+ "train_runtime": 203.1494,
5
+ "train_samples": 28,
6
+ "train_samples_per_second": 1.575,
7
+ "train_steps_per_second": 0.098
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": 4.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": 376.78125,
14
- "epoch": 0.4,
15
- "grad_norm": 9.419937133789062,
16
  "kl": 0.0,
17
  "learning_rate": 5e-07,
18
- "loss": 0.0,
19
- "reward": 0.8982328362762928,
20
- "reward_std": 0.8957028803415596,
21
- "rewards/concensus_correctness_reward_func": 0.12150000035762787,
22
- "rewards/consensus_reward_func": 0.125,
23
  "rewards/cumulative_reward_2": 0.0,
24
- "rewards/final_correctness_reward_func": 0.0,
25
- "rewards/question_recreation_reward_func": 0.28982658532913774,
26
- "rewards/soft_format_reward_func": 0.0,
27
  "rewards/strict_format_reward_func": 0.0,
28
- "rewards/xmlcount_reward_func": 0.36190625093877316,
29
  "step": 2
30
  },
31
  {
32
- "completion_length": 303.0625,
33
- "epoch": 0.8,
34
- "grad_norm": 16.360496520996094,
35
- "kl": 0.0018617127680045087,
36
  "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
- "reward": 1.9974666351918131,
39
- "reward_std": 2.0806010872038314,
40
- "rewards/concensus_correctness_reward_func": 0.8354999963194132,
41
- "rewards/consensus_reward_func": 0.1875,
42
  "rewards/cumulative_reward_2": 0.0,
43
- "rewards/final_correctness_reward_func": 0.125,
44
- "rewards/question_recreation_reward_func": 0.40974785340949893,
45
  "rewards/soft_format_reward_func": 0.0,
46
- "rewards/strict_format_reward_func": 0.03125,
47
- "rewards/xmlcount_reward_func": 0.40846874634735286,
48
  "step": 4
49
  },
50
  {
51
- "completion_length": 338.5625,
52
- "epoch": 1.2,
53
- "grad_norm": 11.249113082885742,
54
- "kl": 0.0014679385494673625,
55
  "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
- "reward": 1.0521508090896532,
58
- "reward_std": 0.8584795482747722,
59
- "rewards/concensus_correctness_reward_func": 0.14687499776482582,
60
- "rewards/consensus_reward_func": 0.1875,
61
  "rewards/cumulative_reward_2": 0.0,
62
- "rewards/final_correctness_reward_func": 0.0,
63
- "rewards/question_recreation_reward_func": 0.33327579603064805,
64
  "rewards/soft_format_reward_func": 0.0,
65
- "rewards/strict_format_reward_func": 0.03125,
66
- "rewards/xmlcount_reward_func": 0.35324999620206654,
67
  "step": 6
68
  },
69
  {
70
- "completion_length": 375.09375,
71
- "epoch": 1.6,
72
- "grad_norm": 5.771604537963867,
73
- "kl": 0.001528985405457206,
74
  "learning_rate": 3.867370395306068e-07,
75
  "loss": 0.0,
76
- "reward": 1.6449143970385194,
77
- "reward_std": 1.950113006751053,
78
- "rewards/concensus_correctness_reward_func": 0.7482500001788139,
79
- "rewards/consensus_reward_func": 0.125,
80
  "rewards/cumulative_reward_2": 0.0,
81
- "rewards/final_correctness_reward_func": 0.125,
82
- "rewards/question_recreation_reward_func": 0.4411331639857963,
83
  "rewards/soft_format_reward_func": 0.0,
84
- "rewards/strict_format_reward_func": 0.03125,
85
- "rewards/xmlcount_reward_func": 0.1742812478914857,
86
  "step": 8
87
  },
88
  {
89
- "completion_length": 337.65625,
90
- "epoch": 2.0,
91
- "grad_norm": 8.124749183654785,
92
- "kl": 0.004178064547886606,
93
  "learning_rate": 3.1137137178519977e-07,
94
  "loss": 0.0,
95
- "reward": 1.1003415104933083,
96
- "reward_std": 0.8792835858184844,
97
- "rewards/concensus_correctness_reward_func": 0.05999999865889549,
98
- "rewards/consensus_reward_func": 0.0625,
99
  "rewards/cumulative_reward_2": 0.0,
100
- "rewards/final_correctness_reward_func": 0.0,
101
- "rewards/question_recreation_reward_func": 0.3608102686703205,
102
  "rewards/soft_format_reward_func": 0.0,
103
- "rewards/strict_format_reward_func": 0.015625,
104
- "rewards/xmlcount_reward_func": 0.6014062347821891,
105
  "step": 10
106
  },
107
  {
108
- "completion_length": 344.0625,
109
- "epoch": 2.4,
110
- "grad_norm": 8.546586990356445,
111
- "kl": 0.004825560077733826,
112
  "learning_rate": 2.2935516363191693e-07,
113
  "loss": 0.0,
114
- "reward": 0.5016123706009239,
115
- "reward_std": 0.663908442016691,
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.0,
120
- "rewards/question_recreation_reward_func": 0.3652373746735975,
121
  "rewards/soft_format_reward_func": 0.0,
122
  "rewards/strict_format_reward_func": 0.0,
123
- "rewards/xmlcount_reward_func": 0.1363750018645078,
124
  "step": 12
125
  },
126
  {
127
- "completion_length": 383.625,
128
- "epoch": 2.8,
129
- "grad_norm": 4.618159294128418,
130
- "kl": 0.0030358081712620333,
131
  "learning_rate": 1.4957614383675767e-07,
132
  "loss": 0.0,
133
- "reward": 0.4713071408914402,
134
- "reward_std": 0.42544175022339914,
135
- "rewards/concensus_correctness_reward_func": 0.0,
136
- "rewards/consensus_reward_func": 0.0,
137
  "rewards/cumulative_reward_2": 0.0,
138
- "rewards/final_correctness_reward_func": 0.0,
139
- "rewards/question_recreation_reward_func": 0.2834946505026892,
140
  "rewards/soft_format_reward_func": 0.0,
141
- "rewards/strict_format_reward_func": 0.015625,
142
- "rewards/xmlcount_reward_func": 0.17218750156462193,
143
  "step": 14
144
  },
145
  {
146
- "completion_length": 370.90625,
147
- "epoch": 3.2,
148
- "grad_norm": 7.157219886779785,
149
- "kl": 0.003981857269536704,
150
  "learning_rate": 8.067960709356478e-08,
151
  "loss": 0.0,
152
- "reward": 1.6145572238601744,
153
- "reward_std": 1.8933260106496164,
154
- "rewards/concensus_correctness_reward_func": 0.7465000003576279,
155
- "rewards/consensus_reward_func": 0.125,
156
  "rewards/cumulative_reward_2": 0.0,
157
- "rewards/final_correctness_reward_func": 0.0625,
158
- "rewards/question_recreation_reward_func": 0.37758851423859596,
159
  "rewards/soft_format_reward_func": 0.0,
160
  "rewards/strict_format_reward_func": 0.0,
161
- "rewards/xmlcount_reward_func": 0.3029687451198697,
162
  "step": 16
163
  },
164
  {
165
- "completion_length": 393.28125,
166
- "epoch": 3.6,
167
- "grad_norm": 5.100037097930908,
168
- "kl": 0.002306824164406862,
169
  "learning_rate": 3.013156219837776e-08,
170
  "loss": 0.0,
171
- "reward": 1.2386403833515942,
172
- "reward_std": 1.3317663265625015,
173
- "rewards/concensus_correctness_reward_func": 0.625,
174
- "rewards/consensus_reward_func": 0.0,
175
  "rewards/cumulative_reward_2": 0.0,
176
- "rewards/final_correctness_reward_func": 0.0,
177
- "rewards/question_recreation_reward_func": 0.2853903556242585,
178
  "rewards/soft_format_reward_func": 0.0,
179
- "rewards/strict_format_reward_func": 0.015625,
180
- "rewards/xmlcount_reward_func": 0.3126249979250133,
181
  "step": 18
182
  },
183
  {
184
- "completion_length": 438.71875,
185
- "epoch": 4.0,
186
- "grad_norm": 6.435215950012207,
187
- "kl": 0.002064306852844311,
188
  "learning_rate": 3.4096741493194193e-09,
189
  "loss": 0.0,
190
- "reward": 1.010270037688315,
191
- "reward_std": 1.0351053967606276,
192
- "rewards/concensus_correctness_reward_func": 0.12162499874830246,
193
- "rewards/consensus_reward_func": 0.25,
194
  "rewards/cumulative_reward_2": 0.0,
195
- "rewards/final_correctness_reward_func": 0.0625,
196
- "rewards/question_recreation_reward_func": 0.4003325153607875,
197
  "rewards/soft_format_reward_func": 0.0,
198
  "rewards/strict_format_reward_func": 0.0,
199
- "rewards/xmlcount_reward_func": 0.1758125089108944,
200
  "step": 20
201
  },
202
  {
203
- "epoch": 4.0,
204
  "step": 20,
205
  "total_flos": 0.0,
206
- "train_loss": 2.5179237127304076e-06,
207
- "train_runtime": 1425.2641,
208
- "train_samples_per_second": 0.225,
209
- "train_steps_per_second": 0.014
210
  }
211
  ],
212
  "logging_steps": 2,
213
  "max_steps": 20,
214
  "num_input_tokens_seen": 0,
215
- "num_train_epochs": 4,
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": 5.0,
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": 344.03125,
14
+ "epoch": 0.5714285714285714,
15
+ "grad_norm": 6.132088661193848,
16
  "kl": 0.0,
17
  "learning_rate": 5e-07,
18
+ "loss": -0.0,
19
+ "reward": 2.487778352573514,
20
+ "reward_std": 2.006606017239392,
21
+ "rewards/concensus_correctness_reward_func": 0.8620624975301325,
22
+ "rewards/consensus_reward_func": 0.25,
23
  "rewards/cumulative_reward_2": 0.0,
24
+ "rewards/final_correctness_reward_func": 0.25,
25
+ "rewards/question_recreation_reward_func": 0.6062783699599095,
26
+ "rewards/soft_format_reward_func": 0.015625,
27
  "rewards/strict_format_reward_func": 0.0,
28
+ "rewards/xmlcount_reward_func": 0.5038125060964376,
29
  "step": 2
30
  },
31
  {
32
+ "completion_length": 341.0,
33
+ "epoch": 1.0,
34
+ "grad_norm": 3.1462059020996094,
35
+ "kl": 0.0008595892310646983,
36
  "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
+ "reward": 1.8018831710020702,
39
+ "reward_std": 1.0011655843506257,
40
+ "rewards/concensus_correctness_reward_func": 0.23699999849001566,
41
+ "rewards/consensus_reward_func": 0.25,
42
  "rewards/cumulative_reward_2": 0.0,
43
+ "rewards/final_correctness_reward_func": 0.16666666666666666,
44
+ "rewards/question_recreation_reward_func": 0.6140498152623574,
45
  "rewards/soft_format_reward_func": 0.0,
46
+ "rewards/strict_format_reward_func": 0.0,
47
+ "rewards/xmlcount_reward_func": 0.5341666688521703,
48
  "step": 4
49
  },
50
  {
51
+ "completion_length": 297.71875,
52
+ "epoch": 1.5714285714285714,
53
+ "grad_norm": 6.190745830535889,
54
+ "kl": 0.0016320943250320852,
55
  "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
+ "reward": 1.9226003549993038,
58
+ "reward_std": 1.0635974742472172,
59
+ "rewards/concensus_correctness_reward_func": 0.24456249782815576,
60
+ "rewards/consensus_reward_func": 0.25,
61
  "rewards/cumulative_reward_2": 0.0,
62
+ "rewards/final_correctness_reward_func": 0.1875,
63
+ "rewards/question_recreation_reward_func": 0.5560690779238939,
64
  "rewards/soft_format_reward_func": 0.0,
65
+ "rewards/strict_format_reward_func": 0.0,
66
+ "rewards/xmlcount_reward_func": 0.6844687629491091,
67
  "step": 6
68
  },
69
  {
70
+ "completion_length": 314.9166666666667,
71
+ "epoch": 2.0,
72
+ "grad_norm": 3.6605167388916016,
73
+ "kl": 0.0024019686485795924,
74
  "learning_rate": 3.867370395306068e-07,
75
  "loss": 0.0,
76
+ "reward": 2.3699574917554855,
77
+ "reward_std": 1.2540641278028488,
78
+ "rewards/concensus_correctness_reward_func": 0.5460833261410395,
79
+ "rewards/consensus_reward_func": 0.3333333333333333,
80
  "rewards/cumulative_reward_2": 0.0,
81
+ "rewards/final_correctness_reward_func": 0.3333333333333333,
82
+ "rewards/question_recreation_reward_func": 0.5782075046251217,
83
  "rewards/soft_format_reward_func": 0.0,
84
+ "rewards/strict_format_reward_func": 0.0,
85
+ "rewards/xmlcount_reward_func": 0.5790000098447005,
86
  "step": 8
87
  },
88
  {
89
+ "completion_length": 311.96875,
90
+ "epoch": 2.571428571428571,
91
+ "grad_norm": 4.422511100769043,
92
+ "kl": 0.004271938196325209,
93
  "learning_rate": 3.1137137178519977e-07,
94
  "loss": 0.0,
95
+ "reward": 2.2210648357868195,
96
+ "reward_std": 1.0689412193605676,
97
+ "rewards/concensus_correctness_reward_func": 0.41612499207258224,
98
+ "rewards/consensus_reward_func": 0.25,
99
  "rewards/cumulative_reward_2": 0.0,
100
+ "rewards/final_correctness_reward_func": 0.3125,
101
+ "rewards/question_recreation_reward_func": 0.5958773451857269,
102
  "rewards/soft_format_reward_func": 0.0,
103
+ "rewards/strict_format_reward_func": 0.0,
104
+ "rewards/xmlcount_reward_func": 0.6465625241398811,
105
  "step": 10
106
  },
107
  {
108
+ "completion_length": 269.7083333333333,
109
+ "epoch": 3.0,
110
+ "grad_norm": 4.302887439727783,
111
+ "kl": 0.006312902851883943,
112
  "learning_rate": 2.2935516363191693e-07,
113
  "loss": 0.0,
114
+ "reward": 1.8300592303276062,
115
+ "reward_std": 1.2033961129685242,
116
+ "rewards/concensus_correctness_reward_func": 0.2336666683355967,
117
+ "rewards/consensus_reward_func": 0.25,
118
  "rewards/cumulative_reward_2": 0.0,
119
+ "rewards/final_correctness_reward_func": 0.08333333333333333,
120
+ "rewards/question_recreation_reward_func": 0.5991008977095286,
121
  "rewards/soft_format_reward_func": 0.0,
122
  "rewards/strict_format_reward_func": 0.0,
123
+ "rewards/xmlcount_reward_func": 0.6639583297073841,
124
  "step": 12
125
  },
126
  {
127
+ "completion_length": 314.96875,
128
+ "epoch": 3.571428571428571,
129
+ "grad_norm": 4.885258197784424,
130
+ "kl": 0.005343124037608504,
131
  "learning_rate": 1.4957614383675767e-07,
132
  "loss": 0.0,
133
+ "reward": 3.495531538501382,
134
+ "reward_std": 0.7231066498789005,
135
+ "rewards/concensus_correctness_reward_func": 1.6215624958276749,
136
+ "rewards/consensus_reward_func": 0.4375,
137
  "rewards/cumulative_reward_2": 0.0,
138
+ "rewards/final_correctness_reward_func": 0.125,
139
+ "rewards/question_recreation_reward_func": 0.6186253735795617,
140
  "rewards/soft_format_reward_func": 0.0,
141
+ "rewards/strict_format_reward_func": 0.0,
142
+ "rewards/xmlcount_reward_func": 0.6928437631577253,
143
  "step": 14
144
  },
145
  {
146
+ "completion_length": 323.0416666666667,
147
+ "epoch": 4.0,
148
+ "grad_norm": 3.448960542678833,
149
+ "kl": 0.004590911810131122,
150
  "learning_rate": 8.067960709356478e-08,
151
  "loss": 0.0,
152
+ "reward": 1.647290966163079,
153
+ "reward_std": 1.4142502636338274,
154
+ "rewards/concensus_correctness_reward_func": 0.23466667036215463,
155
+ "rewards/consensus_reward_func": 0.25,
156
  "rewards/cumulative_reward_2": 0.0,
157
+ "rewards/final_correctness_reward_func": 0.16666666666666666,
158
+ "rewards/question_recreation_reward_func": 0.6292909507950147,
159
  "rewards/soft_format_reward_func": 0.0,
160
  "rewards/strict_format_reward_func": 0.0,
161
+ "rewards/xmlcount_reward_func": 0.3666666764765978,
162
  "step": 16
163
  },
164
  {
165
+ "completion_length": 273.0,
166
+ "epoch": 4.571428571428571,
167
+ "grad_norm": 5.234982013702393,
168
+ "kl": 0.013104470373946242,
169
  "learning_rate": 3.013156219837776e-08,
170
  "loss": 0.0,
171
+ "reward": 2.2405245900154114,
172
+ "reward_std": 0.7255052882392192,
173
+ "rewards/concensus_correctness_reward_func": 0.23793749883770943,
174
+ "rewards/consensus_reward_func": 0.25,
175
  "rewards/cumulative_reward_2": 0.0,
176
+ "rewards/final_correctness_reward_func": 0.375,
177
+ "rewards/question_recreation_reward_func": 0.5015558530576527,
178
  "rewards/soft_format_reward_func": 0.0,
179
+ "rewards/strict_format_reward_func": 0.0,
180
+ "rewards/xmlcount_reward_func": 0.876031257212162,
181
  "step": 18
182
  },
183
  {
184
+ "completion_length": 259.5833333333333,
185
+ "epoch": 5.0,
186
+ "grad_norm": 3.524683952331543,
187
+ "kl": 0.00839625116592894,
188
  "learning_rate": 3.4096741493194193e-09,
189
  "loss": 0.0,
190
+ "reward": 2.6440209125479064,
191
+ "reward_std": 1.0717806244889896,
192
+ "rewards/concensus_correctness_reward_func": 0.6558333337306976,
193
+ "rewards/consensus_reward_func": 0.6666666666666666,
194
  "rewards/cumulative_reward_2": 0.0,
195
+ "rewards/final_correctness_reward_func": 0.08333333333333333,
196
+ "rewards/question_recreation_reward_func": 0.5464792400598526,
197
  "rewards/soft_format_reward_func": 0.0,
198
  "rewards/strict_format_reward_func": 0.0,
199
+ "rewards/xmlcount_reward_func": 0.6917083462079366,
200
  "step": 20
201
  },
202
  {
203
+ "epoch": 5.0,
204
  "step": 20,
205
  "total_flos": 0.0,
206
+ "train_loss": 4.036165773868561e-06,
207
+ "train_runtime": 203.1494,
208
+ "train_samples_per_second": 1.575,
209
+ "train_steps_per_second": 0.098
210
  }
211
  ],
212
  "logging_steps": 2,
213
  "max_steps": 20,
214
  "num_input_tokens_seen": 0,
215
+ "num_train_epochs": 7,
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:7dfd42996f571fced0bd9932b34fefd526e57445bd43e3f78356d7c98ac62d26
3
  size 5944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e71e89f0307a35f08070224668f1f70ed2bded98f9b9c06c83a9fb840307084f
3
  size 5944