qwen2_5_vl_3b_clevr.sh 587 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
6
7
8
set -x

MODEL_PATH=Qwen/Qwen2.5-VL-3B-Instruct  # replace it with your local file path

python3 -m verl.trainer.main \
    config=examples/config.yaml \
    data.train_files=BUAADreamer/clevr_count_70k@train \
    data.val_files=BUAADreamer/clevr_count_70k@test \
chenych's avatar
update  
chenych committed
9
    data.format_prompt=./examples/format_prompt/r1v_format.jinja \
chenych's avatar
chenych committed
10
11
    worker.actor.model.model_path=${MODEL_PATH} \
    worker.rollout.tensor_parallel_size=1 \
chenych's avatar
update  
chenych committed
12
    worker.reward.score_function=./examples/score_function/r1v.py:compute_score \
chenych's avatar
chenych committed
13
14
    trainer.experiment_name=qwen2_5_vl_3b_clevr \
    trainer.n_gpus_per_node=2