search

jingweiz / pytorch-dnc

Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom
MIT License
278 stars 52 forks source link

应该如何将您的程序作为API使用? #10

Open Qchan233 opened 7 years ago

Qchan233 commented 7 years ago

我们希望在学术项目中使用您的代码，需要的功能是基于原有状态运行一步DNC，返回状态和输出。请问哪一个文件实现了这样的功能？

onlytailei commented 7 years ago

并不是很清楚具体需求:

1. 我们希望在学术项目中使用您的代码

请遵循license, 并欢迎引用我们的相关论文

@article{zhang2017neural,
title={Neural SLAM},
author={Zhang, Jingwei and Tai, Lei and Boedecker, Joschka and Burgard, Wolfram and Liu, Ming},
journal={arXiv preprint arXiv:1706.09520},
year={2017}}

2. 需要的功能是基于原有状态运行一步DNC，返回状态和输出

环境封装成类似gym一样的接口,带有step和reset函数,就可以直接运行DNC.

3. 请问哪一个文件实现了这样的功能？

如果是自己的环境,封装成gym接口,详情请参考env, 并在utils完成环境的注册和options的参数修改.