Fixed deadlock issue related with MMSegWandbHook (#2398)

Co-authored-by: WangYudong <yudong.wang@akane.waseda.jp>
open-mmlab · Jan 11, 2023 · c0515a1 · c0515a1
1 parent 320d8c6
commit c0515a1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mmseg/core/hook/wandblogger_hook.py b/mmseg/core/hook/wandblogger_hook.py
@@ -168,7 +168,8 @@ def before_run(self, runner):
             # Log ground truth data
             self._log_data_table()
 
-    @master_only
+    # for the reason of this double-layered structure, refer to
+    # https://github.com/open-mmlab/mmdetection/issues/8145#issuecomment-1345343076
     def after_train_iter(self, runner):
         if self.get_mode(runner) == 'train':
             # An ugly patch. The iter-based eval hook will call the
@@ -178,7 +179,10 @@ def after_train_iter(self, runner):
             return super(MMSegWandbHook, self).after_train_iter(runner)
         else:
             super(MMSegWandbHook, self).after_train_iter(runner)
+        self._after_train_iter(runner)
 
+    @master_only
+    def _after_train_iter(self, runner):
         if self.by_epoch:
             return