[Bugfix][AutoScheduler] Correctly resume status (apache#7614)

trevor-m · May 6, 2021 · 13d9671 · 13d9671
1 parent ae1ad2a
commit 13d9671
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tvm/auto_scheduler/task_scheduler.py b/python/tvm/auto_scheduler/task_scheduler.py
@@ -516,9 +516,9 @@ def _restore_status(self, log_file, num_measures_per_round):
 
             if res.error_no == 0:
                 cost = array_mean(res.costs)
-                if self.best_costs[task_idx] < cost:
+                if cost < self.best_costs[task_idx]:
                     self.best_costs[task_idx] = cost
-                    self.task_best_cts = self.task_cts[task_idx]
+                    self.task_best_cts[task_idx] = self.task_cts[task_idx]
 
         for idx in range(len(self.tasks)):
             if self.task_cts[idx] - self.task_best_cts[idx] > self.early_stopping_task: