This will be the first official public release of the VItamin code base. VItamin is a python package for producing fast gravitational wave posterior samples.
GNU General Public License v3.0
20
stars
7
forks
source link
In CVAE_model.py the truncated normal reconstruction loss is ugly #96
I thought that I'd changed this to be less ugly so that you can just call TruncatedNormal once over the appropriate parameters and then evaluate the log_prob and sum over the dimensions (as done in the von Mises case).
I thought that I'd changed this to be less ugly so that you can just call TruncatedNormal once over the appropriate parameters and then evaluate the log_prob and sum over the dimensions (as done in the von Mises case).
I think it would be sensible to do that here.