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]: