Stonesjtu Pytorch NCE Versions Save

The Noise Contrastive Estimation for softmax output written in Pytorch

neat-nce

6 years ago
  • The main file contains the minimal details required. Many helper functions are moved into utils file.
  • Model's API is simplified a lot.
  • Speed issues remain to be solved.