Torch 模型 onnx 文件的导出和调用(torch训练模型)

推荐整理分享Torch 模型 onnx 文件的导出和调用(torch训练模型)，希望有所帮助，仅作参考，欢迎阅读内容。

文章相关热门搜索词:torch模型保存,torch.nn.functional.grid_sample,torch训练模型,torchvision模型,torchvision模型,torch.nn.functional.normalize,torch模型部署,torch.nn.functional.grid_sample,内容如对您有帮助，希望把文章链接给更多的朋友！

Open Neural Network Exchange (ONNX，开放神经网络交换) 格式，是一个用于表示深度学习模型的标准，可使模型在不同框架之间进行转移

Torch 所定义的模型为动态图，其前向传播是由类方法定义和实现的

但是 Python 代码的效率是比较底下的，试想把动态图转化为静态图，模型的推理速度应当有所提升

Torch 框架中，torch.onnx.export 可以将父类为 nn.Module 的模型导出到 onnx 文件中，最重要的有三个参数：