MatchaLwc commited on
Commit
2413042
·
verified ·
1 Parent(s): 0d707cd

Model save

Browse files
README.md CHANGED
@@ -1,10 +1,8 @@
1
  ---
2
- datasets: DigitalLearningGmbH/MATH-lighteval
3
  library_name: transformers
4
  model_name: qwen25-Math-7b-3epoch
5
  tags:
6
  - generated_from_trainer
7
- - open-r1
8
  - trl
9
  - grpo
10
  licence: license
@@ -12,7 +10,7 @@ licence: license
12
 
13
  # Model Card for qwen25-Math-7b-3epoch
14
 
15
- This model is a fine-tuned version of [None](https://huggingface.co/None) on the [DigitalLearningGmbH/MATH-lighteval](https://huggingface.co/datasets/DigitalLearningGmbH/MATH-lighteval) dataset.
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
18
  ## Quick start
@@ -28,7 +26,7 @@ print(output["generated_text"])
28
 
29
  ## Training procedure
30
 
31
- [<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/1105645918-bit/huggingface/runs/4lyg3uex)
32
 
33
 
34
  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).
 
1
  ---
 
2
  library_name: transformers
3
  model_name: qwen25-Math-7b-3epoch
4
  tags:
5
  - generated_from_trainer
 
6
  - trl
7
  - grpo
8
  licence: license
 
10
 
11
  # Model Card for qwen25-Math-7b-3epoch
12
 
13
+ This model is a fine-tuned version of [None](https://huggingface.co/None).
14
  It has been trained using [TRL](https://github.com/huggingface/trl).
15
 
16
  ## Quick start
 
26
 
27
  ## Training procedure
28
 
29
+ [<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/1105645918-bit/huggingface/runs/cj1e0d7i)
30
 
31
 
32
  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).
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 340917.58753671293,
4
- "train_runtime": 20435.4937,
5
  "train_samples": 7500,
6
- "train_samples_per_second": 0.367,
7
  "train_steps_per_second": 0.003
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 822442.3794969221,
4
+ "train_runtime": 20401.2733,
5
  "train_samples": 7500,
6
+ "train_samples_per_second": 0.368,
7
  "train_steps_per_second": 0.003
8
  }
config.json CHANGED
@@ -23,7 +23,7 @@
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.49.0",
26
- "use_cache": true,
27
  "use_mrope": false,
28
  "use_sliding_window": false,
29
  "vocab_size": 152064
 
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.49.0",
26
+ "use_cache": false,
27
  "use_mrope": false,
28
  "use_sliding_window": false,
29
  "vocab_size": 152064
model-00001-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:41520b81a9c3d56c53546892aa71f706b8205615b76b191808f1df22b519f73e
3
  size 4877660776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95d0e9f55dafb4fba4b654413ccac0799d5ecc0923de9f38a639afa7bd3c35cb
3
  size 4877660776
model-00002-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:510bbaf6488bc84d772d6d484d2ac5f0e8c4ec64e787e8e30765ce6910d62057
3
  size 4932751008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb0e9121a24ad88917e86451c50cf72a4815fd67da828d5a8e32fc0074599ea9
3
  size 4932751008
model-00003-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d6936226b04fd0568523668398ec3d4de37a13f92ae88bcd78979881ad780694
3
  size 4330865200
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:493b1971e3661a1c734c54feba073f3bbc8cab483062d1b2748f13009a6d8c3e
3
  size 4330865200
model-00004-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d4c3350dea854c7937e01daf970a5458ec24c855b8ea029a52ea5ffda5fd2398
3
  size 1089994880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cb01982f51589497085b5c0f412bc6cccef24d303a0f8f4658acf5c05bde000
3
  size 1089994880
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 340917.58753671293,
4
- "train_runtime": 20435.4937,
5
  "train_samples": 7500,
6
- "train_samples_per_second": 0.367,
7
  "train_steps_per_second": 0.003
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 822442.3794969221,
4
+ "train_runtime": 20401.2733,
5
  "train_samples": 7500,
