Fix tune_relay_cuda.py (#6001)
authorLianmin Zheng <lianminzheng@gmail.com>
Tue, 7 Jul 2020 15:18:31 +0000 (08:18 -0700)
committerGitHub <noreply@github.com>
Tue, 7 Jul 2020 15:18:31 +0000 (08:18 -0700)
tutorials/autotvm/tune_relay_cuda.py

index a6fe45b..6a995af 100644 (file)
@@ -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),
     ),
 }