From: Lianmin Zheng Date: Tue, 7 Jul 2020 15:18:31 +0000 (-0700) Subject: Fix tune_relay_cuda.py (#6001) X-Git-Tag: upstream/0.7.0~448 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=723ed593f52e3b247c27c5251b64722fc0e00765;p=platform%2Fupstream%2Ftvm.git Fix tune_relay_cuda.py (#6001) --- diff --git a/tutorials/autotvm/tune_relay_cuda.py b/tutorials/autotvm/tune_relay_cuda.py index a6fe45b..6a995af 100644 --- a/tutorials/autotvm/tune_relay_cuda.py +++ b/tutorials/autotvm/tune_relay_cuda.py @@ -128,11 +128,7 @@ tuning_option = { 'measure_option': autotvm.measure_option( builder=autotvm.LocalBuilder(timeout=10), - #runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150), - runner=autotvm.RPCRunner( - '1080ti', # change the device key to your key - '0.0.0.0', 9190, - number=20, repeat=3, timeout=4, min_repeat_ms=150) + runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150), ), }