【发布时间】:2021-07-15 19:06:02
【问题描述】:
我正在 PyTorch 中使用 RNN 构建文本分类器。我使用的嵌入是 GLOVE。但是,我将可变长度索引引用提供给模型。这将导致可变长度嵌入,我认为它不会起作用。我如何解决这个问题并使所有句子的嵌入输出长度相同?
def forward(self, sentence):
embeds = self.embedding(sentence)
hidden = self.__init__hidden(size)
output, hidden = self.rnn(embeds, hidden)
out = self.hidden2out(output)
另外,如果有人能告诉我如何选择隐藏层的大小,那就太好了。
【问题讨论】:
标签: python machine-learning deep-learning nlp pytorch