added eps to nll est

niki-amini-naieni · Nov 5, 2023 · fc2e9f9 · fc2e9f9
1 parent 74cf59e
commit fc2e9f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nerf/utils.py b/nerf/utils.py
@@ -327,9 +327,9 @@ def nll(truths, preds, vars):
  log_pdf_vals = []
  for px_ind in range(vars.shape[0]):
  log_pdf_vals.append(
- np.log(
+ -np.log(
  multivariate_normal.pdf(truths[px_ind], mean=preds[px_ind], cov=np.diag(vars[px_ind]), allow_singular=False)
- )
+ ) + 1e-12
  )
  return np.mean(log_pdf_vals)