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
799ead5
·
verified ·
1 Parent(s): 00f6eed

Model save

Browse files
Files changed (8) hide show
  1. README.md +2 -4
  2. all_results.json +4 -4
  3. config.json +1 -1
  4. model-00001-of-00002.safetensors +1 -1
  5. model-00002-of-00002.safetensors +1 -1
  6. train_results.json +4 -4
  7. trainer_state.json +0 -0
  8. 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/brzs1l41)
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/oyriy3yi)
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": 9.200112302686571e-07,
4
- "train_runtime": 87.5191,
5
  "train_samples": 35735,
6
- "train_samples_per_second": 2925.077,
7
- "train_steps_per_second": 5.713
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
  }
config.json CHANGED
@@ -22,7 +22,7 @@
22
  "tie_word_embeddings": true,
23
  "torch_dtype": "bfloat16",
24
  "transformers_version": "4.52.3",
25
- "use_cache": true,
26
  "use_sliding_window": false,
27
  "vocab_size": 151936
28
  }
 
22
  "tie_word_embeddings": true,
23
  "torch_dtype": "bfloat16",
24
  "transformers_version": "4.52.3",
25
+ "use_cache": false,
26
  "use_sliding_window": false,
27
  "vocab_size": 151936
28
  }
model-00001-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:807f52796021308bdd2d6e1777c56cea61d79dcc4f52f7fe0a2982d630d096e1
3
  size 4957560304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f4573665232eebed795751922f0b4277040b5f8179a3203b6939b283baa2dee
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:dcceea148a083eef0c2d5de57f6c8b4b5eb6fcec6d1e69d13f6439bb9118b6af
3
  size 1214366696
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ea33d08f9a46ff3ca7ae559ce8964f7e393c5014b3aa32142be908a74e6a09a
3
  size 1214366696
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 9.200112302686571e-07,
4
- "train_runtime": 87.5191,
5
  "train_samples": 35735,
6
- "train_samples_per_second": 2925.077,
7
- "train_steps_per_second": 5.713
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
  }
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:b7eebea411dcd742359401d9414b70d4960ef23a73f06ce6752cda6bf7a9eea9
3
  size 8504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c1f3a29128f5b456b5d8a5db89ea572c1fba6cb3fb8576c6af5680cbefcba54
3
  size 8504