fix reference before assignment victoresque#57

pocca2048 · Aug 27, 2019 · 4280218 · 4280218
1 parent 6bc9dac
commit 4280218
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/base/base_trainer.py b/base/base_trainer.py
@@ -61,6 +61,7 @@ def train(self):
         """
         Full training logic
         """
+        not_improved_count = 0
         for epoch in range(self.start_epoch, self.epochs + 1):
             result = self._train_epoch(epoch)
 
@@ -90,7 +91,6 @@ def train(self):
                                         "Model performance monitoring is disabled.".format(self.mnt_metric))
                     self.mnt_mode = 'off'
                     improved = False
-                    not_improved_count = 0
 
                 if improved:
                     self.mnt_best = log[self.mnt_metric]