crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.26k stars 372 forks source link

Cache FID computation #19

Open neverix opened 2 years ago

neverix commented 2 years ago

Computing Inception takes up to half a minute at the start of the program as of now, so caching it seems obvious.

The only thing I'm not sure about is what to use as key - probably hash for HF, path + modification date for image folder and a plain string for MNIST/CIFAR?