6
+ "train_samples_per_second": 0.368,
7
  "train_steps_per_second": 0.003
8
  }
trainer_state.json CHANGED
@@ -12,13 +12,13 @@
12
  "clip_ratio": 0.0,
13
  "completion_length": 1024.0,
14
  "epoch": 0.017057569296375266,
15
- "grad_norm": 0.09757333248853683,
16
  "kl": 0.0,
17
  "learning_rate": 5e-07,
18
  "loss": 0.0,
19
- "reward": 0.20982143841683865,
20
- "reward_std": 0.3133961074054241,
21
- "rewards/accuracy_reward": 0.20982143841683865,
22
  "rewards/format_reward": 0.0,
23
  "step": 1
24
  },
@@ -26,13 +26,13 @@
26
  "clip_ratio": 0.0,
27
  "completion_length": 1024.0,
28
  "epoch": 0.08528784648187633,
29
- "grad_norm": 0.09003066271543503,
30
- "kl": 0.00010742247104644775,
31
  "learning_rate": 2.5e-06,
32
  "loss": 0.0,
33
- "reward": 0.23046876164153218,
34
- "reward_std": 0.3150851926766336,
35
- "rewards/accuracy_reward": 0.23046876164153218,
36
  "rewards/format_reward": 0.0,
37
  "step": 5
38
  },
@@ -40,13 +40,13 @@
40
  "clip_ratio": 0.0,
41
  "completion_length": 1024.0,
42
  "epoch": 0.17057569296375266,
43
- "grad_norm": 29.32818031311035,
44
- "kl": 0.004455947875976562,
45
  "learning_rate": 2.956412726139078e-06,
46
- "loss": 0.0002,
47
- "reward": 0.44888395071029663,
48
- "reward_std": 0.33523476012051107,
49
- "rewards/accuracy_reward": 0.44888395071029663,
50
  "rewards/format_reward": 0.0,
51
  "step": 10
52
  },
@@ -54,13 +54,13 @@
54
  "clip_ratio": 0.0,
55
  "completion_length": 1024.0,
56
  "epoch": 0.255863539445629,
57
- "grad_norm": 0.0765397846698761,
58
- "kl": 0.010036468505859375,
59
  "learning_rate": 2.7836719084521715e-06,
60
- "loss": 0.0004,
61
- "reward": 0.7040178865194321,
62
- "reward_std": 0.20274759102612733,
63
- "rewards/accuracy_reward": 0.7040178865194321,
64
  "rewards/format_reward": 0.0,
65
  "step": 15
66
  },
@@ -68,13 +68,13 @@
68
  "clip_ratio": 0.0,
69
  "completion_length": 1024.0,
70
  "epoch": 0.3411513859275053,
71
- "grad_norm": 0.06571836024522781,
72
- "kl": 0.01729736328125,
73
  "learning_rate": 2.4946839873611927e-06,
74
- "loss": 0.0007,
75
- "reward": 0.7073661059141159,
76
- "reward_std": 0.18286588205955923,
77
- "rewards/accuracy_reward": 0.7073661059141159,
78
  "rewards/format_reward": 0.0,
79
  "step": 20
80
  },
@@ -82,13 +82,13 @@
82
  "clip_ratio": 0.0,
83
  "completion_length": 1024.0,
84
  "epoch": 0.42643923240938164,
85
- "grad_norm": 0.08071992546319962,
86
- "kl": 0.012841796875,
87
  "learning_rate": 2.1156192081791355e-06,
88
- "loss": 0.0005,
89
- "reward": 0.7102678894996644,
90
- "reward_std": 0.18361264672130345,
91
- "rewards/accuracy_reward": 0.7102678894996644,
92
  "rewards/format_reward": 0.0,
93
  "step": 25
94
  },
@@ -96,13 +96,13 @@
96
  "clip_ratio": 0.0,
97
  "completion_length": 1024.0,
98
  "epoch": 0.511727078891258,
99
- "grad_norm": 0.056903496384620667,
100
- "kl": 0.01241302490234375,
101
  "learning_rate": 1.6808050203829845e-06,
