[tf]中的tf.flags

2018-12-20  本文已影响3人  VanJordan

用来读取命令行参数的,是一个包装后的argparser,只要引入了

import tensorflow as tf
flags = tf.app.flags  
FLAGS = flags.FLAGS  
flags.DEFINE_float('learning_rate', 0.01, 'Initial learning rate.')  
flags.DEFINE_integer('epoch_number', None, 'Number of epochs to run trainer.')

简化形式来写

flags = tf.app.flags
F = flags.FLAGS
fint = flags.DEFINE_integer
fstr = flags.DEFINE_string
fbool = flags.DEFINE_bool
ffloat = flags.DEFINE_float

小实验

上一篇 下一篇

猜你喜欢

热点阅读