TVM编译MXNet模型

编译MXNet模型

  • 从Gluon Model Zoo下载Resnet18模型
  • 编译图
  • 在TVM上执行可移植图形
  • 使用具有预训练权重的MXNet符号

本文是介绍如何使用Relay部署mxnet模型的入门教程。
首先,需要安装mxnet模块。
一个快速的解决方案是

pip install mxnet --user

或请参阅官方安装指南。 https://mxnet.apache.org/versions/master/install/index.html

import mxnet as mx
import tvm
import tvm.relay as relay
import numpy as np

从Gluon Model Zoo下载Resnet18模型

在本节中,我们下载预训练的imagenet模型并对图像进行分类。

from tvm.contrib.download import download_testdata
from mxnet.gluon.model_zoo.vision import get_model
from PIL import Image
from matplotlib import pyplot as plt\

block=get_model("resnet18

你可能感兴趣的:(TVM)