1. 程式人生 > >tf.app.flags函式使用

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