文章目录

• 源码
• 示例代码
• 打印用法
• 指定参数

使用的tensorflow 版本

In [1]: import tensorflow as tf

In [2]: print(tf.__version__)
1.8.0

源码

@tf_export('app.run')
def run(main=None, argv=None):
"""Runs the program with an optional 'main' function and 'argv' list."""

# Define help flags.
_define_help_flags()

# Parse known flags.
argv = flags.FLAGS(_sys.argv if argv is None else argv, known_only=True)

main = main or _sys.modules['__main__'].main

# Call the main function, passing through any arguments
# to the final program.
_sys.exit(main(argv))

示例代码

import tensorflow as tf

flags = tf.app.flags

flags.DEFINE_string('string', 'train', 'This is a string')
flags.DEFINE_float('learning_rate', 0.001, 'This is the rate in training')
flags.DEFINE_boolean('flag', False, 'This is a flag')

FLAGS = flags.FLAGS

def main(argv):
print('string: ', FLAGS.string)
print('learning rate: ', FLAGS.learning_rate)
print('flag:', FLAGS.flag)

if __name__ == '__main__':
tf.app.run()

打印用法

> python3 test.py --help

>> 结果

USAGE: test.py [flags]
flags:

test.py:
--[no]flag: This is a flag
(default: 'false')
--learning_rate: This is the rate in training
(default: '0.001')
(a number)
--string: This is a string
(default: 'train')

Try --helpfull to get a list of all flags.

指定参数

python3 test.py --string 'test' --learning_rate 0.2 --flag False
> string:  test
> learning rate:  0.2
> flag: True

为什么flag 没有生效呢？看help就可以得到答案：
–[no]flag: This is a flag
(default: ‘false’)
原来boolean，设置是用前置[no]来实现

python3 test.py --string 'test' --learning_rate 0.2 --noflag
> string:  test
> learning rate:  0.2
> flag: False

• 点赞
• 收藏
• 分享
• 文章举报