From e258c84824bb57b3d0e5177ba57b1d81adf4a90a Mon Sep 17 00:00:00 2001 From: Zhihao Lin <36994684+LZHgrla@users.noreply.github.com> Date: Mon, 8 Apr 2024 13:05:36 +0800 Subject: [PATCH] Perform evaluation upon training completion (#1529) --- mmengine/runner/loops.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mmengine/runner/loops.py b/mmengine/runner/loops.py index 1f6551ab62..329fd48914 100644 --- 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')