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

版权声明：转载请声明转自Juanlyjack https://blog.csdn.net/m0_38088359/article/details/83546515

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()用来运行带参数项的函数。