search

GlassyWing / nvae

An unofficial toy implementation for NVAE 《A Deep Hierarchical Variational Autoencoder》
Apache License 2.0
108 stars 21 forks source link

Possible to use Discretized mix logistic distribution? #7

Open jfrancis71 opened 3 years ago

jfrancis71 commented 3 years ago

Hi,

Very nice implementation, thank you. I'd be very much interested in using it with discretized mix logistic distribution. I note you seem to have implemented this function in nvae/loss.py.

You refer to it costing too much, is that costing too much in memory or compute time?

I assume line 40 in celeb_vae.py needs to be changed, but I wasn't quite sure how.

Thanks, Julian.

GlassyWing commented 3 years ago

Hi,

Very nice implementation, thank you. I'd be very much interested in using it with discretized mix logistic distribution. I note you seem to have implemented this function in nvae/loss.py.

You refer to it costing too much, is that costing too much in memory or compute time?

I assume line 40 in celeb_vae.py needs to be changed, but I wasn't quite sure how.

Thanks, Julian.

Sorry for the late reply，this is an incomplete implementation, I will try it in my spare time. For now, you can check the official implementation, which implements the method.