Loading scripts/benchmark/env_benchmark/svg/run.sh +1 −1 Original line number Diff line number Diff line Loading @@ -21,6 +21,6 @@ export PYTHONPATH=$(pwd):$PYTHONPATH # Run benchmark echo "Running environment benchmark" python -m vagen.utils_benchmark.env_benchmark --config "$SCRIPT_DIR/benchmark_config.yaml" python -m vagen.env.verify_env --config "$SCRIPT_DIR/benchmark_config.yaml" echo "Benchmark complete. Results saved to env_benchmark_results directory." No newline at end of file scripts/exps/format/sokoban/grounding/run.sh +2 −2 Original line number Diff line number Diff line Loading @@ -39,8 +39,8 @@ python3 -m vagen.trainer.main_ppo \ actor_rollout_ref.actor.kl_loss_coef=0.001 \ actor_rollout_ref.actor.kl_loss_type=mse \ actor_rollout_ref.model.enable_gradient_checkpointing=True \ actor_rollout_ref.actor.fsdp_config.param_offload=False \ actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \ actor_rollout_ref.actor.fsdp_config.param_offload=True \ actor_rollout_ref.actor.fsdp_config.optimizer_offload=True \ actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=1 \ actor_rollout_ref.rollout.tensor_model_parallel_size=2 \ actor_rollout_ref.rollout.name=vllm \ Loading vagen/env/__init__.py +7 −7 Original line number Diff line number Diff line from .sokoban import SokobanEnv, SokobanEnvConfig, SokobanService, SokobanServiceConfig from .frozenlake import FrozenLakeEnv,FrozenLakeEnvConfig, FrozenLakeService # from .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService from .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService # from .svg import SVGEnv, SvgEnvConfig, SVGService, SVGServiceConfig # from .primitive_skill import PrimitiveSkillEnv, PrimitiveSkillEnvConfig, PrimitiveSkillService, PrimitiveSkillServiceConfig # from .alfworld import ALFWorldEnv, ALFWorldEnvConfig, ALFWorldService, ALFWorldServiceConfig Loading @@ -17,12 +17,12 @@ REGISTERED_ENV = { "config_cls": FrozenLakeEnvConfig, "service_cls": FrozenLakeService }, # "navigation": { # "env_cls": NavigationEnv, # "config_cls": NavigationEnvConfig, # "service_cls": NavigationService, # "service_config_cls": NavigationServiceConfig # }, "navigation": { "env_cls": NavigationEnv, "config_cls": NavigationEnvConfig, "service_cls": NavigationService, "service_config_cls": NavigationServiceConfig }, # "svg": { # "env_cls": SVGEnv, # "config_cls": SvgEnvConfig, Loading vagen/utils_benchmark/env_benchmark.py→vagen/env/verify_env.py +0 −0 File moved. View file vagen/utils_benchmark/service_benchmark.py→vagen/env/verify_service.py +0 −0 File moved. View file Loading
scripts/benchmark/env_benchmark/svg/run.sh +1 −1 Original line number Diff line number Diff line Loading @@ -21,6 +21,6 @@ export PYTHONPATH=$(pwd):$PYTHONPATH # Run benchmark echo "Running environment benchmark" python -m vagen.utils_benchmark.env_benchmark --config "$SCRIPT_DIR/benchmark_config.yaml" python -m vagen.env.verify_env --config "$SCRIPT_DIR/benchmark_config.yaml" echo "Benchmark complete. Results saved to env_benchmark_results directory." No newline at end of file
scripts/exps/format/sokoban/grounding/run.sh +2 −2 Original line number Diff line number Diff line Loading @@ -39,8 +39,8 @@ python3 -m vagen.trainer.main_ppo \ actor_rollout_ref.actor.kl_loss_coef=0.001 \ actor_rollout_ref.actor.kl_loss_type=mse \ actor_rollout_ref.model.enable_gradient_checkpointing=True \ actor_rollout_ref.actor.fsdp_config.param_offload=False \ actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \ actor_rollout_ref.actor.fsdp_config.param_offload=True \ actor_rollout_ref.actor.fsdp_config.optimizer_offload=True \ actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=1 \ actor_rollout_ref.rollout.tensor_model_parallel_size=2 \ actor_rollout_ref.rollout.name=vllm \ Loading
vagen/env/__init__.py +7 −7 Original line number Diff line number Diff line from .sokoban import SokobanEnv, SokobanEnvConfig, SokobanService, SokobanServiceConfig from .frozenlake import FrozenLakeEnv,FrozenLakeEnvConfig, FrozenLakeService # from .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService from .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService # from .svg import SVGEnv, SvgEnvConfig, SVGService, SVGServiceConfig # from .primitive_skill import PrimitiveSkillEnv, PrimitiveSkillEnvConfig, PrimitiveSkillService, PrimitiveSkillServiceConfig # from .alfworld import ALFWorldEnv, ALFWorldEnvConfig, ALFWorldService, ALFWorldServiceConfig Loading @@ -17,12 +17,12 @@ REGISTERED_ENV = { "config_cls": FrozenLakeEnvConfig, "service_cls": FrozenLakeService }, # "navigation": { # "env_cls": NavigationEnv, # "config_cls": NavigationEnvConfig, # "service_cls": NavigationService, # "service_config_cls": NavigationServiceConfig # }, "navigation": { "env_cls": NavigationEnv, "config_cls": NavigationEnvConfig, "service_cls": NavigationService, "service_config_cls": NavigationServiceConfig }, # "svg": { # "env_cls": SVGEnv, # "config_cls": SvgEnvConfig, Loading
