def __init__(self, *args, **kwargs):
    super().__init__(*args)
    self.writer = SummaryWriter(os.path.join(self.logdir, self.key))
def __init__(self, *args, **kwargs):
    super().__init__(*args)
    self.writer = SummaryWriter(os.path.join(self.logdir, self.key))