File size: 719 Bytes
4811c23 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | #!/usr/bin/env bash
# S0 冒烟(每个训练配置 1 迭代)
set -euo pipefail
cd "$(dirname "$0")/.."
source "$(conda info --base)/etc/profile.d/conda.sh"; conda activate AdcSR
CUDA_VISIBLE_DEVICES=0 python src/train_lora.py --config configs/config_smoke.yml \
--manifest data/manifest_train.json --real_prob 0.0 --steps 2 --batch_size 1 --grad_accum 1 \
--init_net weight/net_params_200.pkl --out weight/smoke --log_dir logs/smoke_lora || true
CUDA_VISIBLE_DEVICES=0 python src/train_stage2.py --config configs/config_smoke.yml \
--steps 2 --batch_size 1 --grad_accum 1 --teacher osediff --skip_ram \
--init_net weight/net_params_200.pkl --out weight/smoke --log_dir logs/smoke_s2 || true
echo smoke done
|