Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

StyleGAN 2 ADA transfer learning sourcenet error #20

Open
Brainerd95 opened this issue Apr 19, 2022 · 1 comment
Open

StyleGAN 2 ADA transfer learning sourcenet error #20

Brainerd95 opened this issue Apr 19, 2022 · 1 comment

Comments

@Brainerd95
Copy link

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
Copy link

zytxxn commented Apr 26, 2023

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants