tensorflow学习笔记(一):命令行参数

来源:互联网 发布:现货交易 网络销售 编辑:程序博客网 时间:2024/05/22 01:31

全局环境下编写代码

import tensorflow as tfflags = tf.flags #flags是一个文件:flags.py,用于处理命令行参数的解析工作logging = tf.logging#调用flags内部的DEFINE_string函数来制定解析规则flags.DEFINE_string("para_name_1","default_val", "description")flags.DEFINE_bool("para_name_2","default_val", "description")#FLAGS是一个对象,保存了解析后的命令行参数FLAGS = flags.FLAGSdef main(_):    FLAGS.para_name #调用命令行输入的参数if __name__ = "__main__": #使用这种方式保证了,如果此文件被其它文件import的时候,不会执行main中的代码    tf.app.run() #解析命令行参数,调用main函数 main(sys.argv)

调用方法:

~/ python script.py --para_name_1=name --para_name_2=name2# 不传的话,会使用默认值
2 0
原创粉丝点击