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

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

  •  
  •   ViggoSite · 332 天前 · 953 次点击
    这是一个创建于 332 天前的主题,其中的信息可能已经有所发展或是发生改变。

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

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