Stonesjtu / Pytorch-NCE

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

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

Closed Stonesjtu closed 6 years ago