剪枝遇到问题:

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博客

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