Open HouBiaoLiu opened 3 years ago
我稀疏训练用的多卡分布式训练的, 在运行脚本slim_prune_yolov5s对模型进行裁剪时,报错在utils/prune_utils.py中gather_bn_weights函数 是SyncBatchNorm
报错显示的是维度数目不对,项目未使用SyncBatchNorm
我也遇到这个错误,请问一下您是怎解决呢?@HouBiaoLiu
我稀疏训练用的多卡分布式训练的, 在运行脚本slim_prune_yolov5s对模型进行裁剪时,报错在utils/prune_utils.py中gather_bn_weights函数
是SyncBatchNorm