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

# ASFormer multi-seed contact detection
set -e
PYTHON=python
PROJECT=${PULSE_ROOT}
cd $PROJECT
OUT=$PROJECT/results/published_baselines/exp3_asformer_multiseed
mkdir -p $OUT

echo "=== ASFormer Multi-Seed Contact Detection ==="

for SEED in 42 123 456 789 2024; do
    for MOD in emg imu mocap "mocap,emg"; do
        echo "--- seed=$SEED / ${MOD} ---"
        $PYTHON experiments/train_exp3.py \
            --model asformer --modalities $MOD \
            --hidden_dim 64 --epochs 50 --batch_size 32 \
            --lr 1e-3 --weight_decay 1e-4 --downsample 2 \
            --seed $SEED --output_dir $OUT 2>&1 | tail -3
    done
done

echo "=== Done ==="