Unverified Commit b74aa792 authored by Wang Xinjiang's avatar Wang Xinjiang Committed by GitHub
Browse files

Move mode str to the beginning (#634)

parent 2d52809c
...@@ -118,6 +118,7 @@ class LoggerHook(Hook): ...@@ -118,6 +118,7 @@ class LoggerHook(Hook):
runner, runner,
allow_scalar=True, allow_scalar=True,
allow_text=False, allow_text=False,
add_mode=True,
tags_to_skip=('time', 'data_time')): tags_to_skip=('time', 'data_time')):
tags = {} tags = {}
for var, val in runner.log_buffer.output.items(): for var, val in runner.log_buffer.output.items():
...@@ -127,8 +128,9 @@ class LoggerHook(Hook): ...@@ -127,8 +128,9 @@ class LoggerHook(Hook):
continue continue
if isinstance(val, str) and not allow_text: if isinstance(val, str) and not allow_text:
continue continue
tag = f'{var}/{self.get_mode(runner)}' if add_mode:
tags[tag] = val var = f'{self.get_mode(runner)}/{var}'
tags[var] = val
tags.update(self.get_lr_tags(runner)) tags.update(self.get_lr_tags(runner))
tags.update(self.get_momentum_tags(runner)) tags.update(self.get_momentum_tags(runner))
return tags return tags
......
...@@ -71,7 +71,7 @@ class PaviLoggerHook(LoggerHook): ...@@ -71,7 +71,7 @@ class PaviLoggerHook(LoggerHook):
@master_only @master_only
def log(self, runner): def log(self, runner):
tags = self.get_loggable_tags(runner) tags = self.get_loggable_tags(runner, add_mode=False)
if tags: if tags:
self.writer.add_scalars( self.writer.add_scalars(
self.get_mode(runner), tags, self.get_step(runner)) self.get_mode(runner), tags, self.get_step(runner))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment