如题,一个开源模型,希望转换成 onnx 方便使用。有大佬可以的,可以加一下 V:www6688cx
1
tinybaby365 2023-12-25 15:08:37 +08:00
torch.onnx.export 不行吗?
|
2
tool2d 2023-12-25 15:11:48 +08:00
@tinybaby365 估计模型参数很多,就几个参数 torch.onnx.export 应该直接搞定了。
|
3
ViggoSite OP @tinybaby365 应该不行,我自己尝试过,但没有成功;主要是对模型这块,本身也不是很懂。希望找个大佬解决一下。
|
4
shm7 2023-12-25 15:51:43 +08:00
大一点的模型可能转出来有很多个文件,你是碰到这个问题了吗
|
5
opeth 2023-12-25 21:31:15 +08:00
pytorch 转 onnx 确实很多坑
pytorch 版本,onnx 版本,opset 的值都可能导致导出失败 |
6
zhangyichent 2023-12-26 11:50:39 +08:00
开源的直接放地址啊
|