Commit 4a293a38 authored by jameskrw's avatar jameskrw
Browse files

minor

parent cb5a21f9

scripts/examples/crossview_grpo/env_config.yaml

+1 −0
Original line number Diff line number Diff line
@@ -11,6 +11,7 @@ env2: 
    env_config:

        split: test

        render_mode: vision

    train_size: 0

    test_size: 1200

scripts/examples/crossview_grpo/run.sh

+2 −2
Original line number Diff line number Diff line
@@ -62,8 +62,8 @@ python3 -m vagen.trainer.main_ppo \ 
    trainer.experiment_name='grpo_crossview' \

    trainer.n_gpus_per_node=4 \

    trainer.nnodes=1 \

    trainer.save_freq=100 \

    trainer.test_freq=20 \

    trainer.save_freq=150 \

    trainer.test_freq=40 \

    trainer.total_training_steps=300 \

    rollout_manager.max_turns=1 \

    rollout_manager.window_size=5 \

scripts/install.sh

+1 −0
Original line number Diff line number Diff line
@@ -32,4 +32,5 @@ echo "Installing Frozenlake dependencies" 
pip install 'gymnasium'

pip install "gymnasium[toy-text]"



pip install together # together ai api for process reward

echo "Installation complete, to install dependencies for other environments, refer to env/readme"

vagen/env/crossview/env_config.py

+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ from typing import Optional, List, Union 


@dataclass

class CrossViewEnvConfig(BaseEnvConfig):

    split="train"

    split:str ="train"

    image_dir: str = "extracted_images"

    image_size: tuple = (300, 300)

    render_mode: str = "vision"

CRA Git | Maintained and supported by SUSTech CRA and CCSE