Pytorch-NCE icon indicating copy to clipboard operation
Pytorch-NCE copied to clipboard

The Noise Contrastive Estimation for softmax output written in Pytorch

Results 3 Pytorch-NCE issues
Sort by recently updated
recently updated
newest added

Hi, I think there is a bug here: https://github.com/Stonesjtu/Pytorch-NCE/blob/862afc666445dca4ce9d24a3eb1e073255edb92e/nce.py#L198 For RNN model which the last layer before softmax has shape [B * N * D] where time steps `N>1`, I...

I choose different num_layers params, but it still 1, and I rewrite code, give me lots of error.

README.md refers to option `--nce`, for example `python main.py --cuda --noise-ratio 10 --norm-term 9 --nce --train` `example/utils.py` does not have `--nce` in `setup_parser()` Result: ``` think0 tonyr: python main.py --nce...