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):
runner,
allow_scalar=True,
allow_text=False,
add_mode=True,
tags_to_skip=('time', 'data_time')):
tags = {}
for var, val in runner.log_buffer.output.items():
......@@ -127,8 +128,9 @@ class LoggerHook(Hook):
continue
if isinstance(val, str) and not allow_text:
continue
tag = f'{var}/{self.get_mode(runner)}'
tags[tag] = val
if add_mode:
var = f'{self.get_mode(runner)}/{var}'
tags[var] = val
tags.update(self.get_lr_tags(runner))
tags.update(self.get_momentum_tags(runner))
return tags
......
......@@ -71,7 +71,7 @@ class PaviLoggerHook(LoggerHook):
@master_only
def log(self, runner):
tags = self.get_loggable_tags(runner)
tags = self.get_loggable_tags(runner, add_mode=False)
if tags:
self.writer.add_scalars(
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