批量处理图像时的问题

yizt / Grad-CAM.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...

Apache License 2.0

715 stars 166 forks source link

Open YuhuiMa opened 2 years ago

YuhuiMa commented 2 years ago

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

Panghema commented 2 years ago

应该是hook没释放吧

YuhuiMa commented 2 years ago

那应该如何修改代码呢

Panghema commented 2 years ago

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

YuhuiMa commented 2 years ago

好的，我看看。

zhhtu commented 2 years ago

self.handlers.remove(handle)