def close(self):
# save text
history = {"exec": ' '.join(get_args())}
if hasattr(self.callbacks, "callbacks"):
for c in self.callbacks.callbacks:
if isinstance(c, MetricCallback):
history[c.metric_name] = c.history
with (self.save_dir / "results.json").open('w') as f:
json.dump(history, f)
def __init__(self,
save_dir: Optional[str or Path],
report_freq: Optional[int] = None,
is_global_step_epoch: bool = True):
super(TensorboardReporter, self).__init__()
from torch.utils import tensorboard
save_dir = Path(save_dir) / BASIC_DIR_NAME
save_dir.mkdir(exist_ok=True, parents=True)
self.writer = tensorboard.SummaryWriter(save_dir)
self._report_freq = report_freq
self._use_epoch = is_global_step_epoch
self.writer.add_text("exec", ' '.join(get_args()))