Feature Distillation Loss

Problem

Compute the mean squared error between student and teacher feature maps.

Return the MSE over all positions rounded to 4 decimal places.

case 1

in: [[[1,2],[3,4]],[[1,1],[3,6]]]

expected: 1.25

case 2

in: [[[5,5]],[[5,5]]]

expected: 0

case 3

in: [[[0,0],[0,0]],[[1,1],[1,1]]]

expected: 1

Starter codePython

def feature_distill_loss(student, teacher):
    # Your code here
    pass

3 test cases · run your code against them in the editor