pytorch模型转onnx模型trace方式和script方式

pytorch模型转onnx模型trace方式和script方式

pytorch模型转onnx模型trace方式和script方式

• 1、编写模型
• 2、trace方式生成模型
• 3、script方式生成模型
• 4、混合trace和script生成模型
• 5、保存与加载模型
• 6、torchscript介绍
• • 6.1、Mixing Tracing and scripting
• 7、pytorch模型转onnx模型trace方式
• 8、pytorch模型转onnx模型scrip方式

1、编写模型

# -*- coding : UTF-8 -*-
# @file   : model_authoring.py
# @Time   : 2021/11/29 0029 21:53
# @Author : wmz

im