diff --git a/imperative/python/megengine/utils/network.py b/imperative/python/megengine/utils/network.py index 8bcf4d8927766e97e42c0fc59ec41c822291a3b8..8738a5b54526f680955a073a00321e5636fc5c4d 100644 --- a/imperative/python/megengine/utils/network.py +++ b/imperative/python/megengine/utils/network.py @@ -663,6 +663,7 @@ class NodeFilterName(NodeFilter): def __init__(self, node_iter, pattern, ignorecase): super().__init__(node_iter) + self.pattern = pattern self._re = self.make_re(pattern, ignorecase) @classmethod @@ -676,5 +677,5 @@ class NodeFilterName(NodeFilter): def __iter__(self): for i in self._iter: - if self._re.match(i.name): + if self.pattern == i.name or self._re.match(i.name): yield i