TensorFlow插曲之 tf.app.run()

原创于 2017-07-28 16:35:48 发布 · 6.4k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#python #tf.app.run

python 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了TensorFlow中tf.app.run()函数的功能及使用方法，通过示例代码展示了如何利用该函数解析命令行参数并执行主函数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

学习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)