commenting out extra predict interface

Author: sarthakpati

GANDLF/models/imagenet_unet.py

@@ -40,23 +40,24 @@ def forward(self, x):
             labels = self.classification_head(features[-1])
             return labels
 
-    def predict(self, x):
-        """Inference method. Switch model to `eval` mode, call `.forward(x)` with `torch.no_grad()`
+    ## commented out because we are not using this interface
+    # def predict(self, x):
+    #     """Inference method. Switch model to `eval` mode, call `.forward(x)` with `torch.no_grad()`
 
-        Args:
-            x: 4D torch tensor with shape (batch_size, channels, height, width)
+    #     Args:
+    #         x: 4D torch tensor with shape (batch_size, channels, height, width)
 
-        Return:
-            prediction: 4D torch tensor with shape (batch_size, classes, height, width)
+    #     Return:
+    #         prediction: 4D torch tensor with shape (batch_size, classes, height, width)
 
-        """
-        if self.training:
-            self.eval()
+    #     """
+    #     if self.training:
+    #         self.eval()
 
-        with torch.no_grad():
-            x = self.forward(x)
+    #     with torch.no_grad():
+    #         x = self.forward(x)
 
-        return x
+    #     return x
 
 
 class Unet(SegmentationModel):