"test/srt/test_custom_allreduce.py" did not exist on "a53454c55e222fcc7375676b665b7e1276464170"
qwen2_5_vl_7b_geo3k_swanlab.sh 462 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
#!/bin/bash

chenych's avatar
chenych committed
3
4
set -x

chenych's avatar
chenych committed
5
6
export PYTHONUNBUFFERED=1

chenych's avatar
chenych committed
7
8
9
10
11
12
13
14
15
16
MODEL_PATH=Qwen/Qwen2.5-VL-7B-Instruct  # replace it with your local file path

python3 -m verl.trainer.main \
    config=examples/config.yaml \
    data.train_files=hiyouga/geometry3k@train \
    data.val_files=hiyouga/geometry3k@test \
    worker.actor.model.model_path=${MODEL_PATH} \
    trainer.experiment_name=qwen2_5_vl_7b_geo_grpo \
    trainer.logger=['console','swanlab'] \
    trainer.n_gpus_per_node=8