Problem with pretrained model loading

lyndonzheng / Pluralistic-Inpainting

[CVPR 2019]: Pluralistic Image Completion

672 stars 144 forks source link

Problem with pretrained model loading #29

Closed paridhimnnit closed 4 years ago

paridhimnnit commented 4 years ago

Hi,

I am getting the following error while loading the pretrained model in UI mode for Places2 model.

Pretrained network has fewer layers; The following are not initialized. I try to debug it and found the following error in loading state_dict for ResEncoder.

Missing key(s) in state_dict: "block0.conv1.module.bias". "block0.conv1.module.weight_u" ......

I would appreciate any kind of suggestion to resolve it.

lyndonzheng commented 4 years ago

This is a duplicate of https://github.com/lyndonzheng/Pluralistic-Inpainting/issues/9