KMasaki commited on
Commit
78864cd
·
verified ·
1 Parent(s): 7fcfacd

Model save

Browse files
Files changed (4) hide show
  1. README.md +2 -4
  2. all_results.json +4 -4
  3. train_results.json +4 -4
  4. trainer_state.json +0 -0
README.md CHANGED
@@ -1,10 +1,8 @@
1
  ---
2
- datasets: open-r1/OpenR1-Math-220k
3
  library_name: transformers
4
  model_name: Qwen2.5-1.5B-Open-R1-GRPO
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 Qwen2.5-1.5B-Open-R1-GRPO
14
 
15
- This model is a fine-tuned version of [None](https://huggingface.co/None) on the [open-r1/OpenR1-Math-220k](https://huggingface.co/datasets/open-r1/OpenR1-Math-220k) 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/kawamuramasaki/open-r1/runs/p1aly7g1)
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: Qwen2.5-1.5B-Open-R1-GRPO
4
  tags:
5
  - generated_from_trainer
 
6
  - trl
7
  - grpo
8
  licence: license
 
10
 
11
  # Model Card for Qwen2.5-1.5B-Open-R1-GRPO
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/kawamuramasaki/open-r1/runs/jxm7gcj0)
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": 0.04007197510112119,
4
- "train_runtime": 111902.1384,
5
  "train_samples": 93733,
6
- "train_samples_per_second": 0.838,
7
- "train_steps_per_second": 0.03
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 0.08076191042202922,
4
+ "train_runtime": 127083.2818,
5
  "train_samples": 93733,
6
+ "train_samples_per_second": 0.738,
7
+ "train_steps_per_second": 0.026
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "total_flos": 0.0,
3
- "train_loss": 0.04007197510112119,
4
- "train_runtime": 111902.1384,
5
  "train_samples": 93733,
6
- "train_samples_per_second": 0.838,
7
- "train_steps_per_second": 0.03
8
  }
 
1
  {
2
  "total_flos": 0.0,
3
+ "train_loss": 0.08076191042202922,
4
+ "train_runtime": 127083.2818,
5
  "train_samples": 93733,
6
+ "train_samples_per_second": 0.738,
7
+ "train_steps_per_second": 0.026
8
  }
trainer_state.json CHANGED
The diff for this file is too large to render. See raw diff