Commit 509d12f9 authored by Kai Chen's avatar Kai Chen
Browse files

fix init arguments

parent 523f861b
......@@ -4,7 +4,14 @@ from ..hook import Hook
class LoggerHook(Hook):
"""Base class for logger hooks."""
"""Base class for logger hooks.
Args:
interval (int): Logging interval (every k iterations).
ignore_last (bool): Ignore the log of last iterations in each epoch
if less than `interval`.
reset_flag (bool): Whether to clear the output buffer after logging.
"""
__metaclass__ = ABCMeta
......
......@@ -145,12 +145,11 @@ class PaviLoggerHook(LoggerHook):
instance_id=None,
config_file=None,
interval=10,
reset_meter=True,
ignore_last=True):
ignore_last=True,
reset_flag=True):
self.pavi = PaviClient(url, username, password, instance_id)
self.config_file = config_file
super(PaviLoggerHook, self).__init__(interval, reset_meter,
ignore_last)
super(PaviLoggerHook, self).__init__(interval, ignore_last, reset_flag)
def before_run(self, runner):
super(PaviLoggerHook, self).before_run(runner)
......
......@@ -4,13 +4,10 @@ from ...utils import master_only
class TensorboardLoggerHook(LoggerHook):
def __init__(self,
log_dir,
interval=10,
reset_meter=True,
ignore_last=True):
super(TensorboardLoggerHook, self).__init__(interval, reset_meter,
ignore_last)
def __init__(self, log_dir, interval=10, ignore_last=True,
reset_flag=True):
super(TensorboardLoggerHook, self).__init__(interval, ignore_last,
reset_flag)
self.log_dir = log_dir
@master_only
......
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