fprotopapa opened a new pull request #7059:
URL: https://github.com/apache/tvm/pull/7059
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' changed from 185 to 186 in line 126.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
