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,