genforce / sefa

[CVPR 2021] Closed-Form Factorization of Latent Semantics in GANs
https://genforce.github.io/sefa/
MIT License
963 stars 108 forks source link

StyleGAN 2 ADA transfer learning sourcenet error #20

Open Brainerd95 opened 2 years ago

Brainerd95 commented 2 years ago

I trained a couple of models with the pytorch version of StyleGAN 2 ADA and want to use them with SeFa. Most of them run fine, but using models that are based on the transfer learning sourcenet models provided in the official ADA repo results in the following error:

RuntimeError: Error(s) in loading state_dict for StyleGAN2Generator: Missing key(s) in state_dict: "mapping.dense2.weight", "mapping.dense2.bias", "mapping.dense3.weight", "mapping.dense3.bias", "mapping.dense4.weight", "mapping.dense4.bias", "mapping.dense5.weight", "mapping.dense5.bias", "mapping.dense6.weight", "mapping.dense6.bias", "mapping.dense7.weight", "mapping.dense7.bias". Unexpected key(s) in state_dict: "synthesis.layer15.weight", "synthesis.layer15.bias", "synthesis.layer15.noise_strength", "synthesis.layer15.noise", "synthesis.layer15.filter.kernel", "synthesis.layer15.style.weight", "synthesis.layer15.style.bias", "synthesis.layer16.weight", "synthesis.layer16.bias", "synthesis.layer16.noise_strength", "synthesis.layer16.noise", "synthesis.layer16.style.weight", "synthesis.layer16.style.bias", "synthesis.output8.weight", "synthesis.output8.bias", "synthesis.output8.style.weight", "synthesis.output8.style.bias".

I know that those models have a different layer structure, but is there any way to run them with this code anyways?

zytxxn commented 1 year ago

Hello, is the model of transfer learning training wrong? Are other models of training normal?