diff --git a/tests/special_npu/nightly_ci_ascend/run_gspo_qwen3_8b_fsdp2_npu.sh b/tests/special_npu/nightly_ci_ascend/run_gspo_qwen3_8b_fsdp2_npu.sh index 724f84005f7..fba413b4326 100644 --- a/tests/special_npu/nightly_ci_ascend/run_gspo_qwen3_8b_fsdp2_npu.sh +++ b/tests/special_npu/nightly_ci_ascend/run_gspo_qwen3_8b_fsdp2_npu.sh @@ -22,7 +22,7 @@ mkdir -p "$LOG_DIR" rm -rf "$LOG_DIR/$SCRIPT_NAME.log" # ---- user-adjustable (CI defaults keep runtime short) ---- -NGPUS_PER_NODE=${NGPUS_PER_NODE:-16} +NGPUS_PER_NODE=${NGPUS_PER_NODE:-8} TRAIN_BATCH_SIZE=${TRAIN_BATCH_SIZE:-32} PPO_MINI_BATCH_SIZE=${PPO_MINI_BATCH_SIZE:-16} MAX_PROMPT_LENGTH=${MAX_PROMPT_LENGTH:-2048} @@ -158,4 +158,4 @@ PYTHONUNBUFFERED=1 python3 -m verl.trainer.main_ppo \ "${REF[@]}" \ "${ROLLOUT[@]}" \ "${TRAINER[@]}" \ - "$@" 2>&1 | tee "$LOG_DIR/$SCRIPT_NAME.log" \ No newline at end of file + "$@" 2>&1 | tee "$LOG_DIR/$SCRIPT_NAME.log"