From 5c90d9013e9e84e2fa25a00c32ae2ec98eb0bf60 Mon Sep 17 00:00:00 2001
From: Coolshanlan <coolshanlan@gmail.com>
Date: Tue, 6 Dec 2022 17:46:26 +0800
Subject: [PATCH] fix evaluation result record bug

---
 tools/train_net.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/train_net.py b/tools/train_net.py
index e8e3aaeda..cf6c45af3 100644
--- a/tools/train_net.py
+++ b/tools/train_net.py
@@ -67,7 +67,7 @@ def build_hooks(self):
                 )
                 ret[i] = hooks.PeriodicCheckpointer(self.checkpointer, self.cfg.SOLVER.CHECKPOINT_PERIOD)
         return ret
-    
+
     def resume_or_load(self, resume=True):
         checkpoint = self.checkpointer.resume_or_load(self.cfg.MODEL.WEIGHTS, resume=resume)
         if resume and self.checkpointer.has_checkpoint():
@@ -90,6 +90,7 @@ def train_loop(self, start_iter: int, max_iter: int):
                 self.before_step()
                 self.run_step()
                 self.after_step()
+            self.iter += 1
             self.after_train()
 
     def train(self):