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赋值后输出输入值。

参考链接