python-3.x - LSTM 参数

Question

大家好，我正在使用pytorch进行文本分类，我第一次使用44个数据作为训练数据，22个数据作为测试数据，我的词汇量是531，我的lstm参数如下所示，但最后在嵌入我得到这个错误：

IndexError：自我中的索引超出范围

我尝试了不同的 embed_size，但没有成功，你能帮帮我吗？

谢谢

# LSTM parameters
embed_size =300
hidden_size = 256
num_layers = 1

# training parameters
lr = 0.001
num_epochs = 10

标签： python-3.xlstmpython-embedding