Possible to use Discretized mix logistic distribution?

[jfrancis71]

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]

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.