Efficient-AI-Backbones icon indicating copy to clipboard operation
Efficient-AI-Backbones copied to clipboard

Published 20 hours ago • verified publisher icon huawei-noah

g-ghost预训练权重

Open 12moli opened this issue 2 years ago • 8 comments

请问g-ghostnet有没有pytorch的预训练权重呢？

12moli avatar Oct 19 '22 11:10 12moli

我们会尽快release

iamhankai avatar Oct 29 '22 01:10 iamhankai

谢谢您的回复！我把gghost嵌入了yolo，但是由于gghost源码中有用到了lambda创建匿名函数，但torch.save时报错 Can‘t pickle local object ‘_initialize.＜locals＞.ResBasicBlock.init.＜locals>.lambda，貌似无法解析lambda表达式，请问您有解决办法吗？

12moli avatar Oct 29 '22 09:10 12moli

存储模型权重，别存整个模型，torch.save(model.state_dict(), 'save.pt')

iamhankai avatar Oct 31 '22 06:10 iamhankai

谢谢您的回复，在修改期间我又出现了其他的问题。。我嵌入的是yolov5-v6.0，准备把所有的torch.save都修改成只存权重，但在这个保存checkpoint位置好像没办法修改，如图： image ；还有就是嵌入之后会导致后面无法生成混淆矩阵图和PR曲线图等图像，错误：“Using categorical units to plot a list of strings that are all parsable as floats or dates. If these strings should be plotted as numbers, cast to the appropriate data type before plotting.”，请问您有什么解决的办法吗？

12moli avatar Nov 01 '22 08:11 12moli

你这个问题超纲了

iamhankai avatar Nov 01 '22 11:11 iamhankai

好吧，谢谢您！

12moli avatar Nov 01 '22 12:11 12moli

最新g-ghost代码和预训练权重：https://github.com/huawei-noah/Efficient-AI-Backbones/tree/master/g_ghost_pytorch

iamhankai avatar Nov 18 '22 02:11 iamhankai

好吧，谢谢您！

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

songjiahao-wq avatar Nov 24 '22 08:11 songjiahao-wq