HankYe / PAGCP

PAGCP for the compression of YOLOv5
GNU General Public License v3.0
111 stars 11 forks source link

修改网络模型后如何修改相应文件 #9

Open Anoue opened 1 year ago

Anoue commented 1 year ago

就是我将Conv换成了GhostConv之后 还有C3换成了C3Ghost 那我该如何修改文件 来完成减枝呢 一直出错误

HankYe commented 1 year ago

就是我将Conv换成了GhostConv之后 还有C3换成了C3Ghost 那我该如何修改文件 来完成减枝呢 一直出错误

您好,感谢您对工作的关注。由于这段时间任务比较繁重,我们会在晚些时候对修改算子后的模型进行压缩测试,届时会更新comment。如果方便您也可以提供一下报错信息,以便我们后续测试的时候如果遇到可以予以解决。如果您实验比较急的话也可以试试看最新版的torch_pruning,这个可能能解决您目前的问题。最后,再次感谢您的关注和理解!

harryc7 commented 7 months ago

就是我将Conv换成了GhostConv之后 还有C3换成了C3Ghost 那我该如何修改文件 来完成减枝呢 一直出错误

请问你解决了吗