Fixed deadlock issue related with MMSegWandbHook (open-mmlab#2398)

Co-authored-by: WangYudong <[email protected]>

Author: 2 people

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