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

CKPT=${1:?usage: $0 /path/to/model.th /path/to/dataset_dir}
DATA=${2:?usage: $0 /path/to/model.th /path/to/dataset_dir}

H_LAYERS="${H_LAYERS:-2}"
L_LAYERS="${L_LAYERS:-2}"

python evaluate.py \
    --hps fewshot \
    --data_root "$DATA" \
    --restore_path "$CKPT" \
    --restore_ema_path "${CKPT%.th}-ema.th" \
    --use_ema True \
    --latent_dim 1024 \
    --use_rtm True \
    --H_cycles 4 --L_cycles 2 --refinement_steps 2 \
    --H_layers "$H_LAYERS" --L_layers "$L_LAYERS" \
    --num_tokens 4 \
    --rtm_hidden_size 256 \
    --num_fid_samples 5000 \
    --num_pr_samples 1000