caffe加速:合并BatchNorm层和Scale层到Convolution层

时间:2021-09-30 15:26:26
【文件属性】:

文件名称:caffe加速:合并BatchNorm层和Scale层到Convolution层

文件大小:9KB

文件格式:PY

更新时间:2021-09-30 15:26:26

caffe加速 移除dropout 融合BN层

Convolution+BatchNorm+Scale+Relu的组合模块在卷积后进行归一化,可以加速训练收敛。但在推理时BatchNorm非常耗时,可以将训练时学习到的BatchNorm+Scale的线性变换参数融合到卷积层,替换原来的Convolution层中weights和bias,实现在不影响准确度的前提下加速预测时间。


网友评论

  • 不要积分,先给楼主32个
  • 可以 使用 效果不会影响太多,我做的是分割任务
  • 一般般,影响效果