akanimax / pro_gan_pytorch

Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"
MIT License
537 stars 99 forks source link

Code profiling #63

Open akanimax opened 2 years ago

akanimax commented 2 years ago

Doing an informal test the other day, it seems like some operations have slowed down quite a bit since the new changes. Need to add profiling options to the code and remove any possible bottlenecks.