102
- "loss": 0.0005,
103
- "reward": 0.691517885029316,
104
- "reward_std": 0.18483126908540726,
105
- "rewards/accuracy_reward": 0.691517885029316,
106
  "rewards/format_reward": 0.0,
107
  "step": 30
108
  },
@@ -110,13 +110,13 @@
110
  "clip_ratio": 0.0,
111
  "completion_length": 1024.0,
112
  "epoch": 0.5970149253731343,
113
- "grad_norm": 0.07405390590429306,
114
- "kl": 0.010370635986328125,
115
  "learning_rate": 1.2296174432791415e-06,
116
- "loss": 0.0004,
117
- "reward": 0.6821428924798966,
118
- "reward_std": 0.19620487112551926,
119
- "rewards/accuracy_reward": 0.6821428924798966,
120
  "rewards/format_reward": 0.0,
121
  "step": 35
122
  },
@@ -124,13 +124,13 @@
124
  "clip_ratio": 0.0,
125
  "completion_length": 1024.0,
126
  "epoch": 0.6823027718550106,
127
- "grad_norm": 0.06446688622236252,
128
- "kl": 0.03368301391601562,
129
  "learning_rate": 8.029152419343472e-07,
130
- "loss": 0.0014,
131
- "reward": 0.6912946753203869,
132
- "reward_std": 0.1812687825411558,
133
- "rewards/accuracy_reward": 0.6912946753203869,
134
  "rewards/format_reward": 0.0,
135
  "step": 40
136
  },
@@ -138,13 +138,13 @@
138
  "clip_ratio": 0.0,
139
  "completion_length": 1024.0,
140
  "epoch": 0.767590618336887,
141
- "grad_norm": 0.06644736230373383,
142
- "kl": 0.011377716064453125,
143
  "learning_rate": 4.3933982822017883e-07,
144
- "loss": 0.0005,
145
- "reward": 0.688169676065445,
146
- "reward_std": 0.18324931319803,
147
- "rewards/accuracy_reward": 0.688169676065445,
148
  "rewards/format_reward": 0.0,
149
  "step": 45
150
  },
@@ -152,13 +152,13 @@
152
  "clip_ratio": 0.0,
153
  "completion_length": 1024.0,
154
  "epoch": 0.8528784648187633,
155
- "grad_norm": 73763676160.0,
156
- "kl": 99405360.00854492,
157
  "learning_rate": 1.718159615201853e-07,
158
- "loss": 3954644.0,
159
- "reward": 0.6783482417464256,
160
- "reward_std": 0.18882247591391205,
161
- "rewards/accuracy_reward": 0.6783482417464256,
162
  "rewards/format_reward": 0.0,
163
  "step": 50
164
  },
@@ -166,13 +166,13 @@
166
  "clip_ratio": 0.0,
167
  "completion_length": 1024.0,
168
  "epoch": 0.9381663113006397,
169
- "grad_norm": 0.6496619582176208,
170
- "kl": 0.010959625244140625,
171
  "learning_rate": 2.4570139579284723e-08,
172
- "loss": 0.0004,
173
- "reward": 0.7225446715950966,
174
- "reward_std": 0.17012470280751585,
175
- "rewards/accuracy_reward": 0.7225446715950966,
176
  "rewards/format_reward": 0.0,
177
  "step": 55
178
  },
@@ -180,16 +180,16 @@
180
  "clip_ratio": 0.0,
181
  "completion_length": 1024.0,
182
  "epoch": 0.9893390191897654,
183
- "kl": 0.009393056233723959,
184
- "reward": 0.6990327723324299,
185
- "reward_std": 0.184214538273712,
186
- "rewards/accuracy_reward": 0.6990327723324299,
187
  "rewards/format_reward": 0.0,
188
  "step": 58,
189
  "total_flos": 0.0,
190
- "train_loss": 340917.58753671293,
191
- "train_runtime": 20435.4937,
192
- "train_samples_per_second": 0.367,
193
  "train_steps_per_second": 0.003
