| set -euo pipefail | |
| cd "$(dirname "$0")/.." | |
| DATA_DIR="${DATA_DIR:-/path/to/cifar-10}" | |
| SAVE_DIR="${SAVE_DIR:-./runs/cifar10_rtm}" | |
| CKPT_DIR="${CKPT_DIR:?CKPT_DIR (path to a checkpoints/<exp>/ folder) must be set}" | |
| NUM_GPUS="${NUM_GPUS:-1}" | |
| CFG_FILE="${CFG_FILE:-src/configs/CIFAR10/StyleGAN2-ADA-RTM.yaml}" | |
| CMD=(python src/main.py -hdf5 -best | |
| --cfg_file "$CFG_FILE" | |
| --data_dir "$DATA_DIR" | |
| --save_dir "$SAVE_DIR" | |
| -ckpt "$CKPT_DIR" | |
| -metrics fid is prdc | |
| --eval_backbone "InceptionV3_torch" | |
| --post_resizer "friendly" | |
| --num_eval 1 | |
| --num_workers 4 | |
| --seed 0) | |
| if [ "$NUM_GPUS" -gt 1 ]; then | |
| "${CMD[@]}" -DDP | |
| else | |
| "${CMD[@]}" | |
| fi | |