torch量化训练后导出onnx模型并运行

Posted 踟蹰横渡口,彳亍上滩舟。

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch量化训练后导出onnx模型并运行相关的知识，希望对你有一定的参考价值。

1. 依赖项

import io
import numpy as np

from torch import nn
import torch.utils.model_zoo as model_zoo
import torch.onnx

2. 模型结构

# Super Resolution model definition in PyTorch
import torch.nn as nn
import torch.nn

以上是关于torch量化训练后导出onnx模型并运行的主要内容，如果未能解决你的问题，请参考以下文章