def log_diagnostic(self, global_step: int, epoch: int, summary: Diagnostic, **kwargs):
    for key in self.diagnostic_keys:
        self.tracker.append(global_step, summary[key])
def log_diagnostic(self, global_step: int, epoch: int, summary: Diagnostic, **kwargs):
    for key in self.diagnostic_keys:
        self.tracker.append(global_step, summary[key])