gather_bn_weights函数不支持SyncBatchNorm

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5

Apache License 2.0

556 stars 136 forks source link

Open HouBiaoLiu opened 3 years ago

HouBiaoLiu commented 3 years ago

我稀疏训练用的多卡分布式训练的, 在运行脚本slim_prune_yolov5s对模型进行裁剪时，报错在utils/prune_utils.py中gather_bn_weights函数是SyncBatchNorm

ZJU-lishuang commented 3 years ago

报错显示的是维度数目不对，项目未使用SyncBatchNorm

zxsitu commented 3 years ago

我也遇到这个错误，请问一下您是怎解决呢？@HouBiaoLiu