tf.app.flags说明
很多时候在运行python代码的时候我们需要从外部定义参数,从而避免每次都需要改动代码,tf就提供了tf.app.flags功能。
import tensorflow as tf # 学习使用 tf.app.flags 使用,全局变量 # tf.app.flags.DEFINE_string("param_name", "default_val", "description") tf.app.flags.DEFINE_boolean("clean", False, "clean train folder") tf.app.flags.DEFINE_integer("batch_size", 8, "batch size") tf.app.flags.DEFINE_float("dropout", 0.5, "Dropout rate") tf.app.flags.DEFINE_string("log_dir", "./logs", " the log dir") FLAGS=tf.app.flags.FLAGS print(FLAGS.dropout)