PyTorch 量化

import numpy as np
import torch
from torch import nn
from tqdm import tqdm
import tvm
from tvm import relay