194
  }
195
  ],
 
12
  "clip_ratio": 0.0,
13
  "completion_length": 1024.0,
14
  "epoch": 0.017057569296375266,
15
+ "grad_norm": 0.09009335935115814,
16
  "kl": 0.0,
17
  "learning_rate": 5e-07,
18
  "loss": 0.0,
19
+ "reward": 0.2321428693830967,
20
+ "reward_std": 0.3139922395348549,
21
+ "rewards/accuracy_reward": 0.2321428693830967,
22
  "rewards/format_reward": 0.0,
23
  "step": 1
24
  },
 
26
  "clip_ratio": 0.0,
27
  "completion_length": 1024.0,
28
  "epoch": 0.08528784648187633,
29
+ "grad_norm": 0.07332542538642883,
30
+ "kl": 9.924173355102539e-05,
31
  "learning_rate": 2.5e-06,
32
  "loss": 0.0,
33
+ "reward": 0.22935269074514508,
34
+ "reward_std": 0.30691308877430856,
35
+ "rewards/accuracy_reward": 0.22935269074514508,
36
  "rewards/format_reward": 0.0,
37
  "step": 5
38
  },
 
40
  "clip_ratio": 0.0,
41
  "completion_length": 1024.0,
42
  "epoch": 0.17057569296375266,
43
+ "grad_norm": 0.0727960467338562,
44
+ "kl": 0.0027547836303710937,
45
  "learning_rate": 2.956412726139078e-06,
46
+ "loss": 0.0001,
47
+ "reward": 0.4187500193715096,
48
+ "reward_std": 0.33326763547956945,
49
+ "rewards/accuracy_reward": 0.4187500193715096,
50
  "rewards/format_reward": 0.0,
51
  "step": 10
52
  },
 
54
  "clip_ratio": 0.0,
55
  "completion_length": 1024.0,
56
  "epoch": 0.255863539445629,
57
+ "grad_norm": 0.06532283872365952,
58
+ "kl": 0.17901992797851562,
59
  "learning_rate": 2.7836719084521715e-06,
60
+ "loss": 0.0072,
61
+ "reward": 0.7390625298023223,
62
+ "reward_std": 0.17226867079734803,
63
+ "rewards/accuracy_reward": 0.7390625298023223,
64
  "rewards/format_reward": 0.0,
65
  "step": 15
66
  },
 
68
  "clip_ratio": 0.0,
69
  "completion_length": 1024.0,
70
  "epoch": 0.3411513859275053,
71
+ "grad_norm": 0.08123523741960526,
72
+ "kl": 0.020281982421875,
73
  "learning_rate": 2.4946839873611927e-06,
74
+ "loss": 0.0008,
75
+ "reward": 0.7357143208384513,
76
+ "reward_std": 0.16036716918461025,
77
+ "rewards/accuracy_reward": 0.7357143208384513,
78
  "rewards/format_reward": 0.0,
79
  "step": 20
80
  },
 
82
  "clip_ratio": 0.0,
83
  "completion_length": 1024.0,
84
  "epoch": 0.42643923240938164,
85
+ "grad_norm": 98653831168.0,
86
+ "kl": 209715200.01299286,
87
  "learning_rate": 2.1156192081791355e-06,
88
+ "loss": 8388608.0,
89
+ "reward": 0.731919676065445,
90
+ "reward_std": 0.15736169517040252,
91
+ "rewards/accuracy_reward": 0.731919676065445,
92
  "rewards/format_reward": 0.0,
93
  "step": 25
94
  },
 
96
  "clip_ratio": 0.0,
97
  "completion_length": 1024.0,
98
  "epoch": 0.511727078891258,
99
+ "grad_norm": 13029.787109375,
100
+ "kl": 67.636962890625,
101
  "learning_rate": 1.6808050203829845e-06,
102
+ "loss": 2.7023,
103
+ "reward": 0.7205357491970062,
104
+ "reward_std": 0.1578733256086707,
105
+ "rewards/accuracy_reward": 0.7205357491970062,
106
  "rewards/format_reward": 0.0,
