-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Closed
Description
I ran into import errors while following the VTA tutorial for 'Deploy Pretrained Vision Model from MxNet on VTA' under Ubuntu 18.04.5 LTS and Docker images created with Dockerfile.ci_cpu. Scripts launched from tvm base directory.
File: tvm/vta/tutorials/frontend/deploy_classification.py
In line 59 to 61:
import vta
from vta.testing import simulator
from vta.top import graph_pack
Import paths have changed and function call for graph_pack in line 181.
Changed to:
import vta
from vta.python.vta.testing import simulator
from vta.python.vta.top import graphpack as graph_pack
from vta.python import vta
and in line 181
relay_prog = graph_pack.graph_pack(
Similiar issue in file tvm/vta/tutorials/frontend/legacy/deploy_detection.py
Line 58 to 64 and function call in line 222:
from tvm import rpc, autotvm, relay
from tvm.relay.testing import yolo_detection, darknet
from tvm.relay.testing.darknet import __darknetffi__
from tvm.contrib import graph_runtime, graph_runtime, util
from tvm.contrib.download import download_testdata
from vta.testing import simulator
from vta.top import graph_pack
Changed to:
from tvm import rpc, autotvm, relay
from tvm.relay.testing import yolo_detection, darknet
from tvm.relay.testing.darknet import __darknetffi__
from tvm.contrib import graph_runtime, graph_runtime, utils
from tvm.contrib.download import download_testdata
from vta.python.vta.testing import simulator
from vta.python.vta.top import graphpack as graph_pack
from vta.python import vta
and
mod = graph_pack.graph_pack(
Also layer id for stop layer 'cast' is 186 in line 126.
Metadata
Metadata
Assignees
Labels
No labels