【文件属性】:
文件名称:Pytorch-seq2seq-Beam-Search:带有注意力和贪婪搜索束搜索的Seq2Seq模型的PyTorch实现,用于神经机器翻译
文件大小:5.75MB
文件格式:ZIP
更新时间:2021-04-18 08:54:40
pytorch seq2seq beam-search nmt greedy-search
Pytorch-seq2seq-光束搜索
带有注意力和贪婪搜索/波束搜索的Seq2Seq模型,用于在PyTorch中进行神经机器翻译。
此实现着重于以下功能:
用于其他项目的模块化结构
最少的代码以提高可读性
充分利用批处理和GPU。
解码方法贪婪搜索
解码方法波束搜索
此实现依赖于来最大程度地减少数据集管理和预处理部分。
Seq2Seq型号说明
seq2seq的主要结构采用
编码器:双向GRU
解码器:具有注意机制的GRU
注意力
解码方式
贪婪的搜索
光束搜索
要求
CUDA
Python 3.6
PyTorch 1.4
火炬文本
空间
麻木
智慧(可选)
通过这样做下载令牌生成器:
python -m spacy download de
python -m spacy download en
去做
添加日志
更改为支持gpu和cpu,目前是基于gpu的代码实现
【文件预览】:
Pytorch-seq2seq-Beam-Search-master
----.gitignore(1KB)
----log.py(424B)
----LICENSE(1KB)
----note.md(982B)
----utils.py(1KB)
----README.md(2KB)
----.idea()
--------.gitignore(39B)
--------misc.xml(197B)
--------vcs.xml(180B)
--------inspectionProfiles()
--------modules.xml(306B)
----img()
--------李宏毅att3.png(516KB)
--------李宏毅att4.png(602KB)
--------李宏毅att2.png(693KB)
--------李宏毅att1.png(549KB)
--------李宏毅att5.png(594KB)
--------google-seq2seq架构图.png(374KB)
----.data()
--------multi30k()
----model.py(12KB)
----train.py(5KB)