mean-teacher
mean-teacher copied to clipboard
Fix support for models with one output (PyTorch implementation)
The PyTorch code currently crashes if the model has only one output. See comments on issue #7 .