TensorFlow插曲之 tf.app.run()

勤劳的凌菲

6446人浏览 · 2017-07-28 16:35:48

勤劳的凌菲 · 2017-07-28 16:35:48 发布

学习TensorFlow程序代码时，很多时候可以看到tf.app.run()，下面讲一下该函数的作用和用法。

1 函数原型

run(main=None, argv=None)
    Runs the program with an optional 'main' function and 'argv' list.

2 作用

通常用此函数解析命令行后，运行main函数

3 例子

import tensorflow as tf
#导入命令行解析模块
import argparse
import sys

FLAGS=None

def main(_):
    print(sys.argv[0])


if __name__=="__main__": #用这种方式保证了，如果此文件被其他文件import的时候，不会执行main中的代码
    #创建对象
    parse=argparse.ArgumentParser()
    #增加命令行
    parse.add_argument('--dataDir',type=str,default='\\tmp\\tensorflow\\mnist\\inputData',
                    help='Directory for string input data')
    FLAGS, unparsed=parse.parse_known_args()
    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed) #解析命令行参数，调用main函数 main(sys.argv)