add min_repeat_ms to other CUDA tutorials (apache#2526)

merrymercy · Feb 17, 2019 · a3efd19 · a3efd19
1 parent 6d9a1e8
commit a3efd19
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),
     ),
 }