Commit 97a26bf6 by miheer vaidya Committed by Tianqi Chen

Use the best tuner possible (#4397)

* Use the best tuner possible

* Add comment denoting availability of better tuners

* Fix typos and wording
parent f221844b
...@@ -32,6 +32,7 @@ The whole workflow is illustrated by a matrix multiplication example. ...@@ -32,6 +32,7 @@ The whole workflow is illustrated by a matrix multiplication example.
# Install dependencies # Install dependencies
# -------------------- # --------------------
# To use autotvm package in TVM, we need to install some extra 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): # (change "3" to "2" if you use python2):
# #
# .. code-block:: bash # .. code-block:: bash
...@@ -294,7 +295,8 @@ measure_option = autotvm.measure_option( ...@@ -294,7 +295,8 @@ measure_option = autotvm.measure_option(
builder='local', builder='local',
runner=autotvm.LocalRunner(number=5)) 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 = autotvm.tuner.RandomTuner(task)
tuner.tune(n_trial=10, tuner.tune(n_trial=10,
measure_option=measure_option, measure_option=measure_option,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment