Closed anantsakhare closed 1 year ago
After using the Cuda as a device, it consumes GPU memory, there is a memory leak.
causes due to score.backward(retain_graph=True) @pooya-mohammadi
@anantsakhare Can you please tell me how you solved the memory leak problem? I get 'Cuda out of memory' error when I run grad-cam on bunch of images using CUDA.
After using the Cuda as a device, it consumes GPU memory, there is a memory leak.
causes due to score.backward(retain_graph=True) @pooya-mohammadi