minor (3cf7a5da) · Commits · 张泽凯 / VAGEN

vagen/examples/release_experiments/mask_gae_mask_loss_bi_level.sh

+3 −3

Original line number	Diff line number	Diff line
		@@ -43,9 +43,9 @@ python3 -m vagen.trainer.main_ppo \
		actor_rollout_ref.actor.fsdp_config.param_offload=False \
		actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \
		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.tensor_model_parallel_size=1 \
		actor_rollout_ref.rollout.name=vllm \
		actor_rollout_ref.rollout.gpu_memory_utilization=0.2 \
		actor_rollout_ref.rollout.gpu_memory_utilization=0.4 \
		actor_rollout_ref.rollout.enable_chunked_prefill=False \
		actor_rollout_ref.rollout.enforce_eager=False \
		actor_rollout_ref.rollout.free_cache_engine=False \
		@@ -66,7 +66,7 @@ python3 -m vagen.trainer.main_ppo \
		trainer.logger=['console','wandb'] \
		trainer.project_name='vagen' \
		trainer.experiment_name='mask_gae_mask_loss_bi_level' \
		trainer.n_gpus_per_node=4 \
		trainer.n_gpus_per_node=8 \
		trainer.nnodes=1 \
		trainer.save_freq=100 \
		trainer.test_freq=20 \