使用pytorch RNN生成序列

时间:2017-11-01 15:33:38

标签: pytorch rnn sequence-generators

我也在pytorch讨论论坛上问了这个问题(link),但我想我会在这里问一下它是否会有更大的牵引力。

我使用pytorch的{​​{1}}模块训练了一个模型,通过提供批量序列(变化nn.RNN())并在批次中的序列长度不同时正确填充。

我现在正在尝试使用训练有素的RNN模型生成序列。我制作一个长度为1的序列,其中包含我seq_len的索引并将其提供给我训练有素的RNN。不幸的是,我要回来的都是0。有人曾经这样做过吗?

有些代码可以帮助:

GO_TOKEN

有什么想法吗?

0 个答案:

没有答案