durantto:

cifar_shakeshake26


I’m not too sure what’s going on because the model is intended to return two outputs: https://github.com/CuriousAI/mean-teacher/blob/99cd11f646ec2e4cc6f19836be96a72c55923e00/pytorch/mean_teacher/architectures.py#L161 161 .
One thing you can do is set a pdb breakpoint there and see if the model output is actually two outputs.