1

如何将嵌套的整数列表传递给gflags?我可以让我的代码使用类似的东西

flags = tf.app.flags
FLAGS = flags.FLAGS
flags.DEFINE_float('network_nodes', [784, [5, 5, 32], [5, 5, 64], 1200, 10], 'The network structure')

但是尝试在命令行中使用它,例如

python test.py --network_nodes=[784, 100, 10]

结果(不足为奇)出现错误:

test.py: error: argument --network_nodes: invalid int value: '[784, 100, 10]'

如何将嵌套的整数列表传递给gflags(或 TensofFlow 的tf.app.flags)?

4

1 回答 1

3

来自文档:https ://github.com/gflags/python-gflags/blob/master/gflags.py

DEFINE_list: Takes a comma-separated list of strings on the commandline.
         Stores them in a python list object.

因此,您必须将其作为字符串列表传递--network_nodes=784,100,10,然后将字符串转换为浮点数。

于 2016-05-18T17:09:48.443 回答