search

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.07k stars 708 forks source link

加载g-ghost预训练权重 #167

Open zeqianl opened 1 year ago

zeqianl commented 1 year ago

image

在上述代码加载g-ghost预训练权重时报如下错误： image

然后将代码修改为： image 无报错，请问这样修改是对的不？有无问题？

iamhankai commented 1 year ago

是的，把'module.'去掉

zeqianl commented 1 year ago

是的，把'module.'去掉 去掉就又报错了

iamhankai commented 1 year ago

那你用nn.DataParallel包一下model