-
Notifications
You must be signed in to change notification settings - Fork 68
Open
Description
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))
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels