/pytorch-nce2

The Noise Contrastive Estimation for softmax output written in Pytorch

Primary LanguagePythonMIT LicenseMIT

Stargazers