Fixed typo on line 75 of nni/nas/pytorch/mutables.py (#3538)

336d671c · Ankit Aggarwal · GitHub · 1278c439 · 336d671c
Unverified Commit 336d671c authored Apr 21, 2021 by Ankit Aggarwal Committed by GitHub Apr 21, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nni/nas/pytorch/mutables.py nni/nas/pytorch/mutables.py +1 -1

No files found.
--- a/nni/nas/pytorch/mutables.py
+++ b/nni/nas/pytorch/mutables.py
@@ -72,7 +72,7 @@ class Mutable(nn.Module):
        """
        After the search space is parsed, it will be the module name of the mutable.
        """
-        return self._name if hasattr(self, "_name") else "_key"
+        return self._name if hasattr(self, "_name") else self._key

    @name.setter
    def name(self, name):