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賦值后輸出輸入值。

參考鏈接