def mse_loss(teacher_logits, student_logits):
    loss = tf.reduce_mean(tf.nn.l2_loss((teacher_logits - student_logits)))
    return loss
def mse_loss(teacher_logits, student_logits):
    loss = tf.reduce_mean(tf.nn.l2_loss((teacher_logits - student_logits)))
    return loss