yhlleo/RG-UNIT

Error when load pretrained-checkpoint

Opened this issue · 2 comments

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".

Hi @pyx-ustc, it seems that there are some missing layers when loading the model. I'll check it later.

Hi @pyx-ustc , we rewrote the codes for better usages. I'll upload the models soon.