From ba44850e6f6002cbcf195f2657d6d037cec6b424 Mon Sep 17 00:00:00 2001 From: Vivicai1005 Date: Fri, 26 Dec 2025 14:34:27 +0800 Subject: [PATCH] update default paths and disable offloading for AMD qwen3-4B training --- scripts/run-qwen3-4B-amd.sh | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/scripts/run-qwen3-4B-amd.sh b/scripts/run-qwen3-4B-amd.sh index 61d301542..fdeb722cd 100755 --- a/scripts/run-qwen3-4B-amd.sh +++ b/scripts/run-qwen3-4B-amd.sh @@ -15,13 +15,13 @@ set -euxo pipefail ### AMD Support ### -SLIME_DIR="${SLIME_DIR:-/home/yushensu/projects/slime}" # Default path if not set in environment +SLIME_DIR="${SLIME_DIR:-/root}" # Default path if not set in environment export SLIME_DIR -MODEL_DIR="${MODEL_DIR:-/home/yushensu/projects/model}" # Default path if not set in environment +MODEL_DIR="${MODEL_DIR:-/root}" # Default path if not set in environment export MODEL_DIR -DATA_DIR="${DATA_DIR:-/home/yushensu/projects/data}" # Default path if not set in environment +DATA_DIR="${DATA_DIR:-/root}" # Default path if not set in environment export DATA_DIR # For AMD GPU @@ -153,6 +153,8 @@ ray job submit --address="http://127.0.0.1:8265" \ --actor-num-nodes 1 \ --actor-num-gpus-per-node 8 \ --colocate \ + --no-offload-train \ + --no-offload-rollout \ ${MODEL_ARGS[@]} \ ${CKPT_ARGS[@]} \ ${ROLLOUT_ARGS[@]} \