ringodaro commited on
Commit
632d65f
·
verified ·
1 Parent(s): f19d49a

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-soaring_polished_cobra
5
  tags:
@@ -8,13 +8,14 @@ tags:
8
  - grpo
9
  - gensyn
10
  - I am soaring polished cobra
 
11
  - trl
12
  licence: license
13
  ---
14
 
15
  # Model Card for gensyn-checkpoints-soaring_polished_cobra
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-soaring_polished_cobra
5
  tags:
 
8
  - grpo
9
  - gensyn
10
  - I am soaring polished cobra
11
+ - unsloth
12
  - trl
13
  licence: license
14
  ---
15
 
16
  # Model Card for gensyn-checkpoints-soaring_polished_cobra
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
+ "gate_proj",
28
+ "q_proj",
29
+ "o_proj",
30
+ "k_proj",
31
+ "down_proj",
32
+ "up_proj",
33
+ "v_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:ef0c9fc2a22680be6b9b8fa9540f1875dc6c049693303b0d3390ad2c07c40f9d
3
+ size 73911112
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 7.716044910921482e-06,
4
- "train_runtime": 943.9196,
5
- "train_samples": 51,
6
- "train_samples_per_second": 0.339,
7
- "train_steps_per_second": 0.021
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 6.641451051336844e-07,
4
+ "train_runtime": 718.6924,
5
+ "train_samples": 17,
6
+ "train_samples_per_second": 0.445,
7
+ "train_steps_per_second": 0.028
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 7.716044910921482e-06,
4
- "train_runtime": 943.9196,
5
- "train_samples": 51,
6
- "train_samples_per_second": 0.339,
7
- "train_steps_per_second": 0.021
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 6.641451051336844e-07,
4
+ "train_runtime": 718.6924,
5
+ "train_samples": 17,
6
+ "train_samples_per_second": 0.445,
7
+ "train_steps_per_second": 0.028
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": 2.9411764705882355,
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": 320.21875,
14
- "epoch": 0.3137254901960784,
15
- "grad_norm": 18.987831115722656,
16
- "kl": 0.0,
17
  "learning_rate": 5e-07,
18
  "loss": 0.0,
19
- "reward": 0.9052386409603059,
20
- "reward_std": 1.0872065802104771,
21
- "rewards/concensus_correctness_reward_func": 0.0078125,
22
- "rewards/consensus_reward_func": 0.25,
23
  "rewards/cumulative_reward_2": 0.0,
24
- "rewards/final_correctness_reward_func": 0.0625,
25
- "rewards/question_recreation_reward_func": 0.32886362238787115,
26
- "rewards/soft_format_reward_func": 0.015625,
27
  "rewards/strict_format_reward_func": 0.015625,
28
- "rewards/xmlcount_reward_func": 0.22481249924749136,
29
  "step": 2
30
  },
31
  {
32
- "completion_length": 255.65625,
33
- "epoch": 0.6274509803921569,
34
- "grad_norm": 7.594056606292725,
35
- "kl": 0.0013956625989521854,
36
  "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
- "reward": 1.7687878240831196,
39
- "reward_std": 0.8872564652119763,
40
- "rewards/concensus_correctness_reward_func": 0.19506250228732824,
41
- "rewards/consensus_reward_func": 0.1875,
42
  "rewards/cumulative_reward_2": 0.0,
43
- "rewards/final_correctness_reward_func": 0.1875,
44
- "rewards/question_recreation_reward_func": 0.5405378290452063,
45
- "rewards/soft_format_reward_func": 0.0,
46
- "rewards/strict_format_reward_func": 0.0,
47
- "rewards/xmlcount_reward_func": 0.6581875095143914,
48
  "step": 4
49
  },
50
  {
51
- "completion_length": 241.71875,
52
- "epoch": 0.9411764705882353,
53
- "grad_norm": 5.940536022186279,
54
- "kl": 0.0032457121487823315,
55
  "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
- "reward": 3.927860476076603,
58
- "reward_std": 1.7695390642620623,
59
- "rewards/concensus_correctness_reward_func": 2.1019999980926514,
60
- "rewards/consensus_reward_func": 0.375,
61
  "rewards/cumulative_reward_2": 0.0,
62
- "rewards/final_correctness_reward_func": 0.25,
63
- "rewards/question_recreation_reward_func": 0.43264187331078574,
64
  "rewards/soft_format_reward_func": 0.0,
65
  "rewards/strict_format_reward_func": 0.0,
66
- "rewards/xmlcount_reward_func": 0.7682187557220459,
67
  "step": 6
68
  },
