Can you provide pretrained models?

NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)

https://arxiv.org/abs/2007.03898

Other

999 stars 163 forks source link

Can you provide pretrained models? #34

Closed kopepod closed 2 years ago

kopepod commented 2 years ago

Hi, I have no that powerful resources to train the models (2GPUs), could you provide pretrained models for evaluations purposes regarding the latent space? Thanks.

arash-vahdat commented 2 years ago

We have released checkpoints on the MNIST, CIFAR-10, CelebA 64, CelebA HQ 256, and FFHQ datasets. Check here on the readme.