1. 程式人生 > >tensorflow學習記錄-flag設定

tensorflow學習記錄-flag設定

tf定義了tf.app.flags,用於支援接受命令列傳遞引數,相當於接受argv
import tensorflow as tf
flags = tf.flags #flags是一個檔案:flags.py,用於處理命令列引數的解析工作

#第一個是引數名稱,第二個引數是預設值,第三個是引數描述
flags.DEFINE_string("para_1","default_val", "description")
flags.DEFINE_bool("para_2","default_val", "description")

#FLAGS是一個物件,儲存瞭解析後的命令列引數
FLAGS = flags.FLAGS

def main(_):
    FLAGS.para_1 #呼叫命令列輸入的引數

if __name__ = "__main__": #使用這種方式保證了,如果此檔案被其它檔案import的時候,不會執行main中的程式碼
    tf.app.run() #解析命令列引數,呼叫main函式 main(sys.argv)

傳入引數方法:
/ python script.py --para_1=value1 --para_2=value2
# 不傳的話,會使用預設值,注意等號左右沒有空格