Skip to content
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.

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

在下游任务做半监督学习,如何加载coco预训练模型? #16

Open
Stephen-Hao opened this issue Nov 8, 2022 · 1 comment

Comments

@Stephen-Hao
Copy link

Stephen-Hao commented Nov 8, 2022

image
使用mmdetection默认的方式,在config文件里加入load_from的coco预训练路径会报模型参数不匹配的错误,怎么才能正确将coco预训练模型参数加载到student和teacher model里呢

@ligang-cs
Copy link
Owner

Sorry for late reply. You need to modify the code in Line 206 of train.py. Please use:
runner.load_checkpoint(cfg.load_from, revise_keys=[(r'^', 'student.')])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants