File size: 683 Bytes
db75703
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/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}"