TensorFlow 之命令行參數

TensorFlow項目例子中經常出現tf.app.flags，這個好像和tf.flags是一樣，加不加中間的app沒區別（要是不是這樣還請大家指出錯誤之處），它支持應用從命令行接受參數，可以用來指定集羣配置等。

import tensorflow as tf

#調用flags內部的DEFINE_string函數來制定解析規則
tf.flags.DEFINE_string("para_name_1","default_val", "description")
tf.flags.DEFINE_integer("batch_size", 64, "Batch Size (default: 64)")
tf.flags.DEFINE_integer("num_epochs", 10, "Number of training epochs (default: 10)")

#FLAGS是一個對象，保存瞭解析後的命令行參數
FLAGS = tf.flags.FLAGS
FLAGS._parse_flags()#進行解析，加上這一句可以把FLAGS.__flags變成一個字典
print(FLAGS.batch_size)#運行結果輸出64
print(FLAGS.__flags)#運行結果見下圖

當然，以上的都不是重點，重點來了，重點是它可以從命令行接受參數。結果見下圖吧。

可以看到當不傳入參數的時候它可以使用默認值，你也可以很方便的傳入自己的參數。