上传者: 42126865
|
上传时间: 2021-10-12 10:04:26
|
文件大小: 670KB
|
文件类型: -
Tensorflow中的指针网络
TensorFlow实现。
支持多线程数据管道以减少I / O延迟。
要求
Python 2.7
用法
训练模型:
$ python main.py --task=tsp --max_data_length=20 --hidden_dim=512 # download dataset used in the paper
$ python main.py --task=tsp --max_data_length=10 --hidden_dim=128 # generate dataset itself
训练模型:
$ python main.py
$