1、原因: 使用view时维度指定错误,LSTM(input,(h0,c0)) 指定batch_first=True后,input就是(batch_size,seq_len,input_size)否则为input(seq_len, batch, input_size)
lstm_out, self.hidden = self.lstm(
embeds.view(self.batch_size, 200, EMBEDDING_DIM), self.hidden)
2、原因:并不是rnn的错误,而是因为下一函数的输入和这一层输出维度不一样,对照维度信息和尺寸信息修改即可。
推荐报错解决方案:使用pytorch时遇到的问题汇总_vivian_ll的博客-CSDN博客
文章出处登录后可见!
已经登录?立即刷新