Loading vagen/examples/navigation_aico/env_config.yaml +1 −1 Original line number Diff line number Diff line Loading @@ -3,6 +3,6 @@ env1: env_config: render_mode: "vision" train_size: 10000 test_size: 16 test_size: 128 vagen/examples/navigation_aico/run.sh +4 −2 Original line number Diff line number Diff line Loading @@ -19,7 +19,8 @@ python3 -m vagen.trainer.main_ppo \ algorithm.high_level_gamma=0.95 \ data.train_files=data/navigation-vision-debug/train.parquet \ data.val_files=data/navigation-vision-debug/test.parquet \ data.train_batch_size=32 \ data.train_batch_size=64 \ data.val_batch_size=16 \ data.max_prompt_length=1024 \ data.max_response_length=128 \ data.max_trajectory_length=2400 \ Loading Loading @@ -72,6 +73,7 @@ python3 -m vagen.trainer.main_ppo \ rollout_manager.use_gae_mask=True \ trainer.val_before_train=True \ trainer.val_generations_to_log_to_wandb=4 \ rollout_manager.n_trajectory=1 \ rollout_manager.n_trajectory=2 \ rollout_manager.use_service=True \ rollout_manager.timeout=240 \ 2>&1 | tee aico_navigation_vision.log No newline at end of file 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: 8 test_size: 16 env2: env_name: primitive_skill env_config: render_mode: text env_id: "PlaceTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 8 test_size: 16 env3: env_name: primitive_skill env_config: render_mode: text env_id: "PutAppleInDrawer" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 8 test_size: 16 env4: env_name: primitive_skill env_config: render_mode: text env_id: "StackThreeCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 8 No newline at end of file test_size: 16 No newline at end of file vagen/examples/primitive_skill_aico/run.sh +3 −1 Original line number Diff line number Diff line Loading @@ -20,6 +20,7 @@ python3 -m vagen.trainer.main_ppo \ 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.val_batch_size=32 \ data.max_prompt_length=1024 \ data.max_response_length=128 \ data.max_trajectory_length=2400 \ Loading Loading @@ -72,6 +73,7 @@ python3 -m vagen.trainer.main_ppo \ rollout_manager.use_gae_mask=True \ trainer.val_before_train=True \ trainer.val_generations_to_log_to_wandb=8 \ rollout_manager.n_trajectory=1 \ rollout_manager.n_trajectory=2 \ rollout_manager.use_service=True \ rollout_manager.timeout=240 \ 2>&1 | tee aico_primitive_skill_vision.log No newline at end of file vagen/server/server.yaml +5 −5 Original line number Diff line number Diff line Loading @@ -7,13 +7,13 @@ server: debug: false frozenlake: max_workers: 20 max_workers: 48 svg: max_workers: 20 max_workers: 48 model_size: "small" navigation: max_workers: 20 devices: [4,5,6,7] max_workers: 48 devices: [0,1,2,3] primitive_skill: max_workers: 20 max_workers: 48 Loading
vagen/examples/navigation_aico/env_config.yaml +1 −1 Original line number Diff line number Diff line Loading @@ -3,6 +3,6 @@ env1: env_config: render_mode: "vision" train_size: 10000 test_size: 16 test_size: 128
vagen/examples/navigation_aico/run.sh +4 −2 Original line number Diff line number Diff line Loading @@ -19,7 +19,8 @@ python3 -m vagen.trainer.main_ppo \ algorithm.high_level_gamma=0.95 \ data.train_files=data/navigation-vision-debug/train.parquet \ data.val_files=data/navigation-vision-debug/test.parquet \ data.train_batch_size=32 \ data.train_batch_size=64 \ data.val_batch_size=16 \ data.max_prompt_length=1024 \ data.max_response_length=128 \ data.max_trajectory_length=2400 \ Loading Loading @@ -72,6 +73,7 @@ python3 -m vagen.trainer.main_ppo \ rollout_manager.use_gae_mask=True \ trainer.val_before_train=True \ trainer.val_generations_to_log_to_wandb=4 \ rollout_manager.n_trajectory=1 \ rollout_manager.n_trajectory=2 \ rollout_manager.use_service=True \ rollout_manager.timeout=240 \ 2>&1 | tee aico_navigation_vision.log No newline at end of file
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: 8 test_size: 16 env2: env_name: primitive_skill env_config: render_mode: text env_id: "PlaceTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 8 test_size: 16 env3: env_name: primitive_skill env_config: render_mode: text env_id: "PutAppleInDrawer" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 8 test_size: 16 env4: env_name: primitive_skill env_config: render_mode: text env_id: "StackThreeCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube train_size: 10000 test_size: 8 No newline at end of file test_size: 16 No newline at end of file
vagen/examples/primitive_skill_aico/run.sh +3 −1 Original line number Diff line number Diff line Loading @@ -20,6 +20,7 @@ python3 -m vagen.trainer.main_ppo \ 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.val_batch_size=32 \ data.max_prompt_length=1024 \ data.max_response_length=128 \ data.max_trajectory_length=2400 \ Loading Loading @@ -72,6 +73,7 @@ python3 -m vagen.trainer.main_ppo \ rollout_manager.use_gae_mask=True \ trainer.val_before_train=True \ trainer.val_generations_to_log_to_wandb=8 \ rollout_manager.n_trajectory=1 \ rollout_manager.n_trajectory=2 \ rollout_manager.use_service=True \ rollout_manager.timeout=240 \ 2>&1 | tee aico_primitive_skill_vision.log No newline at end of file
vagen/server/server.yaml +5 −5 Original line number Diff line number Diff line Loading @@ -7,13 +7,13 @@ server: debug: false frozenlake: max_workers: 20 max_workers: 48 svg: max_workers: 20 max_workers: 48 model_size: "small" navigation: max_workers: 20 devices: [4,5,6,7] max_workers: 48 devices: [0,1,2,3] primitive_skill: max_workers: 20 max_workers: 48
