Question about vgg16

from tvm import relay

from tvm import nd

import numpy as np

import torch

import torchvision

from op_runtime import op_run

from PIL import Image

from torchvision import transforms

from torchvision import models

import onnx

import tvm

onnx_model = onnx.load_model(‘vgg16.onnx’)

x = np.random.rand(1, 3, 224, 224)

input_name = ‘input0’

shape_dict = {input_name: x.shape}

mod, params = relay.frontend.from_onnx(onnx_model, shape_dict)