Question

我能够从检查点文件恢复模型并提取权重，偏差和batch_norm图层的参数。但是对于多个检查点文件（初始模型等），我找不到BN层的缩放/伽马因子。

例如，在public inceptionV3检查点，我可以找到： InceptionV3/Mixed_5d/Branch_2/Conv2d_0a_1x1/BatchNorm/moving_mean (DT_FLOAT) [64] InceptionV3/Mixed_5d/Branch_2/Conv2d_0a_1x1/BatchNorm/moving_variance (DT_FLOAT) [64] InceptionV3/Mixed_5d/Branch_2/Conv2d_0a_1x1/BatchNorm/beta (DT_FLOAT) [64]

但是，没有任何内容，例如InceptionV3/Mixed_5d/Branch_2/Conv2d_0a_1x1/BatchNorm/gamma。

如何获取伽玛值或默认情况下重新调整为1？

非常感谢！

Answer 1

因此大多数网络都使用SLIM的batch_norm，默认情况下没有缩放/ gamma参数。

比例：如果为True，则乘以gamma。如果为False，则gamma为不曾用过。当下一层是线性的（例如nn.relu）时，这可以是禁用，因为缩放可以由下一层完成。

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/layers/python/layers/layers.py#L365-L386

Answer 2

我对slim库的预训练inceptionV2也有同样的问题。

首先我使用这个arg_scope，我遇到了这个问题：

def _batch_norm_arg_scope(list_ops,
                          use_batch_norm=True,
                          batch_norm_decay=0.9997,
                          batch_norm_epsilon=0.001,
                          batch_norm_scale=False,
                          train_batch_norm=False):
    """Slim arg scope for InceptionV2 batch norm."""
    if use_batch_norm:
        batch_norm_params = {
            'is_training': train_batch_norm,
            'scale': batch_norm_scale,
            'decay': batch_norm_decay,
            'epsilon': batch_norm_epsilon
        }
        normalizer_fn = slim.batch_norm
    else:
        normalizer_fn = None
        batch_norm_params = None

    return slim.arg_scope(list_ops,
                          normalizer_fn=normalizer_fn,
                          normalizer_params=batch_norm_params)

我在slim库中使用arg_scope解决了。

with slim.arg_scope(inception_v2.inception_v2_arg_scope()):

就是：

def inception_arg_scope(weight_decay=0.00004,
                        use_batch_norm=True,
                        batch_norm_decay=0.9997,
                        batch_norm_epsilon=0.001,
                        activation_fn=tf.nn.relu):
  """Defines the default arg scope for inception models.

  Args:
    weight_decay: The weight decay to use for regularizing the model.
    use_batch_norm: "If `True`, batch_norm is applied after each convolution.
    batch_norm_decay: Decay for batch norm moving average.
    batch_norm_epsilon: Small float added to variance to avoid dividing by zero
      in batch norm.
    activation_fn: Activation function for conv2d.

  Returns:
    An `arg_scope` to use for the inception models.
  """
  batch_norm_params = {
      # Decay for the moving averages.
      'decay': batch_norm_decay,
      # epsilon to prevent 0s in variance.
      'epsilon': batch_norm_epsilon,
      # collection containing update_ops.
      'updates_collections': tf.GraphKeys.UPDATE_OPS,
      # use fused batch norm if possible.
      'fused': None,
  }
  if use_batch_norm:
    normalizer_fn = slim.batch_norm
    normalizer_params = batch_norm_params
  else:
    normalizer_fn = None
    normalizer_params = {}
  # Set weight_decay for weights in Conv and FC layers.
  with slim.arg_scope([slim.conv2d, slim.fully_connected],
                      weights_regularizer=slim.l2_regularizer(weight_decay)):
    with slim.arg_scope(
        [slim.conv2d],
        weights_initializer=slim.variance_scaling_initializer(),
        activation_fn=activation_fn,
        normalizer_fn=normalizer_fn,
        normalizer_params=normalizer_params) as sc:
      return sc

恢复Tensorflow模型：在检查点文件中找不到batch_norm图层的gamma / scale

2 个答案: