run multi_code_inversion with stylegan2_ffhq model failed

genforce / mganprior

[CVPR 2020] Image Processing Using Multi-Code GAN Prior

https://genforce.github.io/mganprior/

288 stars 44 forks source link

run multi_code_inversion with stylegan2_ffhq model failed #8

Open Xiaomao136 opened 3 years ago

Xiaomao136 commented 3 years ago

An error occurred when running multi_code. Checking the code found two errors in the derivable_generator.py file. The errors are as follows: line 66 self.stylegan = get_gan_model(gan_model_name).net ---> self.stylegan = get_gan_model(gan_model_name)

line 77 self.stylegan.cuda(device=device) ---> self.stylegan.net.cuda(device=device)