fix the bug that can only create two stage (#4956)

570448ea · changqi97 · GitHub · f24dc27b · 570448ea
Unverified Commit 570448ea authored Jul 01, 2022 by changqi97 Committed by GitHub Jul 01, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nni/algorithms/nas/pytorch/fbnet/mutator.py nni/algorithms/nas/pytorch/fbnet/mutator.py +1 -1

No files found.
--- a/nni/algorithms/nas/pytorch/fbnet/mutator.py
+++ b/nni/algorithms/nas/pytorch/fbnet/mutator.py
@@ -148,7 +148,7 @@ class FBNetMutator(BaseMutator):

            for i in range(left, left + right):
                ops_names_mutable[i] = ops_names
-            left = right
+            left += right

        # Create the mixed op
        for i, mutable in enumerate(self.undedup_mutables):