Jaswanth1210 commited on
Commit
0d411fb
·
1 Parent(s): 17a9ff7

fix: GRPO batch_size must be divisible by num_generations (1→4)

Browse files
Files changed (1) hide show
  1. 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=1)
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()