Fix typo for feature pre-commit

bamaao · Jun 22, 2022 · 28a7093 · 28a7093
1 parent f8c5cae
commit 28a7093
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/yolov6/core/engine.py b/yolov6/core/engine.py
@@ -61,7 +61,7 @@ def train(self):
             for self.epoch in range(self.start_epoch, self.max_epoch):
                 self.train_in_loop()
 
-        except Exception:
+        except Exception as _:
             LOGGER.error('ERROR in training loop or eval/save model.')
             raise   
         finally:
@@ -74,12 +74,12 @@ def train_in_loop(self):
             for self.step, self.batch_data in self.pbar:
                 self.train_in_steps()
                 self.print_details()     
-        except Exception:
+        except Exception as _:
             LOGGER.error('ERROR in training steps.')
             raise
         try:
             self.eval_and_save()
-        except Exception:
+        except Exception as _:
             LOGGER.error('ERROR in evaluate and save model.')
             raise