-
使用PyTorch实现Seq2Seq的序列到序列模型
资源介绍
Seq2Seq-PyTorch
使用PyTorch的序列到序列实现
安装
克隆项目,进入项目目录并执行
python setup.py install
或者
pip install ./
或简单地复制源代码。
推荐使用pip install ./ ,因为您可以先激活虚拟环境,然后再在该环境中安装软件包,而不会影响其他环境。
用法
使用之前,将seq2seq文件夹作为软件包安装或复制到项目目录。 看
一些功能
Trainer支持,尽管内存有限,但可以实现更大(等效)的批处理大小。
去做
支持光束搜索。
修理trainer 。 保存培训检查点时, trainer不会保存最佳时期模型。 因此,如果继续训练,则完成后保存的最佳时期实际上不是整个训练阶段的最佳时期,而是检查点之后的最佳时期。 (不知道培训师是否应在每个检查点保存最好的模型,这会使检查点文件变大。)
(不确定是否有必要。)