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