69
  {
70
- "completion_length": 256.0,
71
- "epoch": 1.156862745098039,
72
- "grad_norm": 9.757232666015625,
73
- "kl": 0.00435605255717581,
74
  "learning_rate": 3.867370395306068e-07,
75
  "loss": 0.0,
76
- "reward": 1.873472045768391,
77
- "reward_std": 0.48061720840632915,
78
- "rewards/concensus_correctness_reward_func": 0.15699999982660467,
79
- "rewards/consensus_reward_func": 0.09090909090909091,
80
  "rewards/cumulative_reward_2": 0.0,
81
- "rewards/final_correctness_reward_func": 0.2727272727272727,
82
- "rewards/question_recreation_reward_func": 0.7095629688013684,
83
  "rewards/soft_format_reward_func": 0.0,
84
  "rewards/strict_format_reward_func": 0.0,
85
- "rewards/xmlcount_reward_func": 0.643272725018588,
86
  "step": 8
87
  },
88
  {
89
- "completion_length": 158.59375,
90
- "epoch": 1.4705882352941178,
91
- "grad_norm": 6.622710227966309,
92
- "kl": 0.010859893664019182,
93
  "learning_rate": 3.1137137178519977e-07,
94
  "loss": 0.0,
95
- "reward": 3.0381277231499553,
96
- "reward_std": 0.5335058589698747,
97
- "rewards/concensus_correctness_reward_func": 1.4662500023841858,
98
- "rewards/consensus_reward_func": 0.4375,
99
  "rewards/cumulative_reward_2": 0.0,
100
  "rewards/final_correctness_reward_func": 0.0,
101
- "rewards/question_recreation_reward_func": 0.36175276967696846,
102
  "rewards/soft_format_reward_func": 0.0,
103
  "rewards/strict_format_reward_func": 0.0,
104
- "rewards/xmlcount_reward_func": 0.77262500859797,
105
  "step": 10
106
  },
107
  {
108
- "completion_length": 283.375,
109
- "epoch": 1.784313725490196,
110
- "grad_norm": 5.638906002044678,
111
- "kl": 0.007406034710584208,
112
  "learning_rate": 2.2935516363191693e-07,
113
  "loss": 0.0,
114
- "reward": 1.6740143587812781,
115
- "reward_std": 1.191217671148479,
116
- "rewards/concensus_correctness_reward_func": 0.1471250019967556,
117
- "rewards/consensus_reward_func": 0.1875,
118
  "rewards/cumulative_reward_2": 0.0,
119
- "rewards/final_correctness_reward_func": 0.25,
120
- "rewards/question_recreation_reward_func": 0.48260811483487487,
121
  "rewards/soft_format_reward_func": 0.0,
122
  "rewards/strict_format_reward_func": 0.0,
123
- "rewards/xmlcount_reward_func": 0.6067812591791153,
124
  "step": 12
125
  },
126
  {
127
- "completion_length": 249.0,
128
- "epoch": 2.0,
129
- "grad_norm": 3.3914954662323,
130
- "kl": 0.011619402540170333,
131
  "learning_rate": 1.4957614383675767e-07,
132
  "loss": 0.0,
133
- "reward": 4.049619214101271,
134
- "reward_std": 2.009789251807061,
135
- "rewards/concensus_correctness_reward_func": 2.756545454263687,
136
- "rewards/consensus_reward_func": 0.09090909090909091,
137
  "rewards/cumulative_reward_2": 0.0,
138
- "rewards/final_correctness_reward_func": 0.09090909090909091,
139
- "rewards/question_recreation_reward_func": 0.4135735107755119,
140
  "rewards/soft_format_reward_func": 0.0,
141
- "rewards/strict_format_reward_func": 0.022727272727272728,
142
- "rewards/xmlcount_reward_func": 0.6749545498327776,
143
  "step": 14
144
  },
