【Tensorflow】tf.app.run() 與 tf.app.flags()
阿新 • • 發佈:2018-12-18
tf.app.flags
tf.app.flags.DEFINE_XXX()
用來定義引數項:
import tensorflow as tf
tf.app.flags.DEFINE_float(
'flag_float', 0.01, 'input a float')
tf.app.flags.DEFINE_string(
'flag_string', 'yes', 'input a string')
tf.app.flags.FLAGS
tf.app.flags.FLAGS
可以用來返回引數:
print(FLAGS.flag_float)
print(FLAGS.flag_string)
tf.app.run()
常見如下主函式:
if __name__ == '__main__':
tf.app.run()
主函式中的tf.app.run()
會呼叫main函式,main裡面先傳遞引數(因此必須在main函式中設定一個引數位置),然後再是執行。如果要更換main名字,只需要在tf.app.run()中傳入一個指定的函式名即可,該函式名內同樣要設定引數。
修改FLAGS引數:
在命令列中輸入:
python xxx.py --flag_float xxx --flag_string xxx
綜上
tf.app.flags()主要用來定義引數項。
而tf.app.run()用來執行帶引數項的函式。