HarukiYqM/Non-Local-Sparse-Attention

How much memory of the GPUS was used for the test?

Closed this issue · 1 comments

HI! Thank you for your sharing. Could you provide the memory of the GPUs used. I always got "RuntimeError: CUDA out of memory" when i test using V100 or 3090.

It can be test on a single 2070. Please follow the instructions and use the option —chop when testing on large images.