#!/bin/bash
# Exp1 v8 Multi-seed: Top configs × 5 seeds to measure variance
# Configs: (1) transformer+imu early, (2) transformer+3mod late, (3) transformer+3mod stacking
# Seeds: 42, 123, 456, 789, 2024

PYTHON=python
BASEDIR=${PULSE_ROOT}
SCRIPT=${BASEDIR}/experiments/train_exp1.py
OUTDIR=${BASEDIR}/results/exp1_v8_multiseed
LOGDIR=${OUTDIR}/slurm_logs
mkdir -p $LOGDIR

COMMON="--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)

# Config 1: Transformer + imu (single, early)
echo "=== Transformer + imu (5 seeds) ==="
for seed in "${SEEDS[@]}"; do
    sbatch -J "ms_tf_imu_s${seed}" -p gpuA800 --gres=gpu:1 -N1 -n1 \
        --cpus-per-task=4 --mem=32G -t 2:00:00 \
        -o "${LOGDIR}/tf_imu_early_s${seed}_%j.out" \
        -e "${LOGDIR}/tf_imu_early_s${seed}_%j.err" \
        --export=ALL \
        --wrap="export PYTHONUNBUFFERED=1; cd ${BASEDIR}; $PYTHON $SCRIPT --model transformer --modalities imu --fusion early --seed $seed --tag s${seed} $COMMON"
    echo "  seed=$seed"
done

# Config 2: Transformer + imu,mocap,emg late fusion
echo ""
echo "=== Transformer + 3mod late (5 seeds) ==="
for seed in "${SEEDS[@]}"; do
    sbatch -J "ms_tf_3m_late_s${seed}" -p gpuA800 --gres=gpu:1 -N1 -n1 \
        --cpus-per-task=4 --mem=64G -t 2:00:00 \
        -o "${LOGDIR}/tf_3mod_late_s${seed}_%j.out" \
        -e "${LOGDIR}/tf_3mod_late_s${seed}_%j.err" \
        --export=ALL \
        --wrap="export PYTHONUNBUFFERED=1; cd ${BASEDIR}; $PYTHON $SCRIPT --model transformer --modalities imu,mocap,emg --fusion late --seed $seed --tag s${seed} $COMMON"
    echo "  seed=$seed"
done

# Config 3: Transformer + imu,mocap,emg stacking fusion
echo ""
echo "=== Transformer + 3mod stacking (5 seeds) ==="
for seed in "${SEEDS[@]}"; do
    sbatch -J "ms_tf_3m_stack_s${seed}" -p gpuA800 --gres=gpu:1 -N1 -n1 \
        --cpus-per-task=4 --mem=64G -t 2:00:00 \
        -o "${LOGDIR}/tf_3mod_stacking_s${seed}_%j.out" \
        -e "${LOGDIR}/tf_3mod_stacking_s${seed}_%j.err" \
        --export=ALL \
        --wrap="export PYTHONUNBUFFERED=1; cd ${BASEDIR}; $PYTHON $SCRIPT --model transformer --modalities imu,mocap,emg --fusion stacking --seed $seed --tag s${seed} $COMMON"
    echo "  seed=$seed"
done

echo ""
echo "Total: 15 jobs | 3 configs × 5 seeds"
echo "Results: $OUTDIR"