Method _freeze

model/depth_wrapper.py:30–34

(self)

        return model

    def _freeze(self):
        # logger.warning(f"======== Freezing Dinov2Wrapper ========")
        self.model.eval()
        for name, param in self.model.named_parameters():
            param.requires_grad = False
        
    def forward(self, x):
        return self.model(x)