107
  "step": 30
108
  },
 
110
  "clip_ratio": 0.0,
111
  "completion_length": 1024.0,
112
  "epoch": 0.5970149253731343,
113
+ "grad_norm": 0.39150699973106384,
114
+ "kl": 2.8228424072265623,
115
  "learning_rate": 1.2296174432791415e-06,
116
+ "loss": 0.1129,
117
+ "reward": 0.7111607402563095,
118
+ "reward_std": 0.16545333191752434,
119
+ "rewards/accuracy_reward": 0.7111607402563095,
120
  "rewards/format_reward": 0.0,
121
  "step": 35
122
  },
 
124
  "clip_ratio": 0.0,
125
  "completion_length": 1024.0,
126
  "epoch": 0.6823027718550106,
127
+ "grad_norm": 91.16580963134766,
128
+ "kl": 28502426.75253143,
129
  "learning_rate": 8.029152419343472e-07,
130
+ "loss": 1146119.4,
131
+ "reward": 0.7203125312924386,
132
+ "reward_std": 0.16900291871279477,
133
+ "rewards/accuracy_reward": 0.7203125312924386,
134
  "rewards/format_reward": 0.0,
135
  "step": 40
136
  },
 
138
  "clip_ratio": 0.0,
139
  "completion_length": 1024.0,
140
  "epoch": 0.767590618336887,
141
+ "grad_norm": 552.0439453125,
142
+ "kl": 116332.38398895264,
143
  "learning_rate": 4.3933982822017883e-07,
144
+ "loss": 4652.1254,
145
+ "reward": 0.7176339611411094,
146
+ "reward_std": 0.1728838672861457,
147
+ "rewards/accuracy_reward": 0.7176339611411094,
148
  "rewards/format_reward": 0.0,
149
  "step": 45
150
  },
 
152
  "clip_ratio": 0.0,
153
  "completion_length": 1024.0,
154
  "epoch": 0.8528784648187633,
155
+ "grad_norm": 0.14540241658687592,
156
+ "kl": 801.6095245361328,
157
  "learning_rate": 1.718159615201853e-07,
158
+ "loss": 32.0004,
159
+ "reward": 0.7058035969734192,
160
+ "reward_std": 0.1750278390944004,
161
+ "rewards/accuracy_reward": 0.7058035969734192,
162
  "rewards/format_reward": 0.0,
163
  "step": 50
164
  },
 
166
  "clip_ratio": 0.0,
167
  "completion_length": 1024.0,
168
  "epoch": 0.9381663113006397,
169
+ "grad_norm": 4.781561851501465,
170
+ "kl": 22835.21565475464,
171
  "learning_rate": 2.4570139579284723e-08,
172
+ "loss": 914.4983,
173
+ "reward": 0.7439732491970062,
174
+ "reward_std": 0.16849460527300836,
175
+ "rewards/accuracy_reward": 0.7439732491970062,
176
  "rewards/format_reward": 0.0,
177
  "step": 55
178
  },
 
180
  "clip_ratio": 0.0,
181
  "completion_length": 1024.0,
182
  "epoch": 0.9893390191897654,
183
+ "kl": 1372.5967407226562,
184
+ "reward": 0.7388393208384514,
185
+ "reward_std": 0.16757233006258807,
186
+ "rewards/accuracy_reward": 0.7388393208384514,
187
  "rewards/format_reward": 0.0,
188
  "step": 58,
189
  "total_flos": 0.0,
190
+ "train_loss": 822442.3794969221,
191
+ "train_runtime": 20401.2733,
192
+ "train_samples_per_second": 0.368,
193
  "train_steps_per_second": 0.003
194
  }
195
  ],
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:701b5383a7c085d106ec2fbb38b885b899773de758141018d0e779bcfbde47d7
3
  size 8184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:249ffb335e211dcd81001eb25e5b0037e40f4ebe8607c3ac99fb99e7d7750469
3
  size 8184