Grad-CAM.pytorch 批量处理图像时的问题

批量处理图像时的问题

Open YuhuiMa opened this issue 3 years ago • 5 comments

您好！我这边用这个代码处理单张图没问题，但python里面写for循环批量处理图片的时候GPU的显存占用会随着处理图片数量的增大而增大，最终会报CUDA out of memory，请问该如何解决？

Oct 21 '21 08:10 YuhuiMa

应该是hook没释放吧

Jun 23 '22 07:06 Panghema

那应该如何修改代码呢

Jun 23 '22 07:06 YuhuiMa

emm 我看了下有这个函数remove_handlers释放了的确认下有没有正常调用吧如果这个有释放我也不清楚了。。

Jun 23 '22 07:06 Panghema

好的，我看看。

Jun 23 '22 07:06 YuhuiMa

self.handlers.remove(handle)

Jul 28 '22 01:07 zhhtu