The input parameters of the Generator

[linziqu]

Hi, thanks for your great work!

I am curious about all the input parameters such as 'noise_mode', 'force_fp' of self.G.synthesis(w, noise_mode='const', force_fp32=True). And I also want to know how to return the featrures of each layer in the self.G.

Thanks

[danielroich]

Hi @linziqu, The 'noise mode' indicates how the noise injected into StyleGAN behaves. StyleGAN actually receives 2 inputs. The latent code and a noise vector can be seen in the original paper. The force_fp handles the float operations when computed in the generator. You can read more in the official documentation "https://github.com/NVlabs/stylegan2-ada-pytorch".

Regarding the features - you can access the StyleGAN layers and thus the weights directly as any other NN in PyTorch. But again I highly recommend reading the official documentation for further instructions.