博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
TensorFlow插曲之 tf.app.run()
阅读量:4147 次
发布时间:2019-05-25

本文共 695 字,大约阅读时间需要 2 分钟。

学习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 argparseimport sysFLAGS=Nonedef 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)

转载地址:http://omjti.baihongyu.com/

你可能感兴趣的文章
日志框架学习2
查看>>
SVN-无法查看log,提示Want to go offline,时间显示1970问题,error主要是 url中 有一层的中文进行了2次encode
查看>>
NGINX
查看>>
Qt文件夹选择对话框
查看>>
1062 Talent and Virtue (25 分)
查看>>
1061 Dating (20 分)
查看>>
1060 Are They Equal (25 分)
查看>>
83. Remove Duplicates from Sorted List(easy)
查看>>
88. Merge Sorted Array(easy)
查看>>
leetcode刷题191 位1的个数 Number of 1 Bits(简单) Python Java
查看>>
leetcode刷题198 打家劫舍 House Robber(简单) Python Java
查看>>
NG深度学习第一门课作业2 通过一个隐藏层的神经网络来做平面数据的分类
查看>>
leetcode刷题234 回文链表 Palindrome Linked List(简单) Python Java
查看>>
NG深度学习第二门课作业1-1 深度学习的实践
查看>>
Ubuntu下安装Qt
查看>>
Qt札记
查看>>
我的vimrc和gvimrc配置
查看>>
hdu 4280
查看>>
禁止使用类的copy构造函数和赋值操作符
查看>>
C++学习路线
查看>>