Perform evaluation upon training completion (#1529)

open-mmlab · Apr 8, 2024 · e258c84 · e258c84
1 parent 2c4516c
commit e258c84
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mmengine/runner/loops.py b/mmengine/runner/loops.py
@@ -98,7 +98,8 @@ def run(self) -> torch.nn.Module:
             self._decide_current_val_interval()
             if (self.runner.val_loop is not None
                     and self._epoch >= self.val_begin
-                    and self._epoch % self.val_interval == 0):
+                    and (self._epoch % self.val_interval == 0
+                         or self._epoch == self._max_epochs)):
                 self.runner.val_loop.run()
 
         self.runner.call_hook('after_train')
@@ -288,7 +289,8 @@ def run(self) -> None:
             self._decide_current_val_interval()
             if (self.runner.val_loop is not None
                     and self._iter >= self.val_begin
-                    and self._iter % self.val_interval == 0):
+                    and (self._iter % self.val_interval == 0
+                         or self._iter == self._max_iters)):
                 self.runner.val_loop.run()
 
         self.runner.call_hook('after_train_epoch')