def log_diagnostic(self, key, step, epoch, summary, **kwargs):
    if (key not in self.loggers):
        self.init_logger(key)
    self.loggers[key].log_diagnostic(step, epoch, summary, **kwargs)
    self.plot_handler.plot()
def log_diagnostic(self, key, step, epoch, summary, **kwargs):
    if (key not in self.loggers):
        self.init_logger(key)
    self.loggers[key].log_diagnostic(step, epoch, summary, **kwargs)
    self.plot_handler.plot()