Open sriderya opened 3 years ago
I have the same problem
I have the same problem
Hi, I am currently facing the same problem as you with regards to style1.weight:
RuntimeError: Error(s) in loading state_dict for Generator: size mismatch for style.1.weight: copying a param with shape torch.Size([512, 1024]) from checkpoint, the shape in current model is torch.Size([512, 512]).
After reading every related problem I'm still not sure why this vector contains 1024 and how I should handle this. I don't think it's because of the auto-config because I trained another model using paper256 and the first mapping layer has again shape [512, 1024]. Could anyone explain to me at which point in training I could change this? And additionally, how I can use/convert my model containing [512, 1024]-vectors? Best wishes, Marcel!
Hello,
I trained a conditional StyleGAN2-ada-pytorch model with a custom dataset. Then, I converted my pkl model to pt model by using https://github.com/dvschultz/stylegan2-ada-pytorch/blob/main/export_weights.py. Then, I integrated my model to Ganspace wrapper and I ran it. This gives me the following error.
I believe that the mismatches arise from conditioning. It looks like something changes after label embeddings are added to unconditional Stylegan2-ada-pytorch network. Is there any solution for this problem or any suggestion to overcome this?
Thanks in advance