摘要: 多个onnx可以合并为一个onnx,这样在c++调用的时候会方便一些。 如果有原始pytorch代码和模型参数,可以在导出时合并。 如果只有onnx文件,可以用下面的方法合并,每个模型各自是各自的输入输出。 import onnx import torch import torch.nn as nn 阅读全文
posted @ 2025-04-19 22:25 Dsp Tian 阅读(149) 评论(0) 推荐(0)