登录 注册
当前位置:主页 > 资源下载 > 13 > PyTorch实现的端到端内存网络,名为pytorch_memn2n,且具备关注层虚拟化功能支持

PyTorch实现的端到端内存网络,名为pytorch_memn2n,且具备关注层虚拟化功能支持

  • 更新:2024-07-30 09:45:20
  • 大小:618KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

端到端目标导向对话框 此存储库包含端到端内存网络的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