def forward_teacher(self, x):
    y = self.teacher_backbone(x).flatten(start_dim=1)
    z = self.teacher_head(y)
    return z
def forward_teacher(self, x):
    y = self.teacher_backbone(x).flatten(start_dim=1)
    z = self.teacher_head(y)
    return z