若我有你 2022-08-24 07:40 采纳率: 100%
浏览 47
已结题

YOLO7导出engine模型失败

YOLO7导出engine模型失败, 好像是内存问题, 有朋友知道怎么解决吗

img

  • 写回答

1条回答 默认 最新

  • kakaccys 2022-08-24 08:47
    关注

    你这是内存缺少崩了,先调小batch_size吧,然后onnx那块儿可以先不调用,先保存原始的模型,之后再加载模型进行压缩剪枝

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 9月6日
  • 已采纳回答 8月29日
  • 创建了问题 8月24日