Using another pretrained StyleGAN2

[JbIPS]

Hi,

I'm playing with your notebook (awesome work btw!) and I try to give it another pretrained GAN from Awesome Pretrained StyleGAN2.

I used the anime one (PyTorch implementation from here) but I get

TypeError                                 Traceback (most recent call last)

[<ipython-input-10-2395bdddca96>](https://localhost:8080/#) in <module>()
     22 
     23 #print(ckpt)
---> 24 generator.load_state_dict(ckpt["g"], strict=False)
     25 
     26 #@title Generate results

TypeError: 'Generator' object is not subscriptable

Do I need further operation on the model to make it compatible?

Thank you

[mchong6]

Maybe the pretrained model is trained using the official stylegan2 repo. The repo I built upon is by rosalinity. To make it compatible you have to convert the model file to one that is compatible with this repo. Look at the readme of the rosalinity repo to see how to do that.

[JbIPS]

My bad, it looks like my file only contains the structure of the network, not the saved weights