tensorflow學習筆記(一):命令列引數
全域性環境下編寫程式碼
import tensorflow as tf
flags = tf.flags #flags是一個檔案:flags.py,用於處理命令列引數的解析工作
logging = tf.logging
#呼叫flags內部的DEFINE_string函式來制定解析規則
flags.DEFINE_string("para_name_1","default_val", "description")
flags.DEFINE_bool("para_name_2","default_val", "description")
#FLAGS是一個物件,儲存瞭解析後的命令列引數
FLAGS = flags.FLAGS
def main(_):
FLAGS.para_name #呼叫命令列輸入的引數
if __name__ == "__main__": #使用這種方式保證了,如果此檔案被其它檔案import的時候,不會執行main中的程式碼
tf.app.run() #解析命令列引數,呼叫main函式 main(sys.argv)
呼叫方法:
~/ python script.py --para_name_1=name --para_name_2=name2
# 不傳的話,會使用預設值