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

[LonglongaaaGo]

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?