要在Caffe中使用batchnorm正确冻结卷积层,您应该:
lr_mult
use_global_stats
true
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 } }
批量标准化层是否被冻结这一事实是否意味着卷积层也被冻结了?
当然不是。但是,通过使用 propagate_down param你可以达到这个效果: 如何防止caffe中特定层的反向计算 。
propagate_down