search

lychenyoko / content-aware-gan-compression

CVPR2021 Content-Aware GAN Compression
MIT License
64 stars 7 forks source link

cal_flops #12

Closed zhoumenghan closed 2 years ago

zhoumenghan commented 2 years ago

Thanks for your open source~ I would like to ask what tool do you use to count the flops of styleGAN2, I use torchstat, but it has not been able to run normally, can you share the code of this part?

lychenyoko commented 2 years ago

You may find the function named StyleGAN2_FLOPCal under the file Util/Calculators.py.