tf.app.flags函式使用
阿新 • • 發佈:2018-11-25
用於支援接受命令列傳遞引數,相當於接受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