Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use the best tuner possible #4397

Merged
merged 3 commits into from
Dec 15, 2019
Merged

Use the best tuner possible #4397

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
3f2decb
Use the best tuner possible
vmiheer Nov 21, 2019
18bf918
Add comment denoting availability of better tuners
vmiheer Dec 14, 2019
42e4498
Fix typos and wording
vmiheer Dec 14, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion tutorials/autotvm/tune_simple_template.py
View file
Open in desktop
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