New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

zeqianl opened this issue Dec 8, 2022 · 3 comments

zeqianl commented Dec 8, 2022

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

然后将代码修改为：

无报错，请问这样修改是对的不？有无问题？

Member

iamhankai commented Dec 8, 2022

是的，把'module.'去掉

Author

zeqianl commented Dec 8, 2022

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

Member

iamhankai commented Dec 9, 2022

那你用nn.DataParallel包一下model

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment