Skip to content

Commit 2d332b3

Browse files
committed
[TVM][Tutorial] mention --task flag for tvmc tune in tvmc_command_line_driver.py
1 parent 030bd33 commit 2d332b3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

gallery/tutorial/tvmc_command_line_driver.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -417,6 +417,11 @@
417417
# process, in terms of number of repetitions (``--repeat`` and ``--number``, for example), the tuning
418418
# algorithm to be used, and so on. Check ``tvmc tune --help`` for more information.
419419
#
420+
# In some situations it might be a good idea, to only tune specific tasks (i.e. the most relevant ones)
421+
# to waste less time tuning simpler workworloads. The flag `--task` offers versatile options to limt
422+
# the tasks used for tuning, e.g. `--task 20,22` or `--task 16-`. All available tasks can be printed
423+
# using `--task list`.
424+
#
420425

421426
################################################################################
422427
# Compiling an Optimized Model with Tuning Data

0 commit comments

Comments
 (0)