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

PhilippvK · PhilippvK · commit 858a722498a4 · 2023-03-22T11:01:20.000+01:00
diff --git a/python/tvm/driver/tvmc/autotuner.py b/python/tvm/driver/tvmc/autotuner.py
@@ -805,7 +805,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),
             ],
         )

Original file line number	Diff line number	Diff line change
`@@ -805,7 +805,7 @@ def tune_tasks(`
`805`	`805`	`early_stopping=early_stopping,`
`806`	`806`	`measure_option=measure_option,`
`807`	`807`	`callbacks=[`
`808`		`- autotvm.callback.progress_bar(trials, prefix=prefix),`
	`808`	`+ autotvm.callback.progress_bar(min(trials, len(tsk.config_space)), prefix=prefix),`
`809`	`809`	`autotvm.callback.log_to_file(log_file),`
`810`	`810`	`],`
`811`	`811`	`)`