
TensorFlow价值迭代网络:基于NIPS 16的实现...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目基于NIPS 2016论文《Value Iteration Network》,采用TensorFlow框架实现了价值迭代网络模型,应用于路径规划等领域。
TensorFlow中的Tamar, A., Wu, Y., Thomas, G., Levine, S. 和 Abbeel, P.的价值迭代网络在2016年神经信息处理系统(NIPS)会议上获得了最佳论文奖,本存储库提供了该研究在TensorFlow上的实现。这个代码基于作者最初的Theano版本。
为了训练模型,请下载16x16和28x28 GridWorld数据集。此外,此存储库还包含一个较小的8x8 GridWorld数据集以供使用方便。
要运行默认参数设置下的8x8 Gridworld模型,请执行以下命令:
```
python3 train.py
```
如果需要监控训练进度,可以将config.log设为True,并启动tensorboard --logdir /tmp/vintf/。日志目录默认设定在/tmp/vintf/,但可以在config.logdir中更改此设置。
注意,默认情况下代码会运行8x8 GridWorld模型。
全部评论 (0)
还没有任何评论哟~


