tf.app.flags函数使用

用于支持接受命令行传递参数,相当于接受argv

flags = tf.app.flags
FLAGS = flags.FLAGS

怎么定义参数:#第一个是参数名称,第二个参数是默认值,第三个是参数描述

1.定义常量:tf.app.flags.DEFINE_integer("", ,"")

flags.DEFINE_integer("cnn_size", 3, "Size of CNN filters")
FLAGS.cnn_size

2.定义浮点数:tf.app.flags.DEFINE_float("", ,"")

flags.DEFINE_float("beta2", 0.1, "Beta2 form adam optimizer")
FLAGS.beta2

3.定义字符串:tf.app.flags.DEFINE_string("", ,"")

flags.DEFINE_string("model_name", "", "model name for save files and tensorboard log")
FLAGS.model_name

4.定义布尔量:tf.app.flags.DEFINE_boolean("", ,"")

flags.DEFINE_boolean("residual", False, "Using residual net")
FLAGS.residual

猜你喜欢

转载自blog.csdn.net/weixin_41855385/article/details/84397063