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

Posted 2021-06-30 踟蹰横渡口,彳亍上滩舟。

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