博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tf.contrib.slim add_arg_scope
阅读量:5066 次
发布时间:2019-06-12

本文共 1337 字,大约阅读时间需要 4 分钟。

  上一篇文章中我们介绍了arg_scope函数,它在每一层嵌套中update当前字典中参数形成新的字典,并入栈。那么这些参数是怎么作用到代码块中的函数的呢?比如说如下情况:

with slim.arg_scope(      [slim.conv2d, slim.separable_conv2d],      weights_initializer=tf.truncated_normal_initializer(          stddev=weights_initializer_stddev),      activation_fn=activation_fn,      normalizer_fn=slim.batch_norm if use_batch_norm else None):    with slim.arg_scope([slim.batch_norm], **batch_norm_params):    slim.conv2d(                features,                num_classes,                kernel_size=kernel_size,                rate=rate,                activation_fn=None,                normalizer_fn=None,                scope=scope))

  原理就是使用add_arg_scope函数装饰op,那么op就能查找栈中字典的参数并使用他们,主要和上篇文章很类似。

def func_with_args(*args, **kwargs):   current_scope = current_arg_scope()   current_args = kwargs   key_func = arg_scope_func_key(func)   if key_func in current_scope:     current_args = current_scope[key_func].copy()     current_args.update(kwargs)   return func(*args, **current_args)

  代码逻辑就是先得到当前字典current_arg_scope,此时为{‘conv2d: kargs, 'separable_2d':kargs, 'batch_norm': batch_norm_params}(这里kargs是我偷懒没把代码中initializer等誊写下来),current_args是代码块中参数,这里是features,num_classes等,key_func是’conv2d‘,循环就是如果在字典中有与之相关的参数,则把参数用到函数中。

结语

  写的好像有些简单,下次有灵感再好好改一下。      最后编辑于11:44:51 2018-07-30

转载于:https://www.cnblogs.com/zzy-tf/p/9389792.html

你可能感兴趣的文章
AngularJS学习篇(一)
查看>>
关于Xshell无法连接centos6.4的问题
查看>>
css3动画——基本准则
查看>>
输入月份和日期,得出是今年第几天
查看>>
pig自定义UDF
查看>>
Kubernetes 运维学习笔记
查看>>
spring security 11种过滤器介绍
查看>>
代码实现导航栏分割线
查看>>
大数据学习系列(8)-- WordCount+Block+Split+Shuffle+Map+Reduce技术详解
查看>>
【AS3代码】播放FLV视频流的三步骤!
查看>>
枚举的使用
查看>>
luogu4849 寻找宝藏 (cdq分治+dp)
查看>>
日志框架--(一)基础篇
查看>>
关于源程序到可运行程序的过程
查看>>
转载:mysql数据库密码忘记找回方法
查看>>
scratch少儿编程第一季——06、人在江湖混,没有背景怎么行。
查看>>
【贪心+DFS】D. Field expansion
查看>>
C# Async与Await的使用
查看>>
Mysql性能调优
查看>>
iOS基础-UIKit框架-多控制器管理-实例:qq界面框架
查看>>