Commit 4f4b0ba3 authored by Yuxin Wu's avatar Yuxin Wu

stupid bugfix

parent 5467e48a
...@@ -196,12 +196,15 @@ class TFEventWriter(TrainingMonitor): ...@@ -196,12 +196,15 @@ class TFEventWriter(TrainingMonitor):
""" """
Write summaries to TensorFlow event file. Write summaries to TensorFlow event file.
""" """
def __init__(self, logdir, max_queue=10, flush_secs=120): def __init__(self, logdir=None, max_queue=10, flush_secs=120):
""" """
Args: Args:
Same as in :class:`tf.summary.FileWriter`. Same as in :class:`tf.summary.FileWriter`.
logdir will be ``logger.get_logger_dir()`` by default. logdir will be ``logger.get_logger_dir()`` by default.
""" """
if logdir is None:
logdir = logger.get_logger_dir()
assert os.path.isdir(logdir), logdir
self._logdir = logdir self._logdir = logdir
self._max_queue = max_queue self._max_queue = max_queue
self._flush_secs = flush_secs self._flush_secs = flush_secs
...@@ -211,7 +214,7 @@ class TFEventWriter(TrainingMonitor): ...@@ -211,7 +214,7 @@ class TFEventWriter(TrainingMonitor):
logdir = logger.get_logger_dir() logdir = logger.get_logger_dir()
if logdir is not None: if logdir is not None:
return super(TFEventWriter, cls).__new__(cls, logdir, max_queue, flush_secs) return super(TFEventWriter, cls).__new__(cls)
else: else:
logger.warn("logger directory was not set. Ignore TFEventWriter.") logger.warn("logger directory was not set. Ignore TFEventWriter.")
return NoOpMonitor() return NoOpMonitor()
......
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