Commit 9acf6a30 authored by jameskrw's avatar jameskrw
Browse files

updates tranining scripts for repoduction

parent 41c6db80

vagen/examples/release_experiments/mask_gae_mask_loss.sh

+1 −1
Original line number Diff line number Diff line
@@ -69,7 +69,7 @@ python3 -m vagen.trainer.main_ppo \ 
    trainer.experiment_name='mask_gae_mask_loss' \

    trainer.n_gpus_per_node=4 \

    trainer.nnodes=1 \

    trainer.save_freq=100 \

    trainer.save_freq=70 \

    trainer.test_freq=20 \

    trainer.total_epochs=15 \

    rollout_manager.max_turns=3 \

vagen/examples/release_experiments/mask_gae_mask_loss_turnwise_reward_bi_level.sh

+1 −1
Original line number Diff line number Diff line
@@ -29,7 +29,7 @@ python3 -m vagen.trainer.main_ppo \ 
    data.train_batch_size=128 \

    data.max_prompt_length=1024 \

    data.max_response_length=128 \

    data.max_trajectory_length=2000 \

    data.max_trajectory_length=1800 \

    data.image_key=images \

    data.truncation=error \

    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \

CRA Git | Maintained and supported by SUSTech CRA and CCSE