ZJU-lishuang / yolov5_prune

yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

运行稀疏训练时程序报错无法更新bn系数 #24

Open wangjinpeng9521 opened 3 years ago

wangjinpeng9521 commented 3 years ago

在稀疏训练后gather bn weight的时候,会提示model中缺少modellist从而报错,请问作者大哥这个问题该如何解决呢,另外入坑加速方向的小弟求大哥联系方式QAQ

wangjinpeng9521 commented 3 years ago

我的vx是283154230

ZJU-lishuang commented 3 years ago

如果是开源数据集,请提供下载路径,用来定位修复bug