查看SSD实现代码时学习的API

1. tf.app.run()和tf.app.flags.DEFINE_float()、tf.app.flags.DEFINE_integer()、tf.app.flags.DEFINE_string()

tf.app.run()   此函数一般不传入参数,直接调用

tf.app.flags.DEFINE_float()   传入所需要的参数名称,参数值,同时还可已传入参数说明,其中参数值的类型与所使用的函数类型一致,即函数名称中DEFINE后所跟的类型,参数名称和参数说明需用字符串类型('illustration')。

在代码中使用tf.app.run()处理flag解析,然后执行主函数,在执行main函数之前首先进行flags的解析,也就是说Tensorflow通过设置flags来传递tf.app.run()所需的参数,可以直接在程序运行前初始化flags,也可以再运行程序的时候设置命令行参数来达到传参的目的。

下面是一段实际使用的代码

import tensorflow as tf
tf.app.flags.DEFINE_float('learning_rate', 0.01, 'Intital learning rate')
tf.app.flags.DEFINE_integer('batch_size', 50, 'num of per SGD ')
tf.app.flags.DEFINE_string('data_path', './data/', 'data save path')
FLAGS = tf.app.flags.flags()
def main():
  learning rate = FLAGS.learning_rate
  batch size = FLAGS.batch_size
  image path = FLAGS.data_path
if _name_ == 'main':
  tf.app.run()

此段代码在程序运行前初始化flags,然后使用tf.app.run()处理flag解析。

2.tf.logging.set_verbosity() 为将要记录的日志设置入口

tf.logging.set_verbosity(tf.logging.DEBUG)


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值