Yolov4模型训练正常,验证一次正常,验证多张图片出现内存泄露
收藏
我在使用paddle的yolo4进行车辆数据检测,模型训练完成后在验证时总是出现运行几张图片就提示内存溢出的错误 验证单张图片时内存会飙升但都能正常得到结果。我检查了一下自己的代码并没有开启多线程,请问下在单独一张张图片检测时怎么会出现内存越来越满。最终造成内存泄露。或者说paddle有自己的内存释放机制,需要手动释放而我没有释放。有遇到过类似问题的大佬么。麻烦帮忙看看。实例化模型检测代码如下:
运行截图:检测同一张图片有时七八次提示内存溢出,有时两三次就提示内存溢出了。
0
收藏
请登录后评论
建议提issue