首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何将ONNX转换回pth格式

如何将ONNX转换回pth格式
EN

Stack Overflow用户
提问于 2022-06-16 03:39:57
回答 1查看 278关注 0票数 1

我有一个模型的onnx格式,我想运行它的快速学习者。可能是这样的

代码语言:javascript
复制
learn = learn.load('model.onnx')

另一种方式是转换回pth格式,但我没有看到任何适当的库在这个任务。在这两种方法中我都需要你的帮助。谢谢。

EN

回答 1

Stack Overflow用户

发布于 2022-06-16 09:35:57

没有本地解决方案,一些人目前正在开发该解决方案:https://github.com/ENOT-AutoDL/onnx2torch

同样清楚的是,.pth检查点,通常只包含权重、偏差等参数。不是像conv2dbatchnorm2dpooling这样的操作。另一方面,onnx模型包含操作和参数,这就是您可以推断它们的原因。如果从onnx中,您只需要权重和偏差就可以将状态加载到已经实现的火炬模型中,那么这可能很容易,如果您想要从onnx自动构建火炬模型,这是很困难的部分。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/72640199

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档