perdim trainmax CTT 2026-07-03 workspace/scripts/slurm/eval_ctt_generated_rollout.sbatch

workspace/scripts/slurm/eval_ctt_generated_rollout.sbatch

@@ -13,10 +13,24 @@
 set -euo pipefail
 
 PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
-CHECKPOINT="${CHECKPOINT:?Set CHECKPOINT to a CTT model.pt}"
+TASK_ID="${SLURM_ARRAY_TASK_ID:-0}"
+if [[ -z "${CHECKPOINT:-}" ]]; then
+  if [[ -n "${CHECKPOINT_TEMPLATE:-}" ]]; then
+    CHECKPOINT="${CHECKPOINT_TEMPLATE//\{seed\}/$TASK_ID}"
+  else
+    echo "Set CHECKPOINT to a CTT model.pt or CHECKPOINT_TEMPLATE with {seed}" >&2
+    exit 2
+  fi
+fi
 SOURCE_INDEX="${SOURCE_INDEX:-data/cil_charts/train/index.json}"
 TARGET_INDEX="${TARGET_INDEX:-data/cil_charts/val/index.json}"
-OUT_DIR="${OUT_DIR:-runs/ctt_residual_rollout_smoke_seed${SLURM_ARRAY_TASK_ID:-0}}"
+if [[ -z "${OUT_DIR:-}" ]]; then
+  if [[ -n "${OUT_DIR_TEMPLATE:-}" ]]; then
+    OUT_DIR="${OUT_DIR_TEMPLATE//\{seed\}/$TASK_ID}"
+  else
+    OUT_DIR="runs/ctt_residual_rollout_smoke_seed${TASK_ID}"
+  fi
+fi
 
 SCRATCH_ROOT="/scratch/$USER/dovla"
 SIF="${SIF:-$SCRATCH_ROOT/containers/pytorch_2.7.1_cuda12.8.sif}"