yolov5_prune
yolov5_prune copied to clipboard
yolo.py中parse_pruned_model 获取每层剩余通道数疑问
您好,有一个问题如图中所示:之前求得的mask_bn中key是类似这种model.0.conv.1 , 但是这里mask_bn使用的key是named_m_bn 带有.bn关键字,运行时会报错显示mask_bn没有这个关键字。请问是之前求得mask_bn时 修改了相应的key值嘛?