From 97a26bf6d11064824290e77925df5b9e5fca4226 Mon Sep 17 00:00:00 2001 From: miheer vaidya Date: Sun, 15 Dec 2019 16:09:16 -0700 Subject: [PATCH] Use the best tuner possible (#4397) * Use the best tuner possible * Add comment denoting availability of better tuners * Fix typos and wording --- tutorials/autotvm/tune_simple_template.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tutorials/autotvm/tune_simple_template.py b/tutorials/autotvm/tune_simple_template.py index dc1b2ce4a4fd..2e877b4cd3cc 100644 --- a/tutorials/autotvm/tune_simple_template.py +++ b/tutorials/autotvm/tune_simple_template.py @@ -32,6 +32,7 @@ # Install dependencies # -------------------- # To use autotvm package in TVM, we need to install some extra dependencies. +# This step (installing xgboost) can be skipped as it doesn't need XGBoost # (change "3" to "2" if you use python2): # # .. code-block:: bash @@ -294,7 +295,8 @@ def matmul(N, L, M, dtype): builder='local', runner=autotvm.LocalRunner(number=5)) -# begin tuning, log records to file `matmul.log` +# Begin tuning with RandomTuner, log records to file `matmul.log` +# You can use alternatives like XGBTuner. tuner = autotvm.tuner.RandomTuner(task) tuner.tune(n_trial=10, measure_option=measure_option,