Loading vagen/examples/svg_aico/env_config.yaml +4 −4 Original line number Diff line number Diff line Loading @@ -2,8 +2,8 @@ env1: env_name: svg env_config: split: train dino_weight: 1.0 structural_weight: 1.0 dino_weight: 5.0 structural_weight: 5.0 train_size: 1000 test_size: 0 Loading @@ -11,8 +11,8 @@ env2: env_name: svg env_config: split: test dino_weight: 1.0 structural_weight: 1.0 dino_weight: 5.0 structural_weight: 5.0 train_size: 0 test_size: 512 Loading vagen/examples/svg_aico/run.sh +2 −2 Original line number Diff line number Diff line Loading @@ -65,8 +65,8 @@ python3 -m vagen.trainer.main_ppo \ trainer.save_freq=70 \ trainer.test_freq=20 \ trainer.total_training_steps=200 \ rollout_manager.max_turns=3 \ rollout_manager.window_size=5 \ rollout_manager.max_turns=2 \ rollout_manager.window_size=3 \ rollout_manager.use_multi_turn_reward=False \ rollout_manager.use_loss_mask=True \ rollout_manager.use_gae_mask=True \ Loading vagen/examples/svg_trico/run.sh +3 −2 Original line number Diff line number Diff line Loading @@ -10,6 +10,7 @@ python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/svg-vision-debug/train.parquet" \ --test_path "data/svg-vision-debug/test.parquet" \ --force_gen # max_trajectory_length = max_prompt_length + max_response_length Loading Loading @@ -64,8 +65,8 @@ python3 -m vagen.trainer.main_ppo \ trainer.save_freq=70 \ trainer.test_freq=20 \ trainer.total_training_steps=200 \ rollout_manager.max_turns=3 \ rollout_manager.window_size=5 \ rollout_manager.max_turns=2 \ rollout_manager.window_size=3 \ rollout_manager.use_multi_turn_reward=False \ rollout_manager.use_loss_mask=True \ rollout_manager.use_gae_mask=True \ Loading Loading
vagen/examples/svg_aico/env_config.yaml +4 −4 Original line number Diff line number Diff line Loading @@ -2,8 +2,8 @@ env1: env_name: svg env_config: split: train dino_weight: 1.0 structural_weight: 1.0 dino_weight: 5.0 structural_weight: 5.0 train_size: 1000 test_size: 0 Loading @@ -11,8 +11,8 @@ env2: env_name: svg env_config: split: test dino_weight: 1.0 structural_weight: 1.0 dino_weight: 5.0 structural_weight: 5.0 train_size: 0 test_size: 512 Loading
vagen/examples/svg_aico/run.sh +2 −2 Original line number Diff line number Diff line Loading @@ -65,8 +65,8 @@ python3 -m vagen.trainer.main_ppo \ trainer.save_freq=70 \ trainer.test_freq=20 \ trainer.total_training_steps=200 \ rollout_manager.max_turns=3 \ rollout_manager.window_size=5 \ rollout_manager.max_turns=2 \ rollout_manager.window_size=3 \ rollout_manager.use_multi_turn_reward=False \ rollout_manager.use_loss_mask=True \ rollout_manager.use_gae_mask=True \ Loading
vagen/examples/svg_trico/run.sh +3 −2 Original line number Diff line number Diff line Loading @@ -10,6 +10,7 @@ python -m vagen.env.create_dataset \ --yaml_path "$SCRIPT_DIR/env_config.yaml" \ --train_path "data/svg-vision-debug/train.parquet" \ --test_path "data/svg-vision-debug/test.parquet" \ --force_gen # max_trajectory_length = max_prompt_length + max_response_length Loading Loading @@ -64,8 +65,8 @@ python3 -m vagen.trainer.main_ppo \ trainer.save_freq=70 \ trainer.test_freq=20 \ trainer.total_training_steps=200 \ rollout_manager.max_turns=3 \ rollout_manager.window_size=5 \ rollout_manager.max_turns=2 \ rollout_manager.window_size=3 \ rollout_manager.use_multi_turn_reward=False \ rollout_manager.use_loss_mask=True \ rollout_manager.use_gae_mask=True \ Loading
