flags命令行传递参数

最新推荐文章于 2023-01-14 23:18:35 发布

原创最新推荐文章于 2023-01-14 23:18:35 发布 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

tensorflow 专栏收录该内容

4 篇文章

订阅专栏

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