File size: 1,117 Bytes
b4b2877 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | #!/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 ==="
|