Skip to content

Commit f339ad8

Browse files
merrymercytmoreau89
authored andcommitted
fix bugs for DCGAN and mobilenet (apache#27)
1 parent 1805e00 commit f339ad8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vta/python/vta/top/op.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,5 +149,5 @@ def schedule_conv2d_transpose(attrs, outs, target):
149149
raise RuntimeError("not support target %s" % target)
150150

151151
with tvm.target.arm_cpu(tvm.target.current_target().model):
152-
return _nn.schedule_conv2d_transpose(attrs, outs, target)
152+
return _nn.schedule_conv2d_transpose(attrs, outs, tvm.target.current_target())
153153

0 commit comments

Comments
 (0)