Commit cb5a21f9 authored by jameskrw's avatar jameskrw
Browse files

udpated cross view

parent 095645e6
Loading
Loading
Loading
Loading
+9 −1
Original line number Diff line number Diff line
env1:
    env_name: crossview
    env_config:
        split: train
        render_mode: vision
    train_size: 10000
    test_size: 512
    test_size: 0

env2:
    env_name: crossview
    env_config:
        split: test
        render_mode: vision
    test_size: 1200
    
+2 −1
Original line number Diff line number Diff line
@@ -18,7 +18,8 @@ python3 -m vagen.trainer.main_ppo \
    algorithm.high_level_gamma=0.95 \
    data.train_files=data/crossview/train.parquet \
    data.val_files=data/crossview/test.parquet \
    data.train_batch_size=64 \
    data.train_batch_size=32 \
    data.val_batch_size=256 \
    data.max_prompt_length=1024 \
    data.max_response_length=648 \
    data.max_trajectory_length=3600 \