批量处理图像时的问题
YuhuiMa opened this issue · 5 comments
YuhuiMa commented
您好！我这边用这个代码处理单张图没问题，但python里面写for循环批量处理图片的时候GPU的显存占用会随着处理图片数量的增大而增大，最终会报CUDA out of memory，请问该如何解决？
Panghema commented
应该是hook没释放吧
YuhuiMa commented
那应该如何修改代码呢
Panghema commented
emm 我看了下有这个函数remove_handlers释放了的 确认下有没有正常调用吧 如果这个有释放 我也不清楚了。。
YuhuiMa commented
好的，我看看。
zhhtu commented
self.handlers.remove(handle)