tensorflow入门笔记（一） tf.app.flags.FLAGS

tf.app.flags.DEFINE_xxx()就是添加命令行的optional argument（可选参数），而tf.app.flags.FLAGS可以从对应的命令行参数取出参数。举例如下：

FLAGS = tf.app.flags.FLAGS
# 基本模型参数
# 定义integer型flag
tf.app.flags.DEFINE_integer(flag_name=\'batch_size\', default_value=128,
                            docstring=\'Number of images to process in a batch.\')
# 定义string型flag
tf.app.flags.DEFINE_string(flag_name=\'data_dir\', default_value=\'cifar-10-binary\',
                           docstring=\'Path to the CIFAR-10 data directory.\')

print(FLAGS.batch_size)
print(FLAGS.data_dir)

在命令行运行后如下：

因为没有给可选参数赋值，所以输出默认值。

给可选参数batch_size和data_dir赋值后输出输入值。

参考链接