前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >tensorflow: arg_scope

tensorflow: arg_scope

作者头像
ke1th
发布2018-01-02 11:29:11
1.1K0
发布2018-01-02 11:29:11
举报

arg_scope

代码语言:javascript
复制
tf.contrib.framework.arg_scope(list_ops_or_scope, **kwargs)
#或者
tf.contrib.slim.arg_scope(list_ops_or_scope, **kwargs)

# 为给定的 list_ops_or_scope 存储默认的参数

示例:

代码语言:javascript
复制
with slim.arg_scope([slim.conv2d, slim.fully_connected],
                        weights_initializer=tf.truncated_normal_initializer(stddev=0.1),
                        weights_regularizer=slim.l2_regularizer(weight_decay),
                        normalizer_fn=slim.batch_norm,
                        normalizer_params=batch_norm_params):

就这样给slim.conv2dslim.fully_connected准备了默认参数。

如何给自定义的函数也附上这种功能

代码语言:javascript
复制
from tensorflow.contrib import framework
from tensorflow.contrib.framework.python.ops.arg_scope import add_arg_scope

@add_arg_scope
def haha(name, age):
    print(name, age)

with framework.arg_scope([haha], age = 15):
    haha("keith")
# 输出
# keith 15
代码语言:javascript
复制
with slim.arg_scope(...) as argScope:
    ...
with slim.arg_scope(argScope):
    ...
# argScope 是一个字典。这个字典可以继续使用,下面的arg_scope配置和上面的是一样的。
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • arg_scope
    • 如何给自定义的函数也附上这种功能
    领券
    问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档