Browse Source

fix(imperative/utils): fix name filter of network varnode

GitOrigin-RevId: 20ead35335
release-1.3
Megvii Engine Team 4 years ago
parent
commit
5e54fdc32b
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/utils/network.py

+ 2
- 1
imperative/python/megengine/utils/network.py View File

@@ -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

Loading…
Cancel
Save