seq2seq.pytorch:使用PyTorch进行序列到序列学习 源码

PyTorch中的Seq2Seq 这是用于在训练序列到序列模型的完整套件。 它由几个模型和代码组成,可以训练和推断使用它们。 使用此代码,您可以训练: 神经机器翻译(NMT)模型 语言模型 图像到字幕的生成 跳过思维的句子表示 和更多... 安装 git clone --recursive https://github.com/eladhoffer/seq2seq.pytorch cd seq2seq.pytorch; python setup.py develop 楷模 当前可用的模型: 简单的Seq2Seq递归模型 带注意解码器的递归Seq2Seq (GNMT)递归模型 变形金刚-来自的仅关注模型 数据集 当前可用的数据集: WMT16 WMT17 OpenSubtitles 2016 COCO图片标题 可以使用3种可用的分割方法对所有数据集进行标记: 基于字符的细