Skip to content

Commit affd9a1

Browse files
committed
modify default value of run_ppo_trainer_veomni.sh to prevent misunderstanding
Signed-off-by: 0oshowero0 <o0shower0o@outlook.com> fix Signed-off-by: 0oshowero0 <o0shower0o@outlook.com>
1 parent 2815eea commit affd9a1

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

.github/workflows/e2e_ppo_trainer_veomni_vllm.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ jobs:
134134
- name: Running GEO3K E2E training tests on 8 L20 GPUs with veomni engine (FSDP_SIZE=8, USP=1)
135135
run: |
136136
ray stop --force
137-
MODEL_ID=Qwen/Qwen3-VL-2B-Instruct TRAIN_FILES=${HOME}/data/geo3k/train.parquet VAL_FILES=${HOME}/data/gsm8k/test.parquet VAL_BEFORE_TRAIN=True NUM_GPUS=8 FSDP_SIZE=4 SP_SIZE=2 EP_SIZE=1 VERL_EXP_NAME="qwen3-2b-vl-function-reward-minimal-fsdp-size8" bash tests/special_e2e/run_ppo_trainer_veomni.sh
137+
bash tests/special_e2e/run_ppo_trainer_veomni.sh
138138
139139
cleanup:
140140
runs-on: ubuntu-latest

tests/special_e2e/run_ppo_trainer_veomni.sh

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,18 @@
22
set -xeuo pipefail
33

44
# Download model if not exists
5-
MODEL_ID=${MODEL_ID:-Qwen/Qwen2.5-0.5B-Instruct}
5+
MODEL_ID=${MODEL_ID:-Qwen/Qwen3-VL-2B-Instruct}
66
MODEL_PATH=${MODEL_PATH:-${HOME}/models/${MODEL_ID}}
77
#huggingface-cli download "${MODEL_ID}" --local-dir "${MODEL_PATH}"
88

9-
TRAIN_FILES=${TRAIN_FILES:-${HOME}/data/gsm8k/train.parquet}
10-
VAL_FILES=${VAL_FILES:-${HOME}/data/gsm8k/test.parquet}
11-
VAL_BEFORE_TRAIN=${VAL_BEFORE_TRAIN:-False}
9+
TRAIN_FILES=${TRAIN_FILES:-${HOME}/data/geo3k/train.parquet}
10+
VAL_FILES=${VAL_FILES:-${HOME}/data/geo3k/test.parquet}
11+
VAL_BEFORE_TRAIN=${VAL_BEFORE_TRAIN:-True}
1212
NUM_GPUS=${NUM_GPUS:-8}
1313
FSDP_SIZE=${FSDP_SIZE:-4}
1414
SP_SIZE=${SP_SIZE:-2}
15-
EP_SIZE=${EP_SIZE:-2}
16-
VERL_EXP_NAME=${VERL_EXP_NAME:-qwen2.5-0.5b-function-reward-minimal-fsdp-size8}
15+
EP_SIZE=${EP_SIZE:-1}
16+
VERL_EXP_NAME=${VERL_EXP_NAME:-qwen3-2b-vl-function-reward-minimal-fsdp-size8}
1717

1818
python3 -m verl.trainer.main_ppo \
1919
model_engine=veomni \

0 commit comments

Comments
 (0)