sharkbn01 commited on
Commit
17dc470
·
verified ·
1 Parent(s): 75b7ccc

End of training

Browse files
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- base_model: Gensyn/Qwen2.5-0.5B-Instruct
3
  library_name: transformers
4
  model_name: gensyn-checkpoints-masked_vocal_salmon
5
  tags:
@@ -8,13 +8,14 @@ tags:
8
  - grpo
9
  - gensyn
10
  - I am masked vocal salmon
 
11
  - trl
12
  licence: license
13
  ---
14
 
15
  # Model Card for gensyn-checkpoints-masked_vocal_salmon
16
 
17
- This model is a fine-tuned version of [Gensyn/Qwen2.5-0.5B-Instruct](https://huggingface.co/Gensyn/Qwen2.5-0.5B-Instruct).
18
  It has been trained using [TRL](https://github.com/huggingface/trl).
19
 
20
  ## Quick start
@@ -39,7 +40,7 @@ This model was trained with GRPO, a method introduced in [DeepSeekMath: Pushing
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
45
 
 
1
  ---
2
+ base_model: Gensyn/Qwen2.5-1.5B-Instruct
3
  library_name: transformers
4
  model_name: gensyn-checkpoints-masked_vocal_salmon
5
  tags:
 
8
  - grpo
9
  - gensyn
10
  - I am masked vocal salmon
11
+ - unsloth
12
  - trl
13
  licence: license
14
  ---
15
 
16
  # Model Card for gensyn-checkpoints-masked_vocal_salmon
17
 
18
+ This model is a fine-tuned version of [Gensyn/Qwen2.5-1.5B-Instruct](https://huggingface.co/Gensyn/Qwen2.5-1.5B-Instruct).
19
  It has been trained using [TRL](https://github.com/huggingface/trl).
20
 
21
  ## Quick start
 
40
 
41
  - TRL: 0.15.2
42
  - Transformers: 4.51.3
43
+ - Pytorch: 2.6.0
44
  - Datasets: 3.5.0
45
  - Tokenizers: 0.21.1
46
 
adapter_config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "Gensyn/Qwen2.5-1.5B-Instruct",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": false,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 16,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "r": 16,
24
+ "rank_pattern": {},
25
+ "revision": null,
26
+ "target_modules": [
27
+ "k_proj",
28
+ "gate_proj",
29
+ "down_proj",
30
+ "q_proj",
31
+ "o_proj",
32
+ "v_proj",
33
+ "up_proj"
34
+ ],
35
+ "task_type": "CAUSAL_LM",
36
+ "trainable_token_indices": null,
37
+ "use_dora": false,
38
+ "use_rslora": false
39
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4aaa02a548ef17ca89d161df6a85af7e0e4203ad9b67593b1e00d234582b821
3
+ size 73911112
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 0.0014741608130862005,
4
- "train_runtime": 143.5727,
5
- "train_samples": 28,
6
- "train_samples_per_second": 2.229,
7
- "train_steps_per_second": 0.139
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 6.654366700331593e-07,
4
+ "train_runtime": 1175.1261,
5
+ "train_samples": 11,
6
+ "train_samples_per_second": 0.272,
7
+ "train_steps_per_second": 0.017
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 0.0014741608130862005,
4
- "train_runtime": 143.5727,
5
- "train_samples": 28,
6
- "train_samples_per_second": 2.229,
7
- "train_steps_per_second": 0.139
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 6.654366700331593e-07,
4
+ "train_runtime": 1175.1261,
5
+ "train_samples": 11,
6
+ "train_samples_per_second": 0.272,
7
+ "train_steps_per_second": 0.017
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": 5.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": 221.9375,
14
- "epoch": 0.5714285714285714,
15
- "grad_norm": 9.694320678710938,
16
- "kl": 0.0,
17
  "learning_rate": 5e-07,
18
- "loss": -0.0,
19
- "reward": 2.8578853271901608,
20
- "reward_std": 1.0625475882552564,
21
- "rewards/concensus_correctness_reward_func": 0.5570000037550926,
22
- "rewards/consensus_reward_func": 0.6875,
23
  "rewards/cumulative_reward_2": 0.0,
24
- "rewards/final_correctness_reward_func": 0.0625,
25
- "rewards/question_recreation_reward_func": 0.6873852936550975,
26
  "rewards/soft_format_reward_func": 0.0,
27
- "rewards/strict_format_reward_func": 0.109375,
28
- "rewards/xmlcount_reward_func": 0.7541250037029386,
29
  "step": 2
30
  },
31
  {
32
- "completion_length": 136.83333333333334,
33
- "epoch": 1.0,
34
- "grad_norm": 5.852896690368652,
35
- "kl": 0.0063058519347881275,
36
  "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
- "reward": 3.7386062343915305,
39
- "reward_std": 1.4028123530248802,
40
- "rewards/concensus_correctness_reward_func": 0.70066666106383,
41
- "rewards/consensus_reward_func": 0.6666666666666666,
42
  "rewards/cumulative_reward_2": 0.0,
43
- "rewards/final_correctness_reward_func": 0.16666666666666666,
44
- "rewards/question_recreation_reward_func": 0.6852311789989471,
45
  "rewards/soft_format_reward_func": 0.0,
46
- "rewards/strict_format_reward_func": 0.3333333333333333,
47
- "rewards/xmlcount_reward_func": 1.1860416730244954,
48
  "step": 4
49
  },
50
  {
51
- "completion_length": 146.78125,
52
- "epoch": 1.5714285714285714,
53
- "grad_norm": 10.775361061096191,
54
- "kl": 0.022688565848511644,
55
  "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
- "reward": 4.566311419010162,
58
- "reward_std": 1.5661333670868771,
59
- "rewards/concensus_correctness_reward_func": 1.4004999957978725,
60
- "rewards/consensus_reward_func": 0.9375,
61
  "rewards/cumulative_reward_2": 0.0,
62
- "rewards/final_correctness_reward_func": 0.0625,
63
- "rewards/question_recreation_reward_func": 0.7594364183023572,
64
  "rewards/soft_format_reward_func": 0.0,
65
- "rewards/strict_format_reward_func": 0.234375,
66
- "rewards/xmlcount_reward_func": 1.1719999946653843,
67
  "step": 6
68
  },
69
  {
70
- "completion_length": 142.91666666666666,
71
- "epoch": 2.0,
72
- "grad_norm": 9.037089347839355,
73
- "kl": 0.31922435003798455,
74
  "learning_rate": 3.867370395306068e-07,
75
- "loss": 0.0002,
76
- "reward": 4.1081428329149885,
77
- "reward_std": 1.029661375455665,
78
- "rewards/concensus_correctness_reward_func": 0.7650833452741305,
79
- "rewards/consensus_reward_func": 0.9166666666666666,
80
  "rewards/cumulative_reward_2": 0.0,
81
- "rewards/final_correctness_reward_func": 0.25,
82
- "rewards/question_recreation_reward_func": 0.6767677310854197,
83
  "rewards/soft_format_reward_func": 0.0,
84
- "rewards/strict_format_reward_func": 0.3125,
85
- "rewards/xmlcount_reward_func": 1.1871250073115032,
86
  "step": 8
87
  },
88
  {
89
- "completion_length": 143.09375,
90
- "epoch": 2.571428571428571,
91
- "grad_norm": 11.110583305358887,
92
- "kl": 0.5744109276565723,
93
  "learning_rate": 3.1137137178519977e-07,
94
- "loss": 0.0006,
95
- "reward": 3.682307220995426,
96
- "reward_std": 1.1703917261911556,
97
- "rewards/concensus_correctness_reward_func": 0.7889375016093254,
98
- "rewards/consensus_reward_func": 0.75,
99
  "rewards/cumulative_reward_2": 0.0,
100
- "rewards/final_correctness_reward_func": 0.1875,
101
- "rewards/question_recreation_reward_func": 0.6156822107732296,
102
  "rewards/soft_format_reward_func": 0.0,
103
- "rewards/strict_format_reward_func": 0.234375,
104
- "rewards/xmlcount_reward_func": 1.1058125048875809,
105
  "step": 10
106
  },
107
  {
108
- "completion_length": 136.41666666666666,
109
- "epoch": 3.0,
110
- "grad_norm": 5.076348304748535,
111
- "kl": 2.3429599769879133,
112
  "learning_rate": 2.2935516363191693e-07,
113
- "loss": 0.0018,
114
- "reward": 4.40448385477066,
115
- "reward_std": 1.6560175865888596,
116
- "rewards/concensus_correctness_reward_func": 0.8267500052849451,
117
- "rewards/consensus_reward_func": 1.1666666666666667,
118
  "rewards/cumulative_reward_2": 0.0,
119
- "rewards/final_correctness_reward_func": 0.08333333333333333,
120
- "rewards/question_recreation_reward_func": 0.8022754968454441,
121
  "rewards/soft_format_reward_func": 0.0,
122
- "rewards/strict_format_reward_func": 0.3125,
123
- "rewards/xmlcount_reward_func": 1.2129583358764648,
124
  "step": 12
125
  },
126
  {
127
- "completion_length": 169.6875,
128
- "epoch": 3.571428571428571,
129
- "grad_norm": 8.082115173339844,
130
- "kl": 7.364466468396131,
131
  "learning_rate": 1.4957614383675767e-07,
132
- "loss": 0.0074,
133
- "reward": 4.209167793393135,
134
- "reward_std": 1.530743383220397,
135
- "rewards/concensus_correctness_reward_func": 0.805249996483326,
136
- "rewards/consensus_reward_func": 1.0625,
137
  "rewards/cumulative_reward_2": 0.0,
138
- "rewards/final_correctness_reward_func": 0.1875,
139
- "rewards/question_recreation_reward_func": 0.6893240767531097,
140
  "rewards/soft_format_reward_func": 0.0,
141
- "rewards/strict_format_reward_func": 0.296875,
142
- "rewards/xmlcount_reward_func": 1.1677187494933605,
143
  "step": 14
144
  },
145
  {
146
- "completion_length": 151.83333333333334,
147
- "epoch": 4.0,
148
- "grad_norm": 23.623441696166992,
149
- "kl": 5.0384774438571185,
150
  "learning_rate": 8.067960709356478e-08,
151
- "loss": 0.0038,
152
- "reward": 3.1199777324994407,
153
- "reward_std": 1.2431689854711294,
154
- "rewards/concensus_correctness_reward_func": 0.32775000234444934,
155
- "rewards/consensus_reward_func": 0.4166666666666667,
156
  "rewards/cumulative_reward_2": 0.0,
157
- "rewards/final_correctness_reward_func": 0.16666666666666666,
158
- "rewards/question_recreation_reward_func": 0.714102791932722,
159
  "rewards/soft_format_reward_func": 0.0,
160
- "rewards/strict_format_reward_func": 0.3125,
161
- "rewards/xmlcount_reward_func": 1.1822916666666667,
162
  "step": 16
163
  },
164
  {
165
- "completion_length": 202.25,
166
- "epoch": 4.571428571428571,
167
- "grad_norm": 9.800145149230957,
168
- "kl": 0.6018333853571676,
169
  "learning_rate": 3.013156219837776e-08,
170
- "loss": 0.0006,
171
- "reward": 3.698561042547226,
172
- "reward_std": 1.2764807561179623,
173
- "rewards/concensus_correctness_reward_func": 0.6848750058561563,
174
- "rewards/consensus_reward_func": 0.9375,
175
  "rewards/cumulative_reward_2": 0.0,
176
- "rewards/final_correctness_reward_func": 0.0625,
177
- "rewards/question_recreation_reward_func": 0.7144985762424767,
178
  "rewards/soft_format_reward_func": 0.0,
179
- "rewards/strict_format_reward_func": 0.203125,
180
- "rewards/xmlcount_reward_func": 1.096062496304512,
181
  "step": 18
182
  },
183
  {
184
- "completion_length": 162.91666666666666,
185
- "epoch": 5.0,
186
- "grad_norm": 7.49417781829834,
187
- "kl": 0.5289312032206605,
188
  "learning_rate": 3.4096741493194193e-09,
189
- "loss": 0.0004,
190
- "reward": 4.249745080868403,
191
- "reward_std": 1.1143968345907826,
192
- "rewards/concensus_correctness_reward_func": 0.8489166696866354,
193
- "rewards/consensus_reward_func": 1.0,
194
  "rewards/cumulative_reward_2": 0.0,
195
- "rewards/final_correctness_reward_func": 0.16666666666666666,
196
- "rewards/question_recreation_reward_func": 0.7353284160296122,
197
  "rewards/soft_format_reward_func": 0.0,
198
- "rewards/strict_format_reward_func": 0.2916666666666667,
199
- "rewards/xmlcount_reward_func": 1.2071666717529297,
200
  "step": 20
201
  },
202
  {
203
- "epoch": 5.0,
204
  "step": 20,
205
  "total_flos": 0.0,
206
- "train_loss": 0.0014741608130862005,
207
- "train_runtime": 143.5727,
208
- "train_samples_per_second": 2.229,
209
- "train_steps_per_second": 0.139
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": {
@@ -227,7 +227,7 @@
227
  }
228
  },
229
  "total_flos": 0.0,
230
- "train_batch_size": 2,
231
  "trial_name": null,
232
  "trial_params": null
233
  }
 
2
  "best_global_step": null,
3
  "best_metric": null,
4
  "best_model_checkpoint": null,
5
+ "epoch": 6.7272727272727275,
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": 472.15625,
14
+ "epoch": 0.7272727272727273,
15
+ "grad_norm": 0.6036210656166077,
16
+ "kl": 0.0008934065590437967,
17
  "learning_rate": 5e-07,
18
+ "loss": 0.0,
19
+ "reward": 1.4180201515555382,
20
+ "reward_std": 1.395761176943779,
21
+ "rewards/concensus_correctness_reward_func": 0.018624999560415745,
22
+ "rewards/consensus_reward_func": 0.5,
23
  "rewards/cumulative_reward_2": 0.0,
24
+ "rewards/final_correctness_reward_func": 0.0,
25
+ "rewards/question_recreation_reward_func": 0.423113907687366,
26
  "rewards/soft_format_reward_func": 0.0,
27
+ "rewards/strict_format_reward_func": 0.03125,
28
+ "rewards/xmlcount_reward_func": 0.4450312480330467,
29
  "step": 2
30
  },
31
  {
32
+ "completion_length": 419.2142857142857,
33
+ "epoch": 1.3636363636363638,
34
+ "grad_norm": 0.5538027882575989,
35
+ "kl": 0.0007605135199680392,
36
  "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
+ "reward": 0.9143300311905997,
39
+ "reward_std": 0.8392358294555119,
40
+ "rewards/concensus_correctness_reward_func": 0.0,
41
+ "rewards/consensus_reward_func": 0.07142857142857142,
42
  "rewards/cumulative_reward_2": 0.0,
43
+ "rewards/final_correctness_reward_func": 0.0,
44
+ "rewards/question_recreation_reward_func": 0.30008004180022646,
45
  "rewards/soft_format_reward_func": 0.0,
46
+ "rewards/strict_format_reward_func": 0.03571428571428571,
47
+ "rewards/xmlcount_reward_func": 0.5071071471486773,
48
  "step": 4
49
  },
50
  {
51
+ "completion_length": 542.4642857142857,
52
+ "epoch": 2.0,
53
+ "grad_norm": 0.39062559604644775,
54
+ "kl": 0.0006340955400706402,
55
  "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
+ "reward": 0.6439937661801066,
58
+ "reward_std": 0.8819772984300341,
59
+ "rewards/concensus_correctness_reward_func": 0.0,
60
+ "rewards/consensus_reward_func": 0.21428571428571427,
61
  "rewards/cumulative_reward_2": 0.0,
62
+ "rewards/final_correctness_reward_func": 0.0,
63
+ "rewards/question_recreation_reward_func": 0.5107437585081372,
64
  "rewards/soft_format_reward_func": 0.0,
65
+ "rewards/strict_format_reward_func": 0.0,
66
+ "rewards/xmlcount_reward_func": -0.08103571246777262,
67
  "step": 6
68
  },
69
  {
70
+ "completion_length": 452.40625,
71
+ "epoch": 2.7272727272727275,
72
+ "grad_norm": 0.484588086605072,
73
+ "kl": 0.0007340605188801419,
74
  "learning_rate": 3.867370395306068e-07,
75
+ "loss": 0.0,
76
+ "reward": 1.2139809615910053,
77
+ "reward_std": 1.1542998775839806,
78
+ "rewards/concensus_correctness_reward_func": 0.013812499586492777,
79
+ "rewards/consensus_reward_func": 0.5625,
80
  "rewards/cumulative_reward_2": 0.0,
81
+ "rewards/final_correctness_reward_func": 0.0,
82
+ "rewards/question_recreation_reward_func": 0.4430122319608927,
83
  "rewards/soft_format_reward_func": 0.0,
84
+ "rewards/strict_format_reward_func": 0.03125,
85
+ "rewards/xmlcount_reward_func": 0.16340623889118433,
86
  "step": 8
87
  },
88
  {
89
+ "completion_length": 580.5714285714286,
90
+ "epoch": 3.3636363636363638,
91
+ "grad_norm": 0.3272243142127991,
92
+ "kl": 0.0006291726936719247,
93
  "learning_rate": 3.1137137178519977e-07,
94
+ "loss": 0.0,
95
+ "reward": 0.9775400928088597,
96
+ "reward_std": 0.9005033576062748,
97
+ "rewards/concensus_correctness_reward_func": 0.0,
98
+ "rewards/consensus_reward_func": 0.2857142857142857,
99
  "rewards/cumulative_reward_2": 0.0,
100
+ "rewards/final_correctness_reward_func": 0.0,
101
+ "rewards/question_recreation_reward_func": 0.49911151826381683,
102
  "rewards/soft_format_reward_func": 0.0,
103
+ "rewards/strict_format_reward_func": 0.0,
104
+ "rewards/xmlcount_reward_func": 0.19271428563765117,
105
  "step": 10
106
  },
107
  {
108
+ "completion_length": 479.92857142857144,
109
+ "epoch": 4.0,
110
+ "grad_norm": 0.37380141019821167,
111
+ "kl": 0.00070970255183056,
112
  "learning_rate": 2.2935516363191693e-07,
113
+ "loss": 0.0,
114
+ "reward": 1.1728975645133428,
115
+ "reward_std": 1.135693175452096,
116
+ "rewards/concensus_correctness_reward_func": 0.0,
117
+ "rewards/consensus_reward_func": 0.35714285714285715,
118
  "rewards/cumulative_reward_2": 0.0,
119
+ "rewards/final_correctness_reward_func": 0.0,
120
+ "rewards/question_recreation_reward_func": 0.5700761207512447,
121
  "rewards/soft_format_reward_func": 0.0,
122
+ "rewards/strict_format_reward_func": 0.05357142857142857,
123
+ "rewards/xmlcount_reward_func": 0.19210714208228247,
124
  "step": 12
125
  },
126
  {
127
+ "completion_length": 569.4375,
128
+ "epoch": 4.7272727272727275,
129
+ "grad_norm": 0.31213265657424927,
130
+ "kl": 0.0007356460555456579,
131
  "learning_rate": 1.4957614383675767e-07,
132
+ "loss": 0.0,
133
+ "reward": 0.5268249623477459,
134
+ "reward_std": 0.9648920074105263,
135
+ "rewards/concensus_correctness_reward_func": 0.004812499973922968,
136
+ "rewards/consensus_reward_func": 0.3125,
137
  "rewards/cumulative_reward_2": 0.0,
138
+ "rewards/final_correctness_reward_func": 0.0,
139
+ "rewards/question_recreation_reward_func": 0.4122624695301056,
140
  "rewards/soft_format_reward_func": 0.0,
141
+ "rewards/strict_format_reward_func": 0.0,
142
+ "rewards/xmlcount_reward_func": -0.20275000482797623,
143
  "step": 14
144
  },
145
  {
146
+ "completion_length": 550.0714285714286,
147
+ "epoch": 5.363636363636363,
148
+ "grad_norm": 0.3187633454799652,
149
+ "kl": 0.0006976349395699799,
150
  "learning_rate": 8.067960709356478e-08,
151
+ "loss": 0.0,
152
+ "reward": 1.2211733332702093,
153
+ "reward_std": 1.0074801189558846,
154
+ "rewards/concensus_correctness_reward_func": 0.010285713842936925,
155
+ "rewards/consensus_reward_func": 0.5714285714285714,
156
  "rewards/cumulative_reward_2": 0.0,
157
+ "rewards/final_correctness_reward_func": 0.0,
158
+ "rewards/question_recreation_reward_func": 0.5048876021589551,
159
  "rewards/soft_format_reward_func": 0.0,
160
+ "rewards/strict_format_reward_func": 0.017857142857142856,
161
+ "rewards/xmlcount_reward_func": 0.11671427318028041,
162
  "step": 16
163
  },
164
  {
165
+ "completion_length": 499.0,
166
+ "epoch": 6.0,
167
+ "grad_norm": 0.48127031326293945,
168
+ "kl": 0.0007189161011150905,
169
  "learning_rate": 3.013156219837776e-08,
170
+ "loss": 0.0,
171
+ "reward": 0.8536344426018851,
172
+ "reward_std": 1.009829261473247,
173
+ "rewards/concensus_correctness_reward_func": 0.010285713842936925,
174
+ "rewards/consensus_reward_func": 0.35714285714285715,
175
  "rewards/cumulative_reward_2": 0.0,
176
+ "rewards/final_correctness_reward_func": 0.0,
177
+ "rewards/question_recreation_reward_func": 0.4199201501905918,
178
  "rewards/soft_format_reward_func": 0.0,
179
+ "rewards/strict_format_reward_func": 0.0,
180
+ "rewards/xmlcount_reward_func": 0.0662857059921537,
181
  "step": 18
182
  },
183
  {
184
+ "completion_length": 409.71875,
185
+ "epoch": 6.7272727272727275,
186
+ "grad_norm": 3.820303440093994,
187
+ "kl": 0.0007408617093460634,
188
  "learning_rate": 3.4096741493194193e-09,
189
+ "loss": 0.0,
190
+ "reward": 1.0658142920583487,
191
+ "reward_std": 1.05014331266284,
192
+ "rewards/concensus_correctness_reward_func": 0.004812499973922968,
193
+ "rewards/consensus_reward_func": 0.375,
194
  "rewards/cumulative_reward_2": 0.0,
195
+ "rewards/final_correctness_reward_func": 0.0,
196
+ "rewards/question_recreation_reward_func": 0.46540803741663694,
197
  "rewards/soft_format_reward_func": 0.0,
198
+ "rewards/strict_format_reward_func": 0.0,
199
+ "rewards/xmlcount_reward_func": 0.22059374302625656,
200
  "step": 20
201
  },
202
  {
203
+ "epoch": 6.7272727272727275,
204
  "step": 20,
205
  "total_flos": 0.0,
206
+ "train_loss": 6.654366700331593e-07,
207
+ "train_runtime": 1175.1261,
208
+ "train_samples_per_second": 0.272,
209
+ "train_steps_per_second": 0.017
210
  }
211
  ],
212
  "logging_steps": 2,
213
  "max_steps": 20,
214
  "num_input_tokens_seen": 0,
215
+ "num_train_epochs": 10,
216
  "save_steps": 25,
217
  "stateful_callbacks": {
218
  "TrainerControl": {
 
227
  }
228
  },
229
  "total_flos": 0.0,
230
+ "train_batch_size": 4,
231
  "trial_name": null,
232
  "trial_params": null
233
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:123588bd2f338759d72e8685564678580a958041438fa52b6cbc5acd86f969e5
3
- size 5880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c2728207bad49d3f20aa37952e67ccc6d3233f5e53095ed17a152bcaff255850
3
+ size 5944