DDPM-2param / cross_model /scripts /run_ddpm_figure6.sh
collins909's picture
Upload 2-parameter conditional DDPM (HI emulation, CAMELS LH params_2, epoch 200) with full training/eval/posterior toolchain
f513198 verified
raw
history blame contribute delete
916 Bytes
#!/bin/bash
#SBATCH --account=<your-slurm-account>
#SBATCH --partition=l40s
#SBATCH --nodes=1
#SBATCH --ntasks=8
#SBATCH --gres=gpu:l40s:1
#SBATCH --time=24:00:00
#SBATCH --job-name=ddpm_fig6
#SBATCH --mail-user=<your-email> # replace before submitting
#SBATCH --output=slurm-ddpm-figure6-%j.out
#SBATCH --error=slurm-ddpm-figure6-%j.err
# Figure 6 style (arXiv:2409.09101-inspired) surrogate posteriors for DDPM-2 / DDPM-6.
# sbatch <DDPM_ROOT>/Models/scripts/run_ddpm_figure6.sh
# sbatch --export=OUTPUT_DIR=/path/to/out,TEST_INDEX=42 .../run_ddpm_figure6.sh --no-six-grid
#
set -euo pipefail
ROOT="<DDPM_ROOT>/Models"
cd "$ROOT"
module load python/miniconda3-py3.12-usr
OUT="${OUTPUT_DIR:-${ROOT}/ddpm_figure6_out}"
IDX="${TEST_INDEX:-56}"
echo "Job=${SLURM_JOB_ID:-local} OUT=${OUT} TEST_INDEX=${IDX}"
python "${ROOT}/scripts/run_ddpm_figure6_suite.py" --output-dir "${OUT}" --test-index "${IDX}" "$@"