textCNN在推理时，dropout没有去掉？

649453932 / Chinese-Text-Classification-Pytorch

中文文本分类，TextCNN，TextRNN，FastText，TextRCNN，BiLSTM_Attention，DPCNN，Transformer，基于pytorch，开箱即用。

MIT License

5.25k stars 1.22k forks source link

Closed littttttlebird closed 4 years ago

littttttlebird commented 4 years ago

在测试的时候，不需要dropout吧?

li199603 commented 4 years ago

测试的时候都有这句model.eval() torch里面就会自动关闭dropout层了

littttttlebird commented 4 years ago

查了一下，确实是这样