File size: 623 Bytes
3ce19a2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | #!/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
|