sefa icon indicating copy to clipboard operation
sefa copied to clipboard

StyleGAN 2 ADA transfer learning sourcenet error

Open Brainerd95 opened this issue 3 years ago • 1 comments

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?

Brainerd95 avatar Apr 19 '22 18:04 Brainerd95

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

zytxxn avatar Apr 26 '23 01:04 zytxxn