Commit 5f0528aa authored by jameskrw's avatar jameskrw
Browse files

minor

parent 9cc8319b
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -64,7 +64,7 @@ python3 -m vagen.trainer.main_ppo \
    trainer.nnodes=1 \
    trainer.save_freq=100 \
    trainer.test_freq=20 \
    trainer.total_epochs=15 \
    trainer.total_training_steps=300 \
    rollout_manager.max_turns=3 \
    rollout_manager.window_size=5 \
    rollout_manager.use_multi_turn_reward=False \
+1 −1
Original line number Diff line number Diff line
@@ -3,4 +3,4 @@ env1:
    env_config:
        render_mode: text
    train_size: 10000  
    test_size: 500
    test_size: 1024
+2 −2
Original line number Diff line number Diff line
@@ -19,7 +19,7 @@ python3 -m vagen.trainer.main_ppo \
    data.train_files=data/sokoban-text-debug/train.parquet \
    data.val_files=data/sokoban-text-debug/test.parquet \
    data.train_batch_size=16 \
    data.val_batch_size=128 \
    data.val_batch_size=256 \
    data.max_prompt_length=1024 \
    data.max_response_length=128 \
    data.max_trajectory_length=1800 \
@@ -64,7 +64,7 @@ python3 -m vagen.trainer.main_ppo \
    trainer.nnodes=1 \
    trainer.save_freq=100 \
    trainer.test_freq=20 \
    trainer.total_epochs=15 \
    trainer.total_training_steps=300 \
    rollout_manager.max_turns=3 \
    rollout_manager.window_size=5 \
    rollout_manager.use_multi_turn_reward=False \
+1 −1
Original line number Diff line number Diff line
@@ -3,4 +3,4 @@ env1:
    env_config:
        render_mode: vision
    train_size: 10000  
    test_size: 500
 No newline at end of file
    test_size: 1024
 No newline at end of file
Loading