Spaces:
Running
Running
Commit ·
0d411fb
1
Parent(s): 17a9ff7
fix: GRPO batch_size must be divisible by num_generations (1→4)
Browse files- train/grpo_train.py +1 -1
train/grpo_train.py
CHANGED
|
@@ -43,7 +43,7 @@ def _parse_args() -> argparse.Namespace:
|
|
| 43 |
p.add_argument("--save-every", type=int, default=100)
|
| 44 |
p.add_argument("--log-to", choices=["jsonl", "wandb"], default="jsonl")
|
| 45 |
p.add_argument("--model", type=str, default="Qwen/Qwen2.5-1.5B-Instruct")
|
| 46 |
-
p.add_argument("--batch-size", type=int, default=
|
| 47 |
p.add_argument("--num-generations", type=int, default=4)
|
| 48 |
p.add_argument("--seed", type=int, default=42)
|
| 49 |
return p.parse_args()
|
|
|
|
| 43 |
p.add_argument("--save-every", type=int, default=100)
|
| 44 |
p.add_argument("--log-to", choices=["jsonl", "wandb"], default="jsonl")
|
| 45 |
p.add_argument("--model", type=str, default="Qwen/Qwen2.5-1.5B-Instruct")
|
| 46 |
+
p.add_argument("--batch-size", type=int, default=4)
|
| 47 |
p.add_argument("--num-generations", type=int, default=4)
|
| 48 |
p.add_argument("--seed", type=int, default=42)
|
| 49 |
return p.parse_args()
|