YOLO7导出engine模型失败, 好像是内存问题, 有朋友知道怎么解决吗
1条回答 默认 最新
- kakaccys 2022-08-24 08:47关注
你这是内存缺少崩了,先调小batch_size吧,然后onnx那块儿可以先不调用,先保存原始的模型,之后再加载模型进行压缩剪枝
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
你这是内存缺少崩了,先调小batch_size吧,然后onnx那块儿可以先不调用,先保存原始的模型,之后再加载模型进行压缩剪枝