Commit 90b10b80 by Attila Dusnoki Committed by Thierry Moreau

Fix wrong n_trial number in autotvm tutorials' progress bar (#4070)

if n_trial is larger then config space.
parent be869546
...@@ -307,7 +307,8 @@ def tune_tasks(tasks, ...@@ -307,7 +307,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
...@@ -204,7 +204,8 @@ def tune_tasks(tasks, ...@@ -204,7 +204,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
...@@ -290,7 +290,8 @@ def tune_tasks(tasks, ...@@ -290,7 +290,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
...@@ -299,7 +299,8 @@ def tune_tasks(tasks, ...@@ -299,7 +299,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
...@@ -201,7 +201,8 @@ def tune_tasks(tasks, ...@@ -201,7 +201,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
...@@ -283,7 +283,8 @@ def tune_tasks(tasks, ...@@ -283,7 +283,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
...@@ -274,7 +274,8 @@ def tune_tasks(tasks, ...@@ -274,7 +274,8 @@ def tune_tasks(tasks,
tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file)) tuner_obj.load_history(autotvm.record.load_from_file(tmp_log_file))
# do tuning # do tuning
tuner_obj.tune(n_trial=min(n_trial, len(tsk.config_space)), n_trial = min(n_trial, len(tsk.config_space))
tuner_obj.tune(n_trial=n_trial,
early_stopping=early_stopping, early_stopping=early_stopping,
measure_option=measure_option, measure_option=measure_option,
callbacks=[ callbacks=[
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment