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 ...@@ -4,7 +4,14 @@ from ..hook import Hook
class LoggerHook(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 __metaclass__ = ABCMeta
......
...@@ -145,12 +145,11 @@ class PaviLoggerHook(LoggerHook): ...@@ -145,12 +145,11 @@ class PaviLoggerHook(LoggerHook):
instance_id=None, instance_id=None,
config_file=None, config_file=None,
interval=10, interval=10,
reset_meter=True, ignore_last=True,
ignore_last=True): reset_flag=True):
self.pavi = PaviClient(url, username, password, instance_id) self.pavi = PaviClient(url, username, password, instance_id)
self.config_file = config_file self.config_file = config_file
super(PaviLoggerHook, self).__init__(interval, reset_meter, super(PaviLoggerHook, self).__init__(interval, ignore_last, reset_flag)
ignore_last)
def before_run(self, runner): def before_run(self, runner):
super(PaviLoggerHook, self).before_run(runner) super(PaviLoggerHook, self).before_run(runner)
......
...@@ -4,13 +4,10 @@ from ...utils import master_only ...@@ -4,13 +4,10 @@ from ...utils import master_only
class TensorboardLoggerHook(LoggerHook): class TensorboardLoggerHook(LoggerHook):
def __init__(self, def __init__(self, log_dir, interval=10, ignore_last=True,
log_dir, reset_flag=True):
interval=10, super(TensorboardLoggerHook, self).__init__(interval, ignore_last,
reset_meter=True, reset_flag)
ignore_last=True):
super(TensorboardLoggerHook, self).__init__(interval, reset_meter,
ignore_last)
self.log_dir = log_dir self.log_dir = log_dir
@master_only @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