Removing confusing softmax, this is done in the consistency loss itself

CuriousAI / mean-teacher

A state-of-the-art semi-supervised method for image recognition

https://arxiv.org/abs/1703.01780

Other

1.59k stars 336 forks source link

Removing confusing softmax, this is done in the consistency loss itself #6

Closed perone closed 6 years ago

perone commented 6 years ago

Just removing the softmax here, this is being done in the consistency losses (softmax_mse_loss and softmax_kl_loss) and is unused here, causing confusion.

tarvaina commented 6 years ago

Thank you!