hailin
/
jd_train
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-08-28 16:30:15 +08:00
parent 5a30fee9a6
commit 8cbc4b948a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
run_ds.sh
View File

@ -14,13 +14,13 @@ SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
DS_CONFIG="${DS_CONFIG:-$SCRIPT_DIR/ds_config_zero3.json}" DS_CONFIG="${DS_CONFIG:-$SCRIPT_DIR/ds_config_zero3.json}"
# ==== 超参数（本地路径；可用 VAR=xxx ./run_ds.sh 覆写）==== # ==== 超参数（本地路径；可用 VAR=xxx ./run_ds.sh 覆写）====
MODEL_NAME_OR_PATH="${MODEL_NAME_OR_PATH:-/home/test/Qwen3-8B}" MODEL_NAME_OR_PATH="${MODEL_NAME_OR_PATH:-/home/test/Qwen3-1.7B}"
# 明确区分训练/评测文件（可按需改成通配符） # 明确区分训练/评测文件（可按需改成通配符）
DATA_GLOB="${DATA_GLOB:-$HOME/datasets/my_corpus/train.jsonl}" DATA_GLOB="${DATA_GLOB:-$HOME/datasets/my_corpus/train.jsonl}"
EVAL_DATA_GLOB="${EVAL_DATA_GLOB:-$HOME/datasets/my_corpus/test.jsonl}" EVAL_DATA_GLOB="${EVAL_DATA_GLOB:-$HOME/datasets/my_corpus/test.jsonl}"
OUTDIR="${OUTDIR:-$HOME/checkpoints/run-qwen3-8b}" OUTDIR="${OUTDIR:-$HOME/checkpoints/run-qwen3-1.7b}"
SEQ_LEN="${SEQ_LEN:-512}" SEQ_LEN="${SEQ_LEN:-512}"
LR="${LR:-2e-5}" LR="${LR:-2e-5}"
GAS="${GAS:-1}" GAS="${GAS:-1}"