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

你可能感兴趣的:(pytorch实践,pytorch,pytorch)