def __init__(self, freq=None, log_dir='.', rank=None, model=None):
'''
Constructor.
Args:
freq (`int` or `None`, default `None`): frequency at which progress is logged.
For example, the value of 2 will cause the progress to be logged every second time when
`:func:cntk.util.TensorBoardFileWriter.update_with_trainer` is invoked.
None indicates that progress is logged only when
`:func:cntk.util.TensorBoardFileWriter.summarize_progress` is invoked.
Must be a positive integer otherwise.
log_dir (`string`, default '.'): directory where to create a TensorBoard event file.
rank (`int` or `None`, default `None`): rank of a worker when using distributed training, or `None` if
training locally. If not `None`, event files will be created in log_dir/rank[rank] rather than log_dir.
model (:class:`cntk.ops.Function` or `None`, default `None`): model graph to plot.
'''
if freq is None:
freq = sys.maxsize
super(TensorBoardProgressWriter,
self).__init__(freq, 0, sys.maxsize, 0)
# Only log either when rank is not specified or when rank is 0.
self.writer = cntk_py.TensorBoardFileWriter(
log_dir, model) if not rank else None
self.closed = False
def __init__(self, freq=None, log_dir='.', rank=None, model=None):
'''
Constructor.
'''
if freq is None:
freq = sys.maxsize
super(TensorBoardProgressWriter, self).__init__(freq, 0, sys.maxsize, 0, sys.maxsize, 0)
# Only log either when rank is not specified or when rank is 0.
self.writer = cntk_py.TensorBoardFileWriter(log_dir, model) if not rank else None
self.closed = False