GalMeasDiff / run.sh
lvjiameng's picture
Upload 4 files
db75703 verified
Raw
History Blame Contribute Delete
683 Bytes
#!/usr/bin/env bash
# Quick demo: conditions β†’ generate β†’ SR β†’ measure β†’ metrics
set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "${ROOT}"
if [[ ! -f weights/00003000-joint.pt ]]; then
echo "[run] weights missing β†’ assembling bundle"
bash scripts/assemble_bundle.sh
fi
CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}" \
python3 -u generate_and_eval.py \
--cond-values "${COND_VALUES:-0.0784,144.40,310.56,551.57,-0.3088,0.2504,5.1633,2.2935}" \
--n-samples "${N_SAMPLES:-2}" \
--outdir "${OUTDIR:-runs/example}" \
--device cuda \
--rank 0 \
--sampling-timesteps "${SAMPLING_TIMESTEPS:-1000}" \
--mc-samples "${MC_SAMPLES:-5}"