From 261f1f33447d5da751d2e0acf4aa435da22ea2fe Mon Sep 17 00:00:00 2001 From: Alexander Kristoffersen Date: Tue, 16 Aug 2022 13:25:41 -0700 Subject: [PATCH] Double Scaling Bug Fix (#282) remove duplicate scaling. --- nerfactory/models/base.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/nerfactory/models/base.py b/nerfactory/models/base.py index 906c77321..29be1eee0 100644 --- a/nerfactory/models/base.py +++ b/nerfactory/models/base.py @@ -175,10 +175,6 @@ class Model(nn.Module): outputs=outputs, batch=batch, metrics_dict=metrics_dict, loss_coefficients=self.loss_coefficients ) - # scaling losses by coefficients. - for loss_name in loss_dict.keys(): - if loss_name in self.loss_coefficients: - loss_dict[loss_name] *= self.loss_coefficients[loss_name] return outputs, loss_dict, metrics_dict def get_metrics_dict(self, outputs, batch) -> Dict[str, torch.Tensor]: