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!