From 5f2739e6341d055db6cc3b13214bd5647bae0934 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Wed, 12 Oct 2022 19:28:58 -0700 Subject: [PATCH] Fix tb and wandb crash (#769) --- nerfstudio/models/nerfacto.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/models/nerfacto.py b/nerfstudio/models/nerfacto.py index a76b0534a..a45fafaab 100644 --- a/nerfstudio/models/nerfacto.py +++ b/nerfstudio/models/nerfacto.py @@ -227,8 +227,8 @@ class NerfactoModel(Model): loss_dict["interlevel_loss"] = self.config.interlevel_loss_mult * interlevel_loss( outputs["weights_list"], outputs["ray_samples_list"] ) - assert metrics_dict is not None and "distortion" in metrics_dict - loss_dict["distortion_loss"] = self.config.distortion_loss_mult * metrics_dict["distortion"] + assert metrics_dict is not None and "distortion" in metrics_dict + loss_dict["distortion_loss"] = self.config.distortion_loss_mult * metrics_dict["distortion"] return loss_dict def get_image_metrics_and_images(