PROSAGA码农传奇-深度学习-冻结VGG19网络中的权重，用于Caffe中的转移学习

冻结VGG19网络中的权重，用于Caffe中的转移学习

作者: Jacob
发布时间: 2024-08-17 07:36:18 (9天前)
转自：

<div class =“post-text”itemprop =“text”>
  <P>
    要在Caffe中使用batchnorm正确冻结卷积层，您应该：
  </p>
  <UL>
    <LI>
      对于卷积层：设置
       <code>
        lr_mult
      </code>
       参数为0
    </LI>
    <LI>
      对于batchnorm图层：设置
       <code>
        lr_mult
      </code>
       参数为0，设定
       <code>
        use_global_stats
      </code>
       至
       <code>
        true
      </code>
      ：
    </LI>
  </UL>
   <pre>
    <code>
      layer {
  name: "bn1"
  type: "BatchNorm"
  bottom: "pool1"
  top: "bn1"
  batch_norm_param {
    use_global_stats: true
  }
  param {
    lr_mult: 0
  }
  param {
    lr_mult: 0
  }
  param {
    lr_mult: 0
  }
}

</code>
  </pre>
  <BLOCKQUOTE>
    <P>
      批量标准化层是否被冻结这一事实是否意味着卷积层也被冻结了？
    </p>
  </BLOCKQUOTE>
  <P>
    当然不是。但是，通过使用
     <code>
      propagate_down
    </code>
     param你可以达到这个效果：
    <a href="https://stackoverflow.com/questions/40610723/how-do-i-to-prevent-backward-computation-in-specific-layers-in-caffe">
      如何防止caffe中特定层的反向计算
    </A>
    。
  </p>
</DIV>