Commit 47b08dcd authored by jameskrw's avatar jameskrw
Browse files

updated server yaml

parent 340bb4fb
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -26,8 +26,8 @@ python3 -m vagen.trainer.main_ppo \
    data.val_files=data/$EXPERIMENT_NAME/test.parquet \
    data.train_batch_size=128 \
    data.max_prompt_length=1024 \
    data.max_response_length=648 \
    data.max_trajectory_length=3000 \
    data.max_response_length=750 \
    data.max_trajectory_length=3500 \
    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
@@ -77,8 +77,8 @@ tmux send-keys -t "$TRAIN_SESSION" "python3 -m vagen.trainer.main_ppo \\
    data.val_files=data/$EXPERIMENT_NAME/test.parquet \\
    data.train_batch_size=128 \\
    data.max_prompt_length=1024 \\
    data.max_response_length=648 \\
    data.max_trajectory_length=3000 \\
    data.max_response_length=750 \\
    data.max_trajectory_length=3500 \\
    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
@@ -77,8 +77,8 @@ tmux send-keys -t "$TRAIN_SESSION" "python3 -m vagen.trainer.main_ppo \\
    data.val_files=data/$EXPERIMENT_NAME/test.parquet \\
    data.train_batch_size=128 \\
    data.max_prompt_length=1024 \\
    data.max_response_length=648 \\
    data.max_trajectory_length=3000 \\
    data.max_response_length=750 \\
    data.max_trajectory_length=3500 \\
    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
@@ -26,8 +26,8 @@ python3 -m vagen.trainer.main_ppo \
    data.val_files=data/$EXPERIMENT_NAME/test.parquet \
    data.train_batch_size=128 \
    data.max_prompt_length=1024 \
    data.max_response_length=648 \
    data.max_trajectory_length=3000 \
    data.max_response_length=750 \
    data.max_trajectory_length=3500 \
    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
@@ -77,8 +77,8 @@ tmux send-keys -t "$TRAIN_SESSION" "python3 -m vagen.trainer.main_ppo \\
    data.val_files=data/$EXPERIMENT_NAME/test.parquet \\
    data.train_batch_size=128 \\
    data.max_prompt_length=1024 \\
    data.max_response_length=648 \\
    data.max_trajectory_length=3000 \\
    data.max_response_length=750 \\
    data.max_trajectory_length=3500 \\
    data.image_key=images \\
    data.truncation=error \\
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \\
Loading