剪枝问题

[lsyss-lsyss]

剪枝完所有指标变成0

[xinxin342]

剪枝的时候你遇到KeyError: 'model.0.act.depth_bn'错误吗？ File "D:\ProgramData\Anaconda3\envs\yolov5\yolov5prune-6.0\prune.py", line 537, in run_prune out_idx = np.squeeze(np.argwhere(np.asarray(maskbndict[layername[:-4] + "bn"].cpu().numpy()))) KeyError: 'model.0.act.depth_bn'

[YHLymj]

剪枝完所有指标变成0

我也遇到了这个问题，请问您解决这个问题了吗？

[1781686573]

请问你解决了吗？我也有这个问题