Missing key(s) and Unexpected key(s) in state_dict : "cab.conv1.weight"

[kravrolens]

Thanks for your excellent work, but testing your dataset with pretrained models,error occurs like this,

RuntimeError: Error(s) in loading state_dict for SPADEGenerator:
Missing key(s) in state_dict: "channelAtt.conv1.weight", "channelAtt.conv1.bias", "channelAtt.conv2.weight", "channelAtt.conv2.bias".
Unexpected key(s) in state_dict: "cab.conv1.weight", "cab.conv1.bias", "cab.conv2.weight", "cab.conv2.bias".

maybe the model you upload doesn't fit the net structure？
The versions torch--1.0.0 and torchvision--0.2.1 are the same as you.

thx!

[Ha0Tang]

@kravrolens can you provide more details?

[kravrolens]

The same problem with issue #5(#5 (comment))

testing with the uploaded pretrained model(facades) and dataset(facades) when I run by sh test_facades.sh

[Ha0Tang]

It works now, please try it again.

[Ha0Tang]

will close this issue now

[kravrolens]

Got it, thanks a lot.