File size: 1,708 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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | #!/bin/bash
# Dispatch 16 eval jobs in parallel — one per (modalities_canonical, t_obs, t_fut) tuple.
set -euo pipefail
PYTHON=python
EVAL=${PULSE_ROOT}/scripts/eval_subset.py
PARTITION=${PARTITION:-gpuA800}
GPU_GRES=${GPU_GRES:-gpu:1}
LOG_DIR=${PULSE_ROOT}/results/eval_logs
mkdir -p "$LOG_DIR"
# 16 distinct subsets enumerated by inspecting all results.json files.
# Each line: <modalities_sorted>|<t_obs>|<t_fut>
SUBSETS=(
"emg,eyetrack,imu|8.0|2.0"
"emg,eyetrack,imu,mocap|8.0|2.0"
"emg,eyetrack,imu,mocap,pressure|8.0|1.0"
"emg,eyetrack,imu,mocap,pressure|8.0|2.0"
"emg,eyetrack,imu,mocap,pressure|8.0|5.0"
"emg,eyetrack,imu,mocap,pressure|8.0|10.0"
"emg,eyetrack,imu,mocap,pressure|8.0|15.0"
"emg,eyetrack,imu,pressure|8.0|2.0"
"emg,eyetrack,mocap,pressure|8.0|2.0"
"emg,imu|8.0|2.0"
"emg,imu,mocap|8.0|2.0"
"emg,imu,mocap,pressure|8.0|2.0"
"eyetrack,imu,mocap,pressure|8.0|2.0"
"imu|8.0|2.0"
"imu,mocap|8.0|2.0"
"mocap|8.0|2.0"
)
idx=0
for entry in "${SUBSETS[@]}"; do
IFS='|' read -r mods t_obs t_fut <<< "$entry"
idx=$((idx+1))
tag=$(echo "${mods}_o${t_obs}_f${t_fut}" | tr ',.' '_')
job_name="evalT10_${idx}_${tag}"
job_name=$(echo "$job_name" | cut -c1-60) # SLURM job names cap at ~60 chars
out="${LOG_DIR}/${tag}.out"
err="${LOG_DIR}/${tag}.err"
cmd="export PYTHONUNBUFFERED=1; ${PYTHON} ${EVAL} --modalities ${mods} --t_obs ${t_obs} --t_fut ${t_fut}"
sbatch -J "${job_name}" -p "${PARTITION}" --gres="${GPU_GRES}" \
-N 1 -n 1 --cpus-per-task=4 --mem=32G \
-t 0:20:00 -o "${out}" -e "${err}" \
--export=ALL --wrap="${cmd}"
echo "submitted ${job_name}"
done
echo ""
echo "All 16 dispatched. Logs: ${LOG_DIR}/"
|