def log_diagnostic(self, global_step: int, epoch: int, summary: Diagnostic, **kwargs):
    summary.log(self.writer, global_step)
def log_diagnostic(self, global_step: int, epoch: int, summary: Diagnostic, **kwargs):
    summary.log(self.writer, global_step)