flags命令行传递参数

flags

import tensorflow as tf
Flags = tf.app.flags
#第一个是参数名称,第二个参数是默认值,第三个是参数描述
tf.app.flags.DEFINE_string('str_name', default, 'descrip1')
tf.app.flags.DEFINE_integer('int_name', default, 'descript2')
tf.app.flags.DEFINE_boolean('bool_name', default, 'descript3') 
tf.app.flags.DEFINE_float('float_name', default, 'descript3')
FLAGS = Flags.FLAGS
调用 FLAGS.name

bool类型传入1认为是true其余均是false
argparse

import argparse
parser = argparse.ArgumentParser(description='manual to this script')
parser.add_argument('--gpus', type=str, default = None)
args = parser.parse_args()
调用 args.name

type:bool, int, str, float
bool类型的解析比较特殊,传入任何值都会被解析成True
sys.argv
sys.argv[0] 文件名
sys.argv[1] 参数1

猜你喜欢

转载自blog.csdn.net/fronde_wu/article/details/82288823
今日推荐