145
  {
146
- "completion_length": 223.875,
147
- "epoch": 2.313725490196078,
148
- "grad_norm": 8.215188026428223,
149
- "kl": 0.014186236599925905,
150
  "learning_rate": 8.067960709356478e-08,
151
  "loss": 0.0,
152
- "reward": 2.642803938128054,
153
- "reward_std": 1.4382841207552701,
154
- "rewards/concensus_correctness_reward_func": 0.8331250017508864,
155
- "rewards/consensus_reward_func": 0.25,
156
  "rewards/cumulative_reward_2": 0.0,
157
- "rewards/final_correctness_reward_func": 0.3125,
158
- "rewards/question_recreation_reward_func": 0.4023352946387604,
159
  "rewards/soft_format_reward_func": 0.0,
160
- "rewards/strict_format_reward_func": 0.015625,
161
- "rewards/xmlcount_reward_func": 0.829218752682209,
162
  "step": 16
163
  },
164
  {
165
- "completion_length": 247.875,
166
- "epoch": 2.627450980392157,
167
- "grad_norm": 7.052781105041504,
168
- "kl": 0.015200054476736113,
169
  "learning_rate": 3.013156219837776e-08,
170
  "loss": 0.0,
171
- "reward": 1.8882133960723877,
172
- "reward_std": 1.274847088381648,
173
- "rewards/concensus_correctness_reward_func": 0.06793750077486038,
174
- "rewards/consensus_reward_func": 0.25,
175
  "rewards/cumulative_reward_2": 0.0,
176
  "rewards/final_correctness_reward_func": 0.3125,
177
- "rewards/question_recreation_reward_func": 0.5125571433454752,
178
- "rewards/soft_format_reward_func": 0.0,
179
- "rewards/strict_format_reward_func": 0.046875,
180
- "rewards/xmlcount_reward_func": 0.698343763127923,
181
  "step": 18
182
  },
183
  {
184
- "completion_length": 253.28125,
185
- "epoch": 2.9411764705882355,
186
- "grad_norm": 7.9368438720703125,
187
- "kl": 0.013815053927828558,
188
  "learning_rate": 3.4096741493194193e-09,
189
  "loss": 0.0,
190
- "reward": 3.0817041154950857,
191
- "reward_std": 2.74389247270301,
192
- "rewards/concensus_correctness_reward_func": 1.3802500003948808,
193
- "rewards/consensus_reward_func": 0.25,
194
  "rewards/cumulative_reward_2": 0.0,
195
- "rewards/final_correctness_reward_func": 0.1875,
196
- "rewards/question_recreation_reward_func": 0.48770411510486156,
197
  "rewards/soft_format_reward_func": 0.0,
198
  "rewards/strict_format_reward_func": 0.0,
199
- "rewards/xmlcount_reward_func": 0.7762499954551458,
200
  "step": 20
201
  },
202
  {
203
- "epoch": 2.9411764705882355,
204
  "step": 20,
205
  "total_flos": 0.0,
206
- "train_loss": 7.716044910921482e-06,
207
- "train_runtime": 943.9196,
208
- "train_samples_per_second": 0.339,
209
- "train_steps_per_second": 0.021
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": {
@@ -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": 4.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": 427.875,
14
+ "epoch": 0.47058823529411764,
15
+ "grad_norm": 0.5876138806343079,
16
+ "kl": 0.0007815321441739798,
17
  "learning_rate": 5e-07,
18
  "loss": 0.0,
19
+ "reward": 1.6895362362265587,
20
+ "reward_std": 1.607804922387004,
21
+ "rewards/concensus_correctness_reward_func": 0.30674999952316284,
22
+ "rewards/consensus_reward_func": 0.375,
23
  "rewards/cumulative_reward_2": 0.0,
24
+ "rewards/final_correctness_reward_func": 0.3125,
25
+ "rewards/question_recreation_reward_func": 0.5297549953684211,
26
+ "rewards/soft_format_reward_func": 0.0,
27
  "rewards/strict_format_reward_func": 0.015625,
28
+ "rewards/xmlcount_reward_func": 0.14990626135841012,
29
  "step": 2
30
  },
31
  {
32
+ "completion_length": 340.0,
33
+ "epoch": 0.9411764705882353,
34
+ "grad_norm": 0.98947674036026,
35
+ "kl": 0.0008097236932371743,
36
  "learning_rate": 4.864543104251586e-07,
37
  "loss": 0.0,
38
+ "reward": 1.1686192080378532,
39
+ "reward_std": 0.8920873831957579,
40
+ "rewards/concensus_correctness_reward_func": 0.0078125,
41
+ "rewards/consensus_reward_func": 0.25,
42
  "rewards/cumulative_reward_2": 0.0,
43
+ "rewards/final_correctness_reward_func": 0.0,
44
+ "rewards/question_recreation_reward_func": 0.562056683935225,
45
+ "rewards/soft_format_reward_func": 0.03125,
46
+ "rewards/strict_format_reward_func": 0.015625,
47
+ "rewards/xmlcount_reward_func": 0.301875002682209,
48
  "step": 4
49
  },
