arg_scope的兩個例子

“”"Contains the arg_scope used for scoping layers arguments.
Allows one to define models much more compactly by eliminating boilerplate
code. This is accomplished through the use of argument scoping (arg_scope).
Example of how to use tf.contrib.framework.arg_scope:

from third_party.tensorflow.contrib.layers.python import layers
arg_scope = tf.contrib.framework.arg_scope
with arg_scope([layers.conv2d], padding='SAME',
               initializer=layers.variance_scaling_initializer(),
               regularizer=layers.l2_regularizer(0.05)):
  net = layers.conv2d(inputs, 64, [11, 11], 4, padding='VALID', scope='conv1')
  net = layers.conv2d(net, 256, [5, 5], scope='conv2')

The first call to conv2d will behave as follows:
layers.conv2d(inputs, 64, [11, 11], 4, padding=‘VALID’,
initializer=layers.variance_scaling_initializer(),
regularizer=layers.l2_regularizer(0.05), scope=‘conv1’)
The second call to conv2d will also use the arg_scope’s default for padding:
layers.conv2d(inputs, 256, [5, 5], padding=‘SAME’,
initializer=layers.variance_scaling_initializer(),
regularizer=layers.l2_regularizer(0.05), scope=‘conv2’)
Example of how to reuse an arg_scope:

with arg_scope([layers.conv2d], padding='SAME',
               initializer=layers.variance_scaling_initializer(),
               regularizer=layers.l2_regularizer(0.05)) as sc:
  net = layers.conv2d(net, 256, [5, 5], scope='conv1')
  ....
with arg_scope(sc):
  net = layers.conv2d(net, 256, [5, 5], scope='conv2')

Example of how to use tf.contrib.framework.add_arg_scope to enable your
function to be called within an arg_scope later:
@tf.contrib.framework.add_arg_scope
def conv2d(*args, **kwargs)
“”"

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章