Hugging Face's logo

Yukang
/
Qwen2.5-3B-Open-R1-Code-GRPO

Text Generation
Transformers
Safetensors
qwen2
Generated from Trainer
open-r1
trl
grpo
conversational
text-generation-inference
Model card Files
xet
Community
Yukang commited on
Commit
614c331
·
verified ·
1 Parent(s): 1fadeb3

Model save

Browse files
Files changed (7) hide show
  1. README.md +2 -4
  2. all_results.json +4 -4
  3. model-00001-of-00002.safetensors +1 -1
  4. model-00002-of-00002.safetensors +1 -1
  5. train_results.json +4 -4
  6. trainer_state.json +0 -0
  7. training_args.bin +1 -1
README.md CHANGED
@@ -1,11 +1,9 @@
1
  ---
2
  base_model: Qwen/Qwen2.5-3B-Instruct
3
- datasets: open-r1/verifiable-coding-problems-python
4
  library_name: transformers
5
  model_name: Qwen2.5-3B-Open-R1-Code-GRPO
6
  tags:
7
  - generated_from_trainer
8
- - open-r1
9
  - trl
10
  - grpo
11
  licence: license
@@ -13,7 +11,7 @@ licence: license
13
 
14
  # Model Card for Qwen2.5-3B-Open-R1-Code-GRPO
15
 
16
- This model is a fine-tuned version of [Qwen/Qwen2.5-3B-Instruct](https://huggingface.co/Qwen/Qwen2.5-3B-Instruct) on the [open-r1/verifiable-coding-problems-python](https://huggingface.co/datasets/open-r1/verifiable-coding-problems-python) dataset.
17
  It has been trained using [TRL](https://github.com/huggingface/trl).
18
 
19
  ## Quick start
@@ -29,7 +27,7 @@ print(output["generated_text"])
29
 
30
  ## Training procedure
31
 
32
- [<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/chenyukang2020-nvidia/huggingface/runs/oyriy3yi)
33
 
34
 
35
  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
  base_model: Qwen/Qwen2.5-3B-Instruct
 
3
  library_name: transformers
4
  model_name: Qwen2.5-3B-Open-R1-Code-GRPO
5
  tags:
6
  - generated_from_trainer
 
7
  - trl
8
  - grpo
9
  licence: license
 
11
 
12
  # Model Card for Qwen2.5-3B-Open-R1-Code-GRPO
13
 
14
+ This model is a fine-tuned version of [Qwen/Qwen2.5-3B-Instruct](https://huggingface.co/Qwen/Qwen2.5-3B-Instruct).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
 
27
 
28
  ## Training procedure
29
 
30
+ [<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/chenyukang2020-nvidia/huggingface/runs/c99iczg5)
31
 
32
 
33
  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": 1.0097834765617772e-05,
4
- "train_runtime": 80.5242,
5
  "train_samples": 35735,
6
- "train_samples_per_second": 3179.169,
7
- "train_steps_per_second": 6.209
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": -1.764968340101791e-06,
4
+ "train_runtime": 73.9907,
5
  "train_samples": 35735,
6
+ "train_samples_per_second": 3459.895,
7
+ "train_steps_per_second": 6.758
8
  }
model-00001-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:68f6c005f51e6799b89abb9e83cd85053f8f05a54a4f194f6cfa64fc76e42de3
3
  size 4957560304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a926a49b6a46bcd523ecf8460974cda609ec195d4a423c1030649f14261c0563
3
  size 4957560304
model-00002-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:46f21cf01ad1a1a895444093101daa61bffe73440097137650e811c072a6d2b3
3
  size 1214366696
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4df150409cdbeb78dc565e4575a3d619a75c4f02d93a55f9d1ec8c545cddae5
3
  size 1214366696
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 1.0097834765617772e-05,
4
- "train_runtime": 80.5242,
5
  "train_samples": 35735,
6
- "train_samples_per_second": 3179.169,
7
- "train_steps_per_second": 6.209
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": -1.764968340101791e-06,
4
+ "train_runtime": 73.9907,
5
  "train_samples": 35735,
6
+ "train_samples_per_second": 3459.895,
7
+ "train_steps_per_second": 6.758
8
  }
trainer_state.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:de80784f5e91d7fd3b630ab022d6b5968dd6d5ce36dbd966c41599d90a607619
3
  size 8504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eabcb0a39ec9cc2897f47bea3ba4ffbeeaf91c51933728506734e4e390ba067a
3
  size 8504