File size: 951 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 --partition=gpuA800
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=4
#SBATCH --gres=gpu:1
#SBATCH --mem=32G
#SBATCH --time=8:00:00
#SBATCH --job-name=ASF_seg_ms
#SBATCH --output=${PULSE_ROOT}/results/pub_multiseed_exp2_%j.log
# ASFormer multi-seed temporal segmentation
set -e
PYTHON=python
PROJECT=${PULSE_ROOT}
cd $PROJECT
OUT=$PROJECT/results/published_baselines/exp2_asformer_multiseed
mkdir -p $OUT
echo "=== ASFormer Multi-Seed Temporal Segmentation ==="
for SEED in 42 123 456 789 2024; do
for MOD in mocap "mocap,emg,eyetrack" "mocap,emg,eyetrack,imu"; do
echo "--- seed=$SEED / ${MOD} ---"
$PYTHON experiments/train_exp2.py \
--model asformer --modalities $MOD \
--hidden_dim 64 --epochs 80 --batch_size 16 \
--lr 5e-4 --weight_decay 1e-4 --downsample 2 \
--seed $SEED --output_dir $OUT 2>&1 | tail -3
done
done
echo "=== Done ==="
|