search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

更改网络结构后该如何进行剪枝 #56

Closed Horizon-Max closed 2 years ago

Horizon-Max commented 3 years ago

博主，您好，请问如果对网络结构进行了更改，应该如何修改代码来进行剪枝呢？谢谢

ZJU-lishuang commented 3 years ago

主要修改cfg中的网络结构，根据修改后网络结构微调代码中报错的地方。