Skip to content

Commit

Permalink
Use the best tuner possible (#4397)
Browse files Browse the repository at this point in the history
* Use the best tuner possible

* Add comment denoting availability of better tuners

* Fix typos and wording
  • Loading branch information
vmiheer authored and tqchen committed Dec 15, 2019
1 parent f221844 commit 97a26bf
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tutorials/autotvm/tune_simple_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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,
Expand Down

0 comments on commit 97a26bf

Please sign in to comment.