标签: pytorch rnn sequence-generators
我也在pytorch讨论论坛上问了这个问题(link),但我想我会在这里问一下它是否会有更大的牵引力。
我使用pytorch的{{1}}模块训练了一个模型,通过提供批量序列(变化nn.RNN())并在批次中的序列长度不同时正确填充。
nn.RNN()
我现在正在尝试使用训练有素的RNN模型生成序列。我制作一个长度为1的序列,其中包含我seq_len的索引并将其提供给我训练有素的RNN。不幸的是,我要回来的都是0。有人曾经这样做过吗?
seq_len
有些代码可以帮助:
GO_TOKEN
有什么想法吗?