search

POSTECH-CVLab / PyTorch-StudioGAN

StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.
https://github.com/MINGUKKANG
Other
3.42k stars 341 forks source link

Question about the time spent in Evaluation #154

Open cs911 opened 2 years ago

cs911 commented 2 years ago

During the Evaluation phase (evaluate fid prdc of the model) when I train the model, sometimes it takes ten minutes, sometimes two or three hours. And the gpu utilization is sometimes very high and sometimes very low. Why? My gpu is 2080ti.

alex4727 commented 2 years ago

Hi, FID and IS evaluation codes are mostly composed of GPU (CUDA) codes so they work faster. PRDC on the other hand, tends to take longer time as it runs on CPU. Besides, the size of dataset increases computing time of PRDC a lot more than FID so you should consider omitting PRDC if your dataset is too big or you don't need them.