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.
# 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 @@ measure_option = autotvm.measure_option(
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,
......
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