Commented out code that raises ValueError for NaN in predictions

stefanklut · Jan 31, 2024 · bc0fbd4 · bc0fbd4
1 parent 29a5dd1
commit bc0fbd4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run.py b/run.py
@@ -184,8 +184,8 @@ def cpu_call(self, original_image: np.ndarray):
             with torch.autocast(device_type=self.cfg.MODEL.DEVICE, enabled=self.cfg.MODEL.AUTOCAST):
                 predictions = self.model([inputs])[0]
 
-            if torch.isnan(predictions["sem_seg"]).any():
-                raise ValueError("NaN in predictions")
+            # if torch.isnan(predictions["sem_seg"]).any():
+            #     raise ValueError("NaN in predictions")
 
         return predictions, height, width