#!/bin/bash
#SBATCH --job-name=ablation_fix
#SBATCH --partition=gpuA800
#SBATCH --gres=gpu:1
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=4
#SBATCH --mem=32G
#SBATCH --time=1:00:00
#SBATCH --output=${PULSE_ROOT}/results/ablation_fix_%j.log

# Fix: mocap+emg late+pretrained — pretrain MOCAP branch (idx=0) instead of emg
set -e
export PYTHONUNBUFFERED=1

PYTHON=python
BASEDIR=${PULSE_ROOT}
SCRIPT=${BASEDIR}/experiments/train_exp1.py
OUTDIR=${BASEDIR}/results/modality_ablation
COMMON="--model transformer --epochs 100 --batch_size 16 --lr 1e-3 --weight_decay 1e-4 --hidden_dim 128 --downsample 5 --patience 15 --proj_dim 0 --output_dir $OUTDIR"
SEEDS=(42 123 456 789 2024)

PT_MOCAP=${BASEDIR}/results/exp1_v8/transformer_mocap_early/model_best.pt

echo "=== Fix: mocap+emg / late+pretrained(mocap, idx=0) ==="
for seed in "${SEEDS[@]}"; do
    echo "  mocap+emg seed=$seed"
    $PYTHON $SCRIPT --modalities mocap,emg --fusion late --seed $seed \
        --pretrained_backbone $PT_MOCAP --freeze_backbone_idx 0 \
        --tag ablation_pt_s${seed} $COMMON 2>&1 | tail -5
done

echo "=== Done ==="