50
  {
51
+ "completion_length": 538.9,
52
+ "epoch": 1.2352941176470589,
53
+ "grad_norm": 0.518266499042511,
54
+ "kl": 0.0006855448475107551,
55
  "learning_rate": 4.472851273490984e-07,
56
  "loss": 0.0,
57
+ "reward": 2.2191823825240133,
58
+ "reward_std": 1.331932282447815,
59
+ "rewards/concensus_correctness_reward_func": 0.525,
60
+ "rewards/consensus_reward_func": 0.9,
61
  "rewards/cumulative_reward_2": 0.0,
62
+ "rewards/final_correctness_reward_func": 0.3,
63
+ "rewards/question_recreation_reward_func": 0.5259323239326477,
64
  "rewards/soft_format_reward_func": 0.0,
65
  "rewards/strict_format_reward_func": 0.0,
66
+ "rewards/xmlcount_reward_func": -0.031749993562698364,
67
  "step": 6
68
  },
69
  {
70
+ "completion_length": 329.9375,
71
+ "epoch": 1.7058823529411766,
72
+ "grad_norm": 0.4131653308868408,
73
+ "kl": 0.0007813851116225123,
74
  "learning_rate": 3.867370395306068e-07,
75
  "loss": 0.0,
76
+ "reward": 1.6284910812973976,
77
+ "reward_std": 0.8130084574222565,
78
+ "rewards/concensus_correctness_reward_func": 0.11649999767541885,
79
+ "rewards/consensus_reward_func": 0.3125,
80
  "rewards/cumulative_reward_2": 0.0,
81
+ "rewards/final_correctness_reward_func": 0.4375,
82
+ "rewards/question_recreation_reward_func": 0.4998348420485854,
83
  "rewards/soft_format_reward_func": 0.0,
84
  "rewards/strict_format_reward_func": 0.0,
85
+ "rewards/xmlcount_reward_func": 0.26215624529868364,
86
  "step": 8
87
  },
88
  {
89
+ "completion_length": 403.2,
90
+ "epoch": 2.0,
91
+ "grad_norm": 0.13836334645748138,
92
+ "kl": 0.0006139612174592912,
93
  "learning_rate": 3.1137137178519977e-07,
94
  "loss": 0.0,
95
+ "reward": 2.0887853801250458,
96
+ "reward_std": 0.6115315139293671,
97
+ "rewards/concensus_correctness_reward_func": 0.0,
98
+ "rewards/consensus_reward_func": 1.1,
99
  "rewards/cumulative_reward_2": 0.0,
100
  "rewards/final_correctness_reward_func": 0.0,
101
+ "rewards/question_recreation_reward_func": 0.5645354151725769,
102
  "rewards/soft_format_reward_func": 0.0,
103
  "rewards/strict_format_reward_func": 0.0,
104
+ "rewards/xmlcount_reward_func": 0.4242500215768814,
105
  "step": 10
106
  },
107
  {
108
+ "completion_length": 444.5625,
109
+ "epoch": 2.4705882352941178,
110
+ "grad_norm": 0.40184539556503296,
111
+ "kl": 0.0006372530478984118,
112
  "learning_rate": 2.2935516363191693e-07,
113
  "loss": 0.0,
114
+ "reward": 1.8296705558896065,
115
+ "reward_std": 1.1184727065265179,
116
+ "rewards/concensus_correctness_reward_func": 0.0,
117
+ "rewards/consensus_reward_func": 0.875,
118
  "rewards/cumulative_reward_2": 0.0,
119
+ "rewards/final_correctness_reward_func": 0.0625,
120
+ "rewards/question_recreation_reward_func": 0.5306080486625433,
121
  "rewards/soft_format_reward_func": 0.0,
122
  "rewards/strict_format_reward_func": 0.0,
123
+ "rewards/xmlcount_reward_func": 0.36156250536441803,
124
  "step": 12
125
  },
