hello, there may be a bug for stylegan.

[wxxxxxxh]

Thanks for your great work in GAN inversion. I find a bug when I set the generator_type as stylegan. I guess that line 69 in derivable_generator.py should be self.stylegan = get_gan_model(gan_model_name)
and line 80 should be self.stylegan.net.cuda(device=device), otherwise, it may cause an error like this: AttributeError: 'StyleGANGeneratorNet' object has no attribute 'net'.

[berylsheep-up]

Thanks for advice!

[YAyaXM]

AttributeError: 'StyleGAN2GeneratorNet' object has no attribute 'net'

I've had this problem

[YAyaXM]

Could you give us some answers @JasonGUTU

[YingqingHe]

I also met this problem, thanks so much for your advice!