search

649453932 / Chinese-Text-Classification-Pytorch

中文文本分类，TextCNN，TextRNN，FastText，TextRCNN，BiLSTM_Attention，DPCNN，Transformer，基于pytorch，开箱即用。
MIT License
5.19k stars 1.22k forks source link

textRNN模型中LSTM问题 #75

Closed ghost closed 3 years ago

ghost commented 3 years ago

你好，首先十分感谢你在文本分类模型中的整理，通过学你的代码，受益匪浅。但是在今天在运行代码的时候遇到一个小小的问题，想请教一下，就是在textRNN以及textRNN_Att中，你在使用LSTM的时候输入x的维度为[batch_size, seq_len, embedding]，但是在torch.nn.LSTM源码中是要求输入x的维度是(seq_len, batch, input_size)，想问一下作者这个地方是不是搞反了，或者说这么做是没问题的？

oyster-cs commented 3 years ago

batch_first=True 了解一下这个参数