#!/bin/bash
# Action Prediction multi-seed: 5 seeds × top 3 modalities
# Best settings from pred_cls3: 8 coarse + prev_action + ds=5 + window=10s
# Total: 15 jobs

PYTHON=python
BASEDIR=${PULSE_ROOT}
TRAIN_SCRIPT=${BASEDIR}/experiments/tasks/train_pred_cls.py
OUTDIR=${BASEDIR}/results/pred_multiseed
LOGDIR=${OUTDIR}/slurm_logs
mkdir -p $LOGDIR

BASE="--mode prediction --coarse --use_prev_action --epochs 80 --batch_size 32 --lr 1e-3 --weight_decay 1e-4 --hidden_dim 128 --dropout 0.2 --downsample 5 --patience 20 --augment --noise_std 0.1 --time_mask_ratio 0.1 --label_smoothing 0.1 --window_sec 10.0 --output_dir $OUTDIR"

# Top 3 from pred_cls3: emg,imu (F1w=0.306), mocap,emg,eyetrack,imu (0.277), mocap,emg,imu (0.272)
TOP_MODS=("emg,imu" "mocap,emg,eyetrack,imu" "mocap,emg,imu")
SEEDS=(42 123 456 789 1024)

for mods in "${TOP_MODS[@]}"; do
    mod_tag=$(echo $mods | tr ',' '-')
    for seed in "${SEEDS[@]}"; do
        sbatch \
            -J "pred_ms_${mod_tag}_s${seed}" \
            -p gpuA800 \
            --gres=gpu:1 \
            -N 1 -n 1 \
            --cpus-per-task=4 \
            --mem=32G \
            -t 2:00:00 \
            -o "${LOGDIR}/${mod_tag}_s${seed}_%j.out" \
            -e "${LOGDIR}/${mod_tag}_s${seed}_%j.err" \
            --export=ALL \
            --wrap="export PYTHONUNBUFFERED=1; cd ${BASEDIR}; $PYTHON $TRAIN_SCRIPT --modalities $mods --seed $seed --tag s${seed} $BASE"
        echo "Submitted: $mods seed=$seed"
    done
done

echo ""
echo "Total: 15 jobs | Prediction Multi-seed"
echo "Results: $OUTDIR"