add min_repeat_ms to other CUDA tutorials (#2526)

apache · Jan 31, 2019 · ad8fc8c · ad8fc8c
1 parent cde1f76
commit ad8fc8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/autotvm/tune_nnvm_cuda.py b/tutorials/autotvm/tune_nnvm_cuda.py
@@ -118,7 +118,7 @@ def get_network(name, batch_size):
 
     'measure_option': autotvm.measure_option(
         builder=autotvm.LocalBuilder(timeout=10),
-        runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4),
+        runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150),
     ),
 }
 
@@ -369,6 +369,6 @@ def tune_and_evaluate(tuning_opt):
         runner=autotvm.RPCRunner(
             '1080ti',  # change the device key to your key
             'localhost', 9190,
-            number=20, repeat=3, timeout=4),
+            number=20, repeat=3, timeout=4, min_repeat_ms=150),
     ),
 }