Commit 70231011 by ziheng Committed by Tianqi Chen

[FIX] Fix target in tutorial (#347)

parent eb9442e0
...@@ -177,15 +177,12 @@ out_shape = (batch_size, num_classes) ...@@ -177,15 +177,12 @@ out_shape = (batch_size, num_classes)
# Pi, which has been proved as a good compilation configuration. # Pi, which has been proved as a good compilation configuration.
if use_rasp: if use_rasp:
target = "llvm -target=armv7l-none-linux-gnueabihf -mcpu=cortex-a53 -mattr=+neon" target = tvm.target.rasp()
else: else:
target = "llvm" target = tvm.target.create('llvm')
graph, lib, params = nnvm.compiler.build(
# use `with tvm.target.rasp` for some target-specified optimization net, target, shape={"data": data_shape}, params=params)
with tvm.target.rasp():
graph, lib, params = nnvm.compiler.build(
net, target, shape={"data": data_shape}, params=params)
# After `nnvm.compiler.build`, you will get three return values: graph, # After `nnvm.compiler.build`, you will get three return values: graph,
# library and the new parameter, since we do some optimization that will # library and the new parameter, since we do some optimization that will
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment