StyleGAN2-ada-pytorch models supported?

[TheodoreGalanos]

Hello and thanks for sharing your code!

I am trying to train an encoder on a stylegan2-ada-pytorch model that I have pretrained. I am using a .pkl file (as is the default output of that repo) and I am getting errors (related to the ada repo in fact). Is it possible to use that model here, or do I need to convert it somehow?

Thanks in advance!

[omertov]

Hi @TheodoreGalanos!
This repository uses the stylegan2 implementation by rosinality: https://github.com/rosinality/stylegan2-pytorch.
To load your trained model you will need to first convert it into a checkpoint compatible with the above repository.

(From a quick glance I saw a relevant issue in stylegan2-ada-pytorch, after which the convert_weight.py script from the rosinality repository can be used)

[TheodoreGalanos]

Thanks just saw that too, would be great. Really want to use your implementation!

[omertov]

For future reference, it seems like there is a way for converting the stylegan2-ada-pytorch model: rosinality/stylegan2-pytorch#206.

Feel free to reopen the issue in case the suggested solution does not work :)