search

autonomousvision / stylegan-xl

[SIGGRAPH'22] StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets
MIT License
961 stars 113 forks source link

Setting for the pretrained models? #80

Open GinIsTheAnswer opened 2 years ago

GinIsTheAnswer commented 2 years ago

Hello! I'm trying to fine tune the FFHQ pretrained model on another dataset. Unfortunately, it fails with the following error:

RuntimeError: output with shape [12] doesn't match the broadcast shape [12, 12]

I wonder if the model has been trained with specific settings, which then don't work for out of the box fine tuning?

woctezuma commented 2 years ago