Open jfrancis71 opened 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.
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.