码迷,mamicode.com
首页 > 移动开发 > 详细

tensorflow学习之——tf.app.flags.DEFINE_XXXX() 使用flags定义命令行参数

时间:2020-03-24 12:42:12      阅读:82      评论:0      收藏:0      [点我收藏+]

标签:def   points   color   class   布尔   框架   网络   name   xxxx   

和C/C++编写main函数中的argv一样,tf框架下也封装了tf.app.flags.DEFINE_XXXX()函数用于定义参数,便于命令行形式传递参数。常见的函数形式如下:

flags.DEFINE_float(参数1,参数2,参数3)  
flags.DEFINE_integer(参数1,参数2,参数3)  
flags.DEFINE_string(参数1,参数2,参数3) 
flags.DEFINE_boolean(参数1,参数2,参数3) 

参数1:定义的参数名称;

参数2:参数默认值;

参数3:对参数的描述;

tf.app.flags.DEFINE_boolean(train, FALSE,train or test) #定义布尔类型的值,判断网络是训练还是测试
tf.app.flags.DEFINE_string(check_points_dir, ./model/,check_points dir) #模型保存路径 
tf.app.flags.DEFINE_integer(epoch, 50,train epoch size) #定义的epoch大小,默认为50
FLAGS = tf.app.flags.FLAGS
def main(XXXX): 
    XXXX
   # 在调用参数时,一般如下使用;FLAGS.参数
   FLAGS.epoch
if __name__ == __main__: XXXX #执行main

 

tensorflow学习之——tf.app.flags.DEFINE_XXXX() 使用flags定义命令行参数

标签:def   points   color   class   布尔   框架   网络   name   xxxx   

原文地址:https://www.cnblogs.com/fclbky/p/12557885.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!