Mean Teacher for semi supervised learning

Hi all,
please can anyone tell me how to solve this issue? I am trying to train a net with the semi supervised method ‘mean teacher’ and have a problem to update the weight of the teacher.

Based on the error message you are running into a shape mismatch while trying to update one of the parameters. Check the shape of the parameters and make sure add_ would get the expected shapes.

1 Like