diff --git a/tutorials/autotvm/tune_nnvm_cuda.py b/tutorials/autotvm/tune_nnvm_cuda.py index 6e0ace462d6f..285940f80845 100644 --- 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), ), }