VariationalRecurrentNeuralNetwork icon indicating copy to clipboard operation
VariationalRecurrentNeuralNetwork copied to clipboard

nll loss maybe got nan

Open jvnext opened this issue 4 years ago • 0 comments

the code def _nll_bernoulli(self, theta, x): return - torch.sum(x*torch.log(theta + EPS) + (1-x)*torch.log(1-theta-EPS)) may got nan loss. i think it should be def _nll_bernoulli(self, theta, x): return - torch.sum(x*torch.log(theta + EPS) + (1-x)*torch.log(1-theta+EPS))

jvnext avatar Apr 06 '22 05:04 jvnext