如题,一个开源模型,希望转换成 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 开源的直接放地址啊 |