search

qqlu / Entity

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation
Other
704 stars 58 forks source link

CUDA out of memory #26

Open hkxiao opened 1 year ago

hkxiao commented 1 year ago

'torch.cuda.empty_cache()' seems don`t work after process one chunk image, I get an RuntimeError: CUDA out of memory.

Runist commented 7 months ago

'torch.cuda.empty_cache()' seems don`t work after process one chunk image, I get an RuntimeError: CUDA out of memory.

Do you have a solution?