Loading vagen/examples/frozen_lake_aico_service/run.sh +1 −1 Original line number Diff line number Diff line Loading @@ -5,7 +5,7 @@ export VLLM_ATTENTION_BACKEND=XFORMERS export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/frozenlake-vision-debug/train.parquet" \ Loading vagen/examples/navigation_aico/run.sh +2 −0 Original line number Diff line number Diff line Loading @@ -6,6 +6,8 @@ export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/navigation-vision-debug/train.parquet" \ Loading vagen/examples/primitive_skill_aico/env_config.yaml +4 −4 Original line number Diff line number Diff line Loading @@ -4,25 +4,25 @@ env1: render_mode: text env_id: "AlignTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 test_size: 8 env2: env_name: primitive_skill env_config: render_mode: text env_id: "PlaceTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 test_size: 8 env3: env_name: primitive_skill env_config: render_mode: text env_id: "PutAppleInDrawer" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 test_size: 8 env4: env_name: primitive_skill env_config: render_mode: text env_id: "StackThreeCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 No newline at end of file test_size: 8 No newline at end of file vagen/examples/primitive_skill_aico/run.sh +4 −1 Original line number Diff line number Diff line Loading @@ -6,6 +6,8 @@ export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server host=127.0.0.1 port=5001 in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/primitive_skill-vision-debug/train.parquet" \ Loading @@ -19,7 +21,7 @@ python3 -m vagen.trainer.main_ppo \ algorithm.high_level_gamma=0.95 \ data.train_files=data/primitive_skill-vision-debug/train.parquet \ data.val_files=data/primitive_skill-vision-debug/test.parquet \ data.train_batch_size=64 \ data.train_batch_size=32 \ data.val_batch_size=32 \ data.max_prompt_length=1024 \ data.max_response_length=128 \ Loading Loading @@ -76,4 +78,5 @@ python3 -m vagen.trainer.main_ppo \ rollout_manager.n_trajectory=2 \ rollout_manager.use_service=True \ rollout_manager.timeout=240 \ rollout_manager.base_url="http://localhost:5001" \ 2>&1 | tee aico_primitive_skill_vision.log No newline at end of file vagen/examples/sokoban_lake_aico/run.sh +1 −1 Original line number Diff line number Diff line Loading @@ -5,7 +5,7 @@ export VLLM_ATTENTION_BACKEND=XFORMERS export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/sokoban-vision-debug/train.parquet" \ Loading Loading
vagen/examples/frozen_lake_aico_service/run.sh +1 −1 Original line number Diff line number Diff line Loading @@ -5,7 +5,7 @@ export VLLM_ATTENTION_BACKEND=XFORMERS export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/frozenlake-vision-debug/train.parquet" \ Loading
vagen/examples/navigation_aico/run.sh +2 −0 Original line number Diff line number Diff line Loading @@ -6,6 +6,8 @@ export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/navigation-vision-debug/train.parquet" \ Loading
vagen/examples/primitive_skill_aico/env_config.yaml +4 −4 Original line number Diff line number Diff line Loading @@ -4,25 +4,25 @@ env1: render_mode: text env_id: "AlignTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 test_size: 8 env2: env_name: primitive_skill env_config: render_mode: text env_id: "PlaceTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 test_size: 8 env3: env_name: primitive_skill env_config: render_mode: text env_id: "PutAppleInDrawer" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 test_size: 8 env4: env_name: primitive_skill env_config: render_mode: text env_id: "StackThreeCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 16 No newline at end of file test_size: 8 No newline at end of file
vagen/examples/primitive_skill_aico/run.sh +4 −1 Original line number Diff line number Diff line Loading @@ -6,6 +6,8 @@ export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server host=127.0.0.1 port=5001 in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/primitive_skill-vision-debug/train.parquet" \ Loading @@ -19,7 +21,7 @@ python3 -m vagen.trainer.main_ppo \ algorithm.high_level_gamma=0.95 \ data.train_files=data/primitive_skill-vision-debug/train.parquet \ data.val_files=data/primitive_skill-vision-debug/test.parquet \ data.train_batch_size=64 \ data.train_batch_size=32 \ data.val_batch_size=32 \ data.max_prompt_length=1024 \ data.max_response_length=128 \ Loading Loading @@ -76,4 +78,5 @@ python3 -m vagen.trainer.main_ppo \ rollout_manager.n_trajectory=2 \ rollout_manager.use_service=True \ rollout_manager.timeout=240 \ rollout_manager.base_url="http://localhost:5001" \ 2>&1 | tee aico_primitive_skill_vision.log No newline at end of file
vagen/examples/sokoban_lake_aico/run.sh +1 −1 Original line number Diff line number Diff line Loading @@ -5,7 +5,7 @@ export VLLM_ATTENTION_BACKEND=XFORMERS export PYTHONHASHSEED=0 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" # run python -m vagen.server.server in a tmux session first python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/sokoban-vision-debug/train.parquet" \ Loading
