我有一个模型的onnx格式,我想运行它的快速学习者。可能是这样的
learn = learn.load('model.onnx')另一种方式是转换回pth格式,但我没有看到任何适当的库在这个任务。在这两种方法中我都需要你的帮助。谢谢。
发布于 2022-06-16 09:35:57
没有本地解决方案,一些人目前正在开发该解决方案:https://github.com/ENOT-AutoDL/onnx2torch。
同样清楚的是,.pth检查点,通常只包含权重、偏差等参数。不是像conv2d,batchnorm2d,pooling这样的操作。另一方面,onnx模型包含操作和参数,这就是您可以推断它们的原因。如果从onnx中,您只需要权重和偏差就可以将状态加载到已经实现的火炬模型中,那么这可能很容易,如果您想要从onnx自动构建火炬模型,这是很困难的部分。
https://stackoverflow.com/questions/72640199
复制相似问题