| #!/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 | |