Commit 70231011 by ziheng Committed by Tianqi Chen

[FIX] Fix target in tutorial (#347)

parent eb9442e0
...@@ -177,14 +177,11 @@ out_shape = (batch_size, num_classes) ...@@ -177,14 +177,11 @@ 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
with tvm.target.rasp():
graph, lib, params = nnvm.compiler.build(
net, target, shape={"data": data_shape}, params=params) 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,
......
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