zhujiapeng / resefa

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

Computing jacobians is extremely slow #5

Open bluestyle97 opened 1 year ago

bluestyle97 commented 1 year 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 1 year 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.