1. 程式人生 > >【Tensorflow】tf.app.run() 與 tf.app.flags()

【Tensorflow】tf.app.run() 與 tf.app.flags()

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()用來執行帶引數項的函式。