mirror of
https://github.com/RYDE-WORK/lnp_ml.git
synced 2026-09-18 15:03:20 +08:00
21 lines
862 B
Bash
21 lines
862 B
Bash
#!/usr/bin/env bash
|
||
set -uo pipefail
|
||
SEED=${SEED:-42}
|
||
MODE=${MODE:-variant} # variant=两卡各跑一个 reg_bypass 变体；shard=两卡分 fold
|
||
|
||
if [ "${MODE}" = "shard" ]; then
|
||
RB=${REG_BYPASS:-off}
|
||
GPU=0 FOLDS=0,1,2 SEED=${SEED} REG_BYPASS=${RB} BATCH=8 \
|
||
setsid nohup bash scripts_run/run_final_cv.sh >/dev/null 2>&1 &
|
||
GPU=1 FOLDS=3,4 SEED=${SEED} REG_BYPASS=${RB} BATCH=4 \
|
||
setsid nohup bash scripts_run/run_final_cv.sh >/dev/null 2>&1 &
|
||
else
|
||
GPU=0 SEED=${SEED} REG_BYPASS=off BATCH=8 MIN_FREE_MB=14000 \
|
||
setsid nohup bash scripts_run/run_final_cv.sh >/dev/null 2>&1 &
|
||
GPU=1 SEED=${SEED} REG_BYPASS=on BATCH=4 MIN_FREE_MB=11000 \
|
||
setsid nohup bash scripts_run/run_final_cv.sh >/dev/null 2>&1 &
|
||
fi
|
||
|
||
sleep 2
|
||
echo "已拉起 MODE=${MODE} SEED=${SEED} PRETRAIN=${PRETRAIN:-<未设置>}"
|
||
echo "日志：models/final_cv/*/gpu*.log" |