| 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 | |