Skip to content

Commit b0b5178

Browse files
author
root
committed
CUDA 调整
1 parent 5d0626f commit b0b5178

5 files changed

+1
-1
lines changed

examples/werewolf/train.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ python -m agentlightning.verl \
3939
actor_rollout_ref.actor.fsdp_config.param_offload=False \
4040
actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \
4141
actor_rollout_ref.rollout.name=vllm \
42-
actor_rollout_ref.rollout.gpu_memory_utilization=0.6 \
42+
actor_rollout_ref.rollout.gpu_memory_utilization=0.4 \
4343
actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=1 \
4444
actor_rollout_ref.ref.fsdp_config.param_offload=False \
4545
algorithm.use_kl_in_reward=False \

0 commit comments

Comments
 (0)