def log_diagnostic(self, global_step: int, epoch: int, summary: Diagnostic, **kwargs):
    raise NotImplementedError
def log_diagnostic(self, global_step: int, epoch: int, summary: Diagnostic, **kwargs):
    raise NotImplementedError