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