迷你seq2seq
在PyTorch中注意神经机器翻译的最小Seq2Seq模型。
此实现重点在于以下功能:
用于其他项目的模块化结构
最少的代码可读性
充分利用批次和GPU。
此实现依赖于来最大程度地减少数据集管理和预处理部分。
型号说明
编码器:双向GRU
解码器:具有注意机制的GRU
注意:
要求
GPU和CUDA
Python3
火炬
火炬文本
空间
麻木
智慧(可选)
通过这样做下载令牌生成器:
python -m spacy download de
python -m spacy download en
参考文献
基于以下实现
1