Fix typo

Tudor33 · Aug 4, 2020 · 0892e66 · 0892e66
1 parent 9535466
commit 0892e66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sdk/pynni/nni/nas/pytorch/cream/trainer.py b/src/sdk/pynni/nni/nas/pytorch/cream/trainer.py
@@ -266,7 +266,7 @@ def raw_sgd(w, g):
                     (prec1 > self.best_children_pool[-1][1] and cand_flops < self.best_children_pool[-1][2])):
                 val_prec1 = prec1
                 training_data = deepcopy(input_data[:self.slices].detach())
-                if self.best_children_pool:
+                if not self.best_children_pool:
                     features = deepcopy(output[:self.slices].detach())
                 else:
                     features = deepcopy(teacher_output[:self.slices].detach())