Commit 057c26f9 authored by jameskrw's avatar jameskrw
Browse files

updated scripts

parent 41ad9685
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -29,8 +29,8 @@ python3 -m vagen.trainer.main_ppo \
    data.train_batch_size=128 \
    data.val_batch_size=8 \
    data.max_prompt_length=1024 \
    data.max_response_length=190 \
    data.max_trajectory_length=4080 \
    data.max_response_length=256 \
    data.max_trajectory_length=5000 \
    data.image_key=images \
    data.truncation=error \
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \
+2 −2
Original line number Diff line number Diff line
@@ -29,8 +29,8 @@ python3 -m vagen.trainer.main_ppo \
    data.train_batch_size=128 \
    data.val_batch_size=8 \
    data.max_prompt_length=1024 \
    data.max_response_length=190 \
    data.max_trajectory_length=4080 \
    data.max_response_length=256 \
    data.max_trajectory_length=5000 \
    data.image_key=images \
    data.truncation=error \
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \
+2 −2
Original line number Diff line number Diff line
@@ -29,8 +29,8 @@ python3 -m vagen.trainer.main_ppo \
    data.train_batch_size=128 \
    data.val_batch_size=8 \
    data.max_prompt_length=1024 \
    data.max_response_length=190 \
    data.max_trajectory_length=4080 \
    data.max_response_length=256 \
    data.max_trajectory_length=5000 \
    data.image_key=images \
    data.truncation=error \
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \