1. 首页
  2. 人工智能
  3. 深度学习
  4. caffe加速:合并BatchNorm层和Scale层到Convolution层

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

上传者: 2019-07-25 20:02:07上传 PY文件 8.62KB 热度 26次
Convolution+BatchNorm+Scale+Relu的组合模块在卷积后进行归一化,可以加速训练收敛。但在推理时BatchNorm非常耗时,可以将训练时学习到的BatchNorm+Scale的线性变换参数融合到卷积层,替换原来的Convolution层中weights和bias,实现在不影响准确度的前提下加速预测时间。
下载地址
用户评论
码姐姐匿名网友 2019-07-25 20:02:08

可以 使用 效果不会影响太多,我做的是分割任务

码姐姐匿名网友 2019-07-25 20:02:08

一般般,影响效果