Error when load pretrained-checkpoint
Opened this issue · 2 comments
pyx-ustc commented
Interesting work!
Can you please provide the final pretrained-checkpoint for testing? Because I am trying to use your code to train the celeba, but failed.
When I run the command "sh ./scripts/train_retrieval_gmm.sh 0 0", it fails to load the pretrained celeba_gmm_gen.pt. Can you give me some tips to solve it ?
RuntimeError: Error(s) in loading state_dict for AdaINGenRaw:
Missing key(s) in state_dict: "enc_style.model.3.conv.weight", "enc_style.model.3.conv.bias", "enc_style.model.4.conv.weight", "enc_style.model.4.conv.bias", "enc_style.model.5.conv.weight", "enc_style.model.5.conv.bias".