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

您好,我遇到了train.py中的run_fusion函数报错如下: #31

Open
Wbzb opened this issue Nov 25, 2023 · 2 comments
Open

您好,我遇到了train.py中的run_fusion函数报错如下: #31

Wbzb opened this issue Nov 25, 2023 · 2 comments

Comments

@Wbzb
Copy link

Wbzb commented Nov 25, 2023

RuntimeError: Error(s) in loading state_dict for FusionNet:
Missing key(s) in state_dict: "decode4.bn.weight", "decode4.bn.bias", "decode4.bn.running_mean", "decode4.bn.running_var", "decode3.bn.weight", "decode3.bn.bias", "decode3.bn.running_mean", "decode3.bn.running_var", "decode2.bn.weight", "decode2.bn.bias", "decode2.bn.running_mean", "decode2.bn.running_var", "decode1.bn.weight", "decode1.bn.bias", "decode1.bn.running_mean", "decode1.bn.running_var".

@Linfeng-Tang
Copy link
Owner

看这个错误显示需要加载预训练的权重,但是没有加载上,有可能是load了一个错误的融合权重哈。这个需要再检查一下代码。

@HaoyuYao
Copy link

HaoyuYao commented Dec 4, 2023

需要加载fusionmodel_final.pth才可以

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

3 participants