-
PyTorch实现的端到端内存网络,名为pytorch_memn2n,且具备关注层虚拟化功能支持
资源介绍
端到端目标导向对话框
此存储库包含端到端内存网络的PyTorch实现,如论文“ 还有一个用于复制T1-T5 bAbI任务结果的代码和一个jupyter笔记本文件,用于可视化学习模型的内存注意。
要求
- python 3.6
- pytorch 0.3.0
跑步
首先,您需要下载。
要运行培训,请使用以下模式:
python train.py /path/to/dataset/train_set_file.txt /path/to/dataset/dev_set_file.txt /path/to/dataset/candidates_file.txt
有不同的命令行参数可用于调整模型和训练参数。 有关完整列表,请运行
python train.py -h
为了进行评估,请使用:
python eval.py /path/to/saved/model/dir /path/to/dat