more printing

niki-amini-naieni · Nov 5, 2023 · 916af68 · 916af68
1 parent 9a0c104
commit 916af68
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nerf/utils.py b/nerf/utils.py
@@ -326,6 +326,7 @@ def nll(truths, preds, vars):
  vars = vars.flatten(end_dim=-2).cpu().numpy()
  log_pdf_vals = []
  for px_ind in range(vars.shape[0]):
+ print(np.diag(vars[px_ind]))
  log_pdf_vals.append(
  np.log(
  multivariate_normal(truths[px_ind], preds[px_ind], np.diag(vars[px_ind]))