LSTM_attention方法中encoder部分LSTM cell的输入维度的规则貌似错了

Ligouhai-bigone / Sentiment-Analysis-Chinese-pytorch

Sentiment Analysis-Pytorch(情感分析的Pytorch实现)

MIT License

57 stars 7 forks source link

LSTM_attention方法中encoder部分LSTM cell的输入维度的规则貌似错了 #1

Open siyu1992 opened 4 years ago

siyu1992 commented 4 years ago

在Sentiment-Analysis-Chinese-pytorch.py文件的LSTM_attention方法中， LSTM_attention的forward方法，encoder的LSTM输入如下： states, hidden = self.encoder(embeddings.permute([0, 1, 2]))#[batch, seq_len, embed_dim]

在LSTM中也没有定义batch_first=True，这样LSTM输入的上下文是不是会有问题，序列都变成按列读了，这个是不是需要修复一下，或者有别的用意呢？谢谢

ipengx1029 commented 2 years ago

我也觉得是应该是需要batch_first的