本文共 695 字,大约阅读时间需要 2 分钟。
学习TensorFlow程序代码时,很多时候可以看到tf.app.run(),下面讲一下该函数的作用和用法。
run(main=None, argv=None) Runs the program with an optional 'main' function and 'argv' list.
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/