Commit 43612654 authored by jameskrw's avatar jameskrw
Browse files

minor

parent c4b89752
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=150 \
    data.max_trajectory_length=4000 \
    data.max_response_length=200 \
    data.max_trajectory_length=4080 \
    data.image_key=images \
    data.truncation=error \
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \
+3 −3
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=150 \
    data.max_trajectory_length=4000 \
    data.max_response_length=200 \
    data.max_trajectory_length=4080 \
    data.image_key=images \
    data.truncation=error \
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \
+3 −3
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=150 \
    data.max_trajectory_length=4000 \
    data.max_response_length=200 \
    data.max_trajectory_length=4080 \
    data.image_key=images \
    data.truncation=error \
    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \
+1 −1
Original line number Diff line number Diff line
@@ -13,7 +13,7 @@ svg:
  model_size: "small"
navigation:
  max_workers: 48
  devices: [0,1,2,4,5,6,7]
  devices: [0,1,2,3]
primitive_skill:
  max_workers: 48