xyx112 2022-12-30 09:17 采纳率: 100%
浏览 24
已结题

linux环境下模型换砖pt转onnx时校验报错

img

在linux环境下pt文件转onnx,用makebin校验时报错。

  • 写回答

1条回答 默认 最新

  • 北下关吴中生 优质创作者: 人工智能技术领域 2022-12-30 09:42
    关注

    找到你pt转onnx文件的代码。
    找到parser.add_argument('--opset', type=int, default=12, help='ONNX: opset version')这条语句,将默认值修改为11,如:parser.add_argument('--opset', type=int, default=11, help='ONNX: opset version')。然后重新将.pt转化成.onnx文件即可解决该问题。

    从图中来看,12的版本不可以,转成11或者10应该是没问题的。

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

报告相同问题?

问题事件

  • 系统已结题 1月7日
  • 已采纳回答 12月30日
  • 创建了问题 12月30日