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.

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

WARNING - The model and loaded state dict do not match exactly #23

Closed
Yaoxingtian opened this issue Oct 18, 2022 · 1 comment
Closed

WARNING - The model and loaded state dict do not match exactly #23

Yaoxingtian opened this issue Oct 18, 2022 · 1 comment

Comments

@Yaoxingtian
Copy link

你好:
我使用的是segnext_tiny_512x512_ade_160k.pth，以及配置文件segnext.tiny.512x512.ade.160k.py，训练的时候有警告 WARNING - The model and loaded state dict do not match exactly，用small模型也是一样的警告。
请问模型是有什么问题吗？ 谢谢
@uyzhang
Copy link
Collaborator

uyzhang commented Oct 18, 2022 •
edited by MenghaoGuo
Loading

Because some parameters of the pretrained model such as 'classification head' do not need to be used in training segmentation model.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@Yaoxingtian @uyzhang @MenghaoGuo