Stonesjtu / Pytorch-NCE

The Noise Contrastive Estimation for softmax output written in Pytorch
MIT License
318 stars 46 forks source link

Add importance sampling as another loss function and batched NCE/IS for speed #9

Closed Stonesjtu closed 6 years ago