Skip to content

#7058 [Tutorial] Import errors in deploy_detection.py and deploy_classification.py #7059

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tmoreau89 merged 7 commits into from
Dec 11, 2020
Merged

#7058 [Tutorial] Import errors in deploy_detection.py and deploy_classification.py #7059

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
9a5f904
Update deploy_classification.py
Dec 8, 2020
f85a707
Update deploy_detection.py
Dec 8, 2020
cc2ee40
Update deploy_classification.py
Dec 9, 2020
d2651d0
Update deploy_detection.py
Dec 9, 2020
cb65569
reverse changes in deploy_classification.py
Dec 9, 2020
ab64252
Reverse change in import paths
Dec 9, 2020
931ee1c
Update deploy_detection.py
Dec 10, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions vta/tutorials/frontend/legacy/deploy_detection.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@
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 import graph_runtime, utils
from tvm.contrib.download import download_testdata
from vta.testing import simulator
from vta.top import graph_pack
Expand Down Expand Up @@ -123,15 +123,15 @@
target = env.target if device == "vta" else env.target_vta_cpu

pack_dict = {
"yolov3-tiny": ["nn.max_pool2d", "cast", 4, 185],
"yolov3-tiny": ["nn.max_pool2d", "cast", 4, 186],
}

# Name of Darknet model to compile
# The ``start_pack`` and ``stop_pack`` labels indicate where
# to start and end the graph packing relay pass: in other words
# where to start and finish offloading to VTA.
# the number 4 indicate the the ``start_pack`` index is 4, the
# number 185 indicate the ``stop_pack index`` is 185, by using
# number 186 indicate the ``stop_pack index`` is 186, by using
# name and index number, here we can located to correct place
# where to start/end when there are multiple ``nn.max_pool2d``
# or ``cast``, print(mod.astext(show_meta_data=False)) can help
Expand Down