1. tf.app.flags，用於支援接受命令列傳遞引數

import tensorflow as tf
#第一個是引數名稱，第二個引數是預設值，第三個是引數描述
tf.app.flags.DEFINE_string('str_name', 'def_v_1',"descrip1")
tf.app.flags.DEFINE_integer('int_name', 10,"descript2")
tf.app.flags.DEFINE_boolean('bool_name', False, "descript3")
FLAGS = tf.app.flags.FLAGS

呼叫引數：FLAGS.int_name = 14
 

python tt.py --str_name test_str --int_name 99 --bool_name True

1. 使用argparse包

parser = argparse.ArgumentParser()
parser.add_argument("--input_dir", help="path to folder containing images")
parser.add_argument("--mode", required=True, choices=["train", "test", "export"])
parser.add_argument("--output_dir"
 
, required=True, help="where to put output files")
parser.add_argument("--seed", type=int)

a = parser.parse_args()
呼叫引數a.mode = "train"