yolov8剪枝:剪枝后通道数不匹配

剪枝遇到问题:

RuntimeError: Given groups=1, weight of size [64, 64, 3, 3], expected input[1, 28, 80, 80] to have 64 channels, but got 28 channels instead
错误如上所述。

解决办法:

 这里的threshold×0.8即可。

其他系数没试过,你也可以试试。

分析:

只是个人理解:

随着factor的提升,input channels也增高,说明剪枝剪多了

随着threshold的降低,input channels降低。

引用:Jetson nano部署剪枝YOLOv8_爱听歌的周童鞋的博客-CSDN博客

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值