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

NomalDecoder & num_bits #31

Open PeterMcGor opened 2 years ago

PeterMcGor commented 2 years ago

Hi and thanks for sharing the code.

I am doing some tests with medical images, 16-bit and one channel input, therefore employing the NormalDecoder. The implementation takes as optional parameter num_bits but is not used as for the DiscMixLogistic. Is the implementation complete? or is just included for some kind of legacy?

Thanks in advance!