#!/bin/bash
# SpatialDiffuseSlot Phase-1: Semanticist DiT-L/ViT (tok_L init, trunk FROZEN)
# + our 85 multi-res attn-pool tokens + our spatial-align mask.
# Usage: GPUS=0,1,2,3 bash train_spatial_l.sh
# Phase-2 (나중): config freeze_dit:false + ckpt_path=<phase1 step> 로 resume.
set -e
cd "$(dirname "$0")"
VENV=/NHNHOME/WORKSPACE/0226010398_A/sr_diffusion/clevr_sudoku/MNIST_debug/venv
[ -d "$VENV" ] && source "$VENV/bin/activate"
export CUDA_VISIBLE_DEVICES="${GPUS:-0,1,2,3}"
export TORCH_HOME=/NHNHOME/WORKSPACE/0226010398_A/torch_cache
NG=$(echo "$CUDA_VISIBLE_DEVICES" | tr ',' '\n' | wc -l)
accelerate launch --num_processes "$NG" --multi_gpu \
  train_net.py --cfg configs/tokenizer_l_spatial.yaml