Skip to content

Commit

Permalink
Add command
Browse files Browse the repository at this point in the history
  • Loading branch information
lewtun committed Feb 19, 2025
1 parent fb6e4ae commit 89ded43
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,13 @@ Then make sure your dataset contains a `verification_info` column with the follo
}
```

See the [training config](./recipes/Qwen2.5-1.5B-Instruct/grpo/config_demo_code.yaml) for an example with `Qwen2.5-1.5B-Instruct`.
For example, to train a smol model on Python problems, run:

```shell
ACCELERATE_LOG_LEVEL=info accelerate launch --config_file recipes/accelerate_configs/zero2.yaml \
--num_processes=7 src/open_r1/grpo.py \
--config recipes/Qwen2.5-1.5B-Instruct/grpo/config_demo_code.yaml
```

### Launching jobs on a Slurm cluster

Expand Down

0 comments on commit 89ded43

Please sign in to comment.