批量处理图像时的问题

[YuhuiMa]

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

[Panghema]

应该是hook没释放吧

[YuhuiMa]

那应该如何修改代码呢

[Panghema]

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

[YuhuiMa]

好的，我看看。

[zhhtu]

self.handlers.remove(handle)