NVlabs / stylegan3

Official PyTorch implementation of StyleGAN3
Other
6.28k stars 1.1k forks source link

Loading generator weight to generator object without persistent class #240

Open swj0418 opened 1 year ago

swj0418 commented 1 year ago

I want to load the pretrained generator weights to a generator object that I instantiate inside my program. I need to do this because I wanted to tweak some intermediate activations, with requires me to change the base code slightly. However, I could not find a way to load the pretrained weights to generator object. Do you have support for this? Thanks.

PDillis commented 8 months ago

The README has a pretty straightforward explanation on how to load the network here. If you have a more specific question, feel free to ask.