open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-27 17:23:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

first bits of veRL example

Browse source
This commit is contained in:
Andreas Koepf 2025-02-01 21:19:27 +00:00
parent bb620a1066
commit e671b97ab4
5 changed files with 562 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
examples/veRL/launch_on_4gpu.sh Executable file
View file

@ -0,0 +1,9 @@
#!/bin/bash
export N_GPUS=4
export BASE_MODEL=meta-llama/Llama-3.2-1B-Instruct
export ROLLOUT_TP_SIZE=2
export EXPERIMENT_NAME=chain_sum_llama
export VLLM_ATTENTION_BACKEND=XFORMERS
bash ./train.sh