Out of memory using U-IDS/P-IDS metrics

zsyzzsoft / co-mod-gan

[ICLR 2021, Spotlight] Large Scale Image Completion via Co-Modulated Generative Adversarial Networks

Other

445 stars 67 forks source link

Out of memory using U-IDS/P-IDS metrics #58

Open LonglongaaaGo opened 1 year ago

LonglongaaaGo commented 1 year ago

Hi @zsyzzsoft , Thank you for your awesome project. Could you help me with this: When I apply the U-IDS/P-IDS metrics on the 512*512 Places2 dataset (validation set, 36500 images). It occupies more than 64GB of memory.
However, I have no more memory I can use on my machine. So, is there any method that can solve this problem?