NVlabs / stylegan2-ada

StyleGAN2 with adaptive discriminator augmentation (ADA) - Official TensorFlow implementation
https://arxiv.org/abs/2006.06676
Other
1.8k stars 501 forks source link

Pkl weights conversion between stlyegan2-ada and stylegan? #12

Open 100330706 opened 3 years ago

100330706 commented 3 years ago

Is it possible to convert the stylegan2 pkl weights to stylegan2-ada pkl weights and vicevers?

woctezuma commented 3 years ago

In my opinion, it depends on the config. Network architectures can differ if you don't use the right config.

https://github.com/NVlabs/stylegan2-ada/blob/9b78595d6f17a0a2299093b1297b0dc6953ea0d1/train.py#L161-L171

seawee1 commented 3 years ago

Shouldn't both networks in general be the same, as stylegan2-ada is more of a training technique rather than an architectural redesign of stylegan2?

woctezuma commented 3 years ago

You are correct. If you choose the right config, you will have the same network architecture as in StyleGAN2.