mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed deadlock issue related with MMSegWandbHook (#2398) 

Co-authored-by: WangYudong <yudong.wang@akane.waseda.jp>
pull/2489/head
MilkClouds 2022-12-11 05:35:29 +00:00 committed by MeowZheng
parent 320d8c60f3
commit c0515a1be5
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmseg/core/hook/wandblogger_hook.py
View File

@ -168,7 +168,8 @@ class MMSegWandbHook(WandbLoggerHook):
# Log ground truth data # Log ground truth data
self._log_data_table() 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): def after_train_iter(self, runner):
if self.get_mode(runner) == 'train': if self.get_mode(runner) == 'train':
# An ugly patch. The iter-based eval hook will call the # An ugly patch. The iter-based eval hook will call the
@ -178,7 +179,10 @@ class MMSegWandbHook(WandbLoggerHook):
return super(MMSegWandbHook, self).after_train_iter(runner) return super(MMSegWandbHook, self).after_train_iter(runner)
else: else:
super(MMSegWandbHook, self).after_train_iter(runner) super(MMSegWandbHook, self).after_train_iter(runner)
self._after_train_iter(runner)
@master_only
def _after_train_iter(self, runner):
if self.by_epoch: if self.by_epoch:
return return