上传者: 42128988
|
上传时间: 2021-11-19 18:43:06
|
文件大小: 1.03MB
|
文件类型: -
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种可用的分割方法对所有数据集进行标记:
基于字符的细