#!/usr/bin/env bash
set -euo pipefail

CKPT=${1:?usage: $0 /path/to/model.th}
DATA_ROOT="${DATA_ROOT:-./datasets/cifar10}"

python evaluate.py \
    --hps cifar10 \
    --data_root "$DATA_ROOT" \
    --restore_path "$CKPT" \
    --restore_ema_path "${CKPT%.th}-ema.th" \
    --use_ema True \
    --use_se True \
    --width 768 \
    --dec_blocks '1x1,4m1,4x2,8m4,8x5,16m8,16x5,32m16,32x5' \
    --latent_dim 128 \
    --use_rtm True \
    --H_cycles 4 --L_cycles 1 --refinement_steps 4 \
    --num_tokens 4 \
    --rtm_hidden_size 256 \
    --num_fid_samples 50000 \
    --num_pr_samples 50000 \
    --test_batch_size 256