From 47b1ef9898a65a540a1a577e831a1f164ebbe8da Mon Sep 17 00:00:00 2001 From: machenmusik Date: Thu, 9 Feb 2023 10:26:15 -0500 Subject: [PATCH] fix error with loss not defined --- nerfstudio/model_components/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/model_components/losses.py b/nerfstudio/model_components/losses.py index 228f4bb34..b65a361a0 100644 --- a/nerfstudio/model_components/losses.py +++ b/nerfstudio/model_components/losses.py @@ -326,5 +326,5 @@ def robust_rgb_loss(rgb_gt,rgb_outputs,patch_size=16): # final_w[:, # (patch_size//2-patch_size//4):(patch_size//2+patch_size//4), # (patch_size//2-patch_size//4):(patch_size//2+patch_size//4)] = weight_r8[:,None,None] - # loss = torch.mean(residuals.squeeze()*weight_r8[:,None,None]) + loss = torch.mean(residuals.squeeze()*weight_r8[:,None,None]) return loss \ No newline at end of file