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)





---
[Visit Topic](https://discuss.tvm.apache.org/t/question-about-vgg16/11797/1) to 
respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/a8ada49d698cd5a05230c073e8fa6d2a4eec0f74b1d0fb6fd99dc55e88f9e162).

Reply via email to