PhotoSketch icon indicating copy to clipboard operation
PhotoSketch copied to clipboard

Can't load model

Open HADJHENNI opened this issue 2 years ago • 2 comments

Hello, I'm trying to run the code, but each time i try to load the model i get this error : RuntimeError: Error(s) in loading state_dict for ResnetGenerator: Missing key(s) in state_dict: "model.1.bias", "model.4.bias", "model.7.bias", "model.10.conv_block.1.bias", "model.10.conv_block.6.bias", "model.11.conv_block.1.bias", "model.11.conv_block.6.bias", "model.12.conv_block.1.bias", "model.12.conv_block.6.bias", "model.13.conv_block.1.bias", "model.13.conv_block.6.bias", "model.14.conv_block.1.bias", "model.14.conv_block.6.bias", "model.15.conv_block.1.bias", "model.15.conv_block.6.bias", "model.16.conv_block.1.bias", "model.16.conv_block.6.bias", "model.17.conv_block.1.bias", "model.17.conv_block.6.bias", "model.18.conv_block.1.bias", "model.18.conv_block.6.bias", "model.19.bias", "model.22.bias". .....

any suggestions to solve this issue ? Thank you

HADJHENNI avatar Mar 17 '22 10:03 HADJHENNI

Hi, if you are using compatible model definition and checkpoint, then the cause is most likely the inconsistency with the saving and loading of nn.DataParellel and/or weight devices, for which you can check out this post.

mtli avatar Mar 17 '22 22:03 mtli

@HADJHENNI Have you solved the issue? met with the same issue... @mtli Did you train the model using 3 GPU cards?

mychina75 avatar Jan 11 '23 10:01 mychina75