[TVMC] Use len of AutoTVM config space in progress bar

apache · Mar 21, 2023 · 1aff401 · 1aff401
1 parent 2a57e82
commit 1aff401
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/driver/tvmc/autotuner.py b/python/tvm/driver/tvmc/autotuner.py
@@ -806,7 +806,7 @@ def tune_tasks(
             early_stopping=early_stopping,
             measure_option=measure_option,
             callbacks=[
-                autotvm.callback.progress_bar(trials, prefix=prefix),
+                autotvm.callback.progress_bar(min(trials, len(tsk.config_space)), prefix=prefix),
                 autotvm.callback.log_to_file(log_file),
             ],
         )