def __init__(self, model, teacher, dataset, *args, **kwargs):
    super().__init__(model, dataset, *args, **kwargs)
    self.teacher = teacher
def __init__(self, model, teacher, dataset, *args, **kwargs):
    super().__init__(model, dataset, *args, **kwargs)
    self.teacher = teacher