zhujiapeng / resefa

[ICML 2022] Region-Based Semantic Factorization in GANs
https://zhujiapeng.github.io/resefa/
MIT License
71 stars 4 forks source link

Computing jacobians is extremely slow #5

Open bluestyle97 opened 2 years ago

bluestyle97 commented 2 years ago

Hi, I find the process of computing jacobians is extremely slow when testing with a 512x512 pretrained stylegan2 model, is this normal?

zhujiapeng commented 2 years ago

Yes, it is normal. It will take about one hour when using a 1024x1024 model on 2080 Ti GPU. StyleGAN3 will take a much longer time. We will release the code to speed up the process of computing the jacobian in the future.