search

naver-ai / StyleMapGAN

Official pytorch implementation of StyleMapGAN (CVPR 2021)
https://www.youtube.com/watch?v=qCapNyRA_Ng
Other
458 stars 81 forks source link

What's the difference between normal generator and small generator? #19

Open nguyenquyem99dt opened 3 years ago

nguyenquyem99dt commented 3 years ago

Because I have a limited memory GPU, I just can train model with small generator. But I don't enough understand about the difference between normal G and small G. Can you explain the main difference, please?

blandocs commented 2 years ago

StyleMapGAN-Light is 2.5X smaller than the original version. Stylemap resizer accounts for a large portion of the network’s size, so we reduce the number of channels of feature maps in the stylemap resizer. The reconstruction image lacks some detail, but StyleMapGAN-Light still outperforms baselines, and FIDlerp is even better than the original version.

Please see appendix B in the supplementary paper.

nguyenquyem99dt commented 2 years ago

Thank you for your explanation.