tf.flags用法

from absl import flags
from absl import app
 
FLAGS = flags.FLAGS
 
#1、第一个是参数名称，第二个参数是默认值，第三个是参数描述
flags.DEFINE_string('model', None, 'model to run')
 
def main(argv):
    print('Hello World')
    print('selected model', FLAGS.model)
 
if __name__ == '__main__':
    app.run(main)  #2、执行main函数

输出

$ python flag_try.py

Hello world
selected model None



$ python flag_try.py --model 'My model'
Hello world
selected model My model