V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
ViggoSite
V2EX  ›  外包

求大佬可以将 pth 模型导出 onnx,付费求

  •  
  •   ViggoSite · 2023-12-25 15:04:59 +08:00 · 978 次点击
    这是一个创建于 366 天前的主题,其中的信息可能已经有所发展或是发生改变。

    如题,一个开源模型,希望转换成 onnx 方便使用。有大佬可以的,可以加一下 V:www6688cx

    6 条回复    2023-12-26 11:50:39 +08:00
    tinybaby365
        1
    tinybaby365  
       2023-12-25 15:08:37 +08:00
    torch.onnx.export 不行吗?
    tool2d
        2
    tool2d  
       2023-12-25 15:11:48 +08:00
    @tinybaby365 估计模型参数很多,就几个参数 torch.onnx.export 应该直接搞定了。
    ViggoSite
        3
    ViggoSite  
    OP
       2023-12-25 15:14:01 +08:00
    @tinybaby365 应该不行,我自己尝试过,但没有成功;主要是对模型这块,本身也不是很懂。希望找个大佬解决一下。
    shm7
        4
    shm7  
       2023-12-25 15:51:43 +08:00
    大一点的模型可能转出来有很多个文件,你是碰到这个问题了吗
    opeth
        5
    opeth  
       2023-12-25 21:31:15 +08:00
    pytorch 转 onnx 确实很多坑
    pytorch 版本,onnx 版本,opset 的值都可能导致导出失败
    zhangyichent
        6
    zhangyichent  
       2023-12-26 11:50:39 +08:00
    开源的直接放地址啊
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1119 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 24ms · UTC 18:46 · PVG 02:46 · LAX 10:46 · JFK 13:46
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.