126
  {
127
+ "completion_length": 349.3125,
128
+ "epoch": 2.9411764705882355,
129
+ "grad_norm": 0.648057758808136,
130
+ "kl": 0.0008520848932676017,
131
  "learning_rate": 1.4957614383675767e-07,
132
  "loss": 0.0,
133
+ "reward": 1.988405354321003,
134
+ "reward_std": 1.3347271010279655,
135
+ "rewards/concensus_correctness_reward_func": 0.2265625,
136
+ "rewards/consensus_reward_func": 0.5,
137
  "rewards/cumulative_reward_2": 0.0,
138
+ "rewards/final_correctness_reward_func": 0.3125,
139
+ "rewards/question_recreation_reward_func": 0.5402803681790829,
140
  "rewards/soft_format_reward_func": 0.0,
141
+ "rewards/strict_format_reward_func": 0.03125,
142
+ "rewards/xmlcount_reward_func": 0.37781250290572643,
143
  "step": 14
144
  },
145
  {
146
+ "completion_length": 408.5,
147
+ "epoch": 3.235294117647059,
148
+ "grad_norm": 0.6705062389373779,
149
+ "kl": 0.0009508899878710508,
150
  "learning_rate": 8.067960709356478e-08,
151
  "loss": 0.0,
152
+ "reward": 1.2143768668174744,
153
+ "reward_std": 0.8865768074989319,
154
+ "rewards/concensus_correctness_reward_func": 0.0,
155
+ "rewards/consensus_reward_func": 0.4,
156
  "rewards/cumulative_reward_2": 0.0,
157
+ "rewards/final_correctness_reward_func": 0.0,
158
+ "rewards/question_recreation_reward_func": 0.6206269025802612,
159
  "rewards/soft_format_reward_func": 0.0,
160
+ "rewards/strict_format_reward_func": 0.0,
161
+ "rewards/xmlcount_reward_func": 0.19375,
162
  "step": 16
163
  },
164
  {
165
+ "completion_length": 320.40625,
166
+ "epoch": 3.7058823529411766,
167
+ "grad_norm": 1.4544986486434937,
168
+ "kl": 0.0009329580789199099,
169
  "learning_rate": 3.013156219837776e-08,
170
  "loss": 0.0,
171
+ "reward": 2.1640911400318146,
172
+ "reward_std": 1.306618556380272,
173
+ "rewards/concensus_correctness_reward_func": 0.27806249633431435,
174
+ "rewards/consensus_reward_func": 0.625,
175
  "rewards/cumulative_reward_2": 0.0,
176
  "rewards/final_correctness_reward_func": 0.3125,
177
+ "rewards/question_recreation_reward_func": 0.5214036731049418,
178
+ "rewards/soft_format_reward_func": 0.015625,
179
+ "rewards/strict_format_reward_func": 0.03125,
180
+ "rewards/xmlcount_reward_func": 0.3802500069141388,
181
  "step": 18
182
  },
183
  {
184
+ "completion_length": 395.4,
185
+ "epoch": 4.0,
186
+ "grad_norm": 0.2986851930618286,
187
+ "kl": 0.0007005963008850813,
188
  "learning_rate": 3.4096741493194193e-09,
189
  "loss": 0.0,
190
+ "reward": 1.7007000684738158,
191
+ "reward_std": 0.8157501250505448,
192
+ "rewards/concensus_correctness_reward_func": 0.028499999642372133,
193
+ "rewards/consensus_reward_func": 0.9,
194
  "rewards/cumulative_reward_2": 0.0,
195
+ "rewards/final_correctness_reward_func": 0.2,
196
+ "rewards/question_recreation_reward_func": 0.4101000681519508,
197
  "rewards/soft_format_reward_func": 0.0,
198
  "rewards/strict_format_reward_func": 0.0,
199
+ "rewards/xmlcount_reward_func": 0.16210000067949296,
200
  "step": 20
201
  },
202
  {
203
+ "epoch": 4.0,
204
  "step": 20,
205
  "total_flos": 0.0,
206
+ "train_loss": 6.641451051336844e-07,
207
+ "train_runtime": 718.6924,
208
+ "train_samples_per_second": 0.445,
209
+ "train_steps_per_second": 0.028
210
  }
211
  ],
212
  "logging_steps": 2,
213
  "max_steps": 20,
214
  "num_input_tokens_seen": 0,
215
+ "num_train_epochs": 5,
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:44ab3cb72ef003efdc56d9938ef1b19649d4088db0b36611fd01fbeb752fdfc4
3
- size 5880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60d32258253d459158070110279c739eea3792bcfbf0af7487e04a785c458a2f
3
+ size 5944