RuntimeError: self.net.module.load_state_dict(obj['net'])

[taotaoyuhust]

using the pretrained model throw out this error:

RuntimeError: Error(s) in loading state_dict for EMANet:
Missing key(s) in state_dict: "extractor.4.0.conv1.weight", "extractor.4.0.bn1.weight", "extractor.4.0.bn1.bias", "extractor.4.0.bn1.running_mean", "extractor.4.0.bn1.running_var", "extractor.4.0.conv2.weight", ......
Unexpected key(s) in state_dict: "layer1.0.0.conv1.weight", "layer1.0.0.bn1.weight", "layer1.0.0.b

someone can help?

[Xia Li 李夏]

Sorry, I forgot to change the keys' names

[taotaoyuhust]

3x for reply, so how to solve it?

[Xia Li 李夏]

I will upload a usable pth sooner.

[happy]

@XiaLiPKU ，
hello author,
Thank you for your work, I want to ask you a question ,your EMAnet is used segment, if I am use in classification task .What do you think of the effect? If it is necessary to try , the emau structure need to change?

[illutheplanet]

same problem here

[illutheplanet]

it occurs to me:
RuntimeError: Error(s) in loading state_dict for EMANet:
Unexpected key(s) in state_dict: "extractor.6.0.conv1.weight", "extractor.6.0.bn1.weight", "extractor.6.0.bn1.bias", "extractor.6.0.bn1.running_mean", "extractor.6.0.bn1.running_var", "extractor.6.0.bn1.num_batches_tracked", "extractor.6.0.conv2.weight", "e

[Xia Li 李夏]

using the pretrained model throw out this error:

RuntimeError: Error(s) in loading state_dict for EMANet:
Missing key(s) in state_dict: "extractor.4.0.conv1.weight", "extractor.4.0.bn1.weight", "extractor.4.0.bn1.bias", "extractor.4.0.bn1.running_mean", "extractor.4.0.bn1.running_var", "extractor.4.0.conv2.weight", ......
Unexpected key(s) in state_dict: "layer1.0.0.conv1.weight", "layer1.0.0.bn1.weight", "layer1.0.0.b

someone can help?

Now, I have updated the pretrained model and the link.
You can download it again and direct use it in your code

[Xia Li 李夏]

it occurs to me:
RuntimeError: Error(s) in loading state_dict for EMANet:
Unexpected key(s) in state_dict: "extractor.6.0.conv1.weight", "extractor.6.0.bn1.weight", "extractor.6.0.bn1.bias", "extractor.6.0.bn1.running_mean", "extractor.6.0.bn1.running_var", "extractor.6.0.bn1.num_batches_tracked", "extractor.6.0.conv2.weight", "e

Now I have updated the pth file. Please download it again.