Not able to reproduce results for CNN-for-Text-Classification

[sksq96]

Hey,

I am trying to reproduce this notebook, but the loss do not go down as advertised.

[0/5] mean_loss : 1.80
[1/5] mean_loss : 1.64
[2/5] mean_loss : 1.64
[3/5] mean_loss : 1.62
[4/5] mean_loss : 1.76

I am using PyTorch v0.4 on CUDA. My hypothesis is the newer version broke something.

>torch.__version__
'0.4.0a0+a3e9151'

Thanks!

[DSKSD]

Hi,
Thanks for letting me know.
I don't test it on v0.4 yet. I'll check this out soon.
I wonder you initialize embedding matrix with pretrained Word2vec model.
If not, you should train your model much longer.