Method train

model/splat_model.py:43–47

(self, mode=True)

                self.lpips_loss = torch.compile(self.lpips_loss)
        
    def train(self, mode=True):
        super().train(mode)
        if 'lpips_loss' in self.__dict__:
            self.lpips_loss.eval()
        return self

    def state_dict(self, **kwargs):
        """Remove non-trainable modules (LPIPS, tracker) from state dict before saving."""