pytorch 转化到caffe

1. 没有转换成功,因为caffe不支持gather layer. 而pytorch yolov3层,非常容易产生gather, 就算是tensor.view(), slice 都会产生gather运算符。

2. tensorrt也是最近才支持gather运算符。何况已经很久不更新的caffe.

mpi caffe的可以支持gather 层, https://github.com/craftGBD/caffe-GBD/blob/master/src/caffe/layers/gather_layer.cpp

但是目标是要转换到nnie,里面应该没有mpi。

3. 什么是merge batchnorm?

把batchnorm 合并到前面的卷积层里面,从而去掉了bn,scale层,起到了加速效果。为什么能合并呢,卷积不是全连接,wx+b, 但是输入通过im2col后,卷积操作就化成了矩阵的wx+b。

y = wx+b

z = gamma * (y-batch_mean)/sqrt(var) + beta

从而:z = gamma*w/sqrt(var) * x + (gamma * b / sqrt(var) + beta)

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值