2022.08.30 19:23

2gunsu · Aug 30, 2022 · dbbc806 · dbbc806
1 parent d40ebb0
commit dbbc806
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/model/detector/monocon_detector.py b/model/detector/monocon_detector.py
@@ -59,7 +59,10 @@ def forward(self, data_dict: Dict[str, Any], return_loss: bool = True) -> Tuple[
             return pred_dict
 
 
-    def batch_eval(self, data_dict: Dict[str, Any], get_vis_format: bool = False) -> Dict[str, Any]:
+    def batch_eval(self, 
+                   data_dict: Dict[str, Any], 
+                   get_vis_format: bool = False) -> Dict[str, Any]:
+
         if self.training:
             raise Exception(f"Model is in training mode. Please use '.eval()' first.")