pytorch把模型保存为onnx

最新推荐文章于 2025-04-28 00:16:08 发布

weixin_45415546

最新推荐文章于 2025-04-28 00:16:08 发布

阅读量4.4k

点赞数

分类专栏： pytorch 文章标签： onnx pytorch

本文链接：https://blog.youkuaiyun.com/weixin_45415546/article/details/100928019

版权

pytorch 专栏收录该内容

1 篇文章

订阅专栏

普通方式

torch_out = torch.onnx._export(model,img_tensor,"test.onnx",export_params=True)

动态batchsize方式

  input_names = ['input']
  output_names = ['output']
  #model  input_tensor  根据实际情况设置
  torch.onnx.export(model, input_tensor, 'example.onnx',
  export_params=True, verbose=False, input_names = input_names,
  output_names=output_names, opset_version=10,dynamic_axes={"input":{0:"batch_size"},
                                  "output":{0:"batch_size"}})