diff --git a/nerfstudio/models/bakedsdf.py b/nerfstudio/models/bakedsdf.py index 5af507f1..8a8cac4b 100644 --- a/nerfstudio/models/bakedsdf.py +++ b/nerfstudio/models/bakedsdf.py @@ -256,9 +256,8 @@ def sample_and_forward_field(self, ray_bundle: RayBundle): return samples_and_field_outputs def get_loss_dict(self, outputs, batch, metrics_dict=None): - loss_dict = {} image = batch["image"].to(self.device) - loss_dict["rgb_loss"] = self.rgb_loss(image, outputs["rgb"]) + loss_dict = super().get_loss_dict(outputs, batch, metrics_dict) if self.training: # eikonal loss