search

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

prune_yolov5s.py文件第152行代码的疑问 #112

Open bingo789 opened 1 year ago

bingo789 commented 1 year ago

pruned_model = prune_model_keep_size2(model, CBL_idx, CBL_idx, CBLidx2mask)的第二个参数是不是应该改成 prune_idx ？

函数定义是 def prune_model_keep_size2(model, prune_idx, CBL_idx, CBLidx2mask): ......

ZJU-lishuang commented 1 year ago

https://github.com/tanluren/yolov3-channel-and-layer-pruning/blob/master/prune.py#L142