YOLO7导出engine模型失败, 好像是内存问题, 有朋友知道怎么解决吗
你这是内存缺少崩了,先调小batch_size吧,然后onnx那块儿可以先不调用,先保存原始的模型,之后再加载模型进行压缩剪枝