TensorFlow里的tf.flags 使用方法说明

import tensorflow as tf
#定义flags变量
flags = tf.flags
FLAGS = flags.FLAGS

#定义变量里的值
flags.DEFINE_string("path","/data","data's path")
flags.DEFINE_double("money",20.0,"the value of money")
flags.DEFINE_integer("step",20,"training step")
flags.DEFINE_bool("Train",True,"train or not")

#使用变量里的值
FLAGS.path
FLAGS.step

#必须初始化的值,常放在main函数中
flags.mark_flag_as_required("path")
  flags.mark_flag_as_required("step")
  flags.mark_flag_as_required("train")

 

上一篇:自顶向下redis4.0(3)命令与dict


下一篇:TF的FLAGS字典获取