def __init__(self, *args, diagnostic_keys=['loss'], **kwargs):
    super().__init__(*args)
    self.diagnostic_keys = diagnostic_keys
    self.tracker = DataTracker(label=self.key)
def __init__(self, *args, diagnostic_keys=['loss'], **kwargs):
    super().__init__(*args)
    self.diagnostic_keys = diagnostic_keys
    self.tracker = DataTracker(label=self.key)