search

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

CelebA-HQ 256 Dataset #13

Closed leoliu37 closed 3 years ago

leoliu37 commented 3 years ago

Thanks for the interesting work.

I want to reproduce the NVAE model in CelebAHQ 256 dataset. Could you provide the main NVAE training scripts with detailed parameters in the Celeba-HQ dataset?

Thanks for your help.

arash-vahdat commented 3 years ago

We just updated the README with the training commands for all the datasets, in addition to the pre-trained checkpoints.

leoliu37 commented 3 years ago

We just updated the README with the training commands for all the datasets, in addition to the pre-trained checkpoints.

Thanks for your help.