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

建议backbone部分都提供加载预训练权值的接口? #7

Open
dsp6414 opened this issue Nov 7, 2020 · 1 comment
Open

Comments

@dsp6414
Copy link

dsp6414 commented Nov 7, 2020

感觉好像只有resnet可以加载预训练权值

@wuzhihao7788
Copy link
Owner

目前是只有resnet可提供预加载模型。其他模型结构也可通过配置文件from选项变相的预加载模型。但是如果类别不一致的话。可能需要修改一下。最后一层模型输出的的结构。你的建议很好,下一个版本,我可以把此部分修改意见加入进去

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