fix some error

open-mmlab · Aug 10, 2022 · 7004e56 · 7004e56
1 parent 8f13648
commit 7004e56
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mmrazor/structures/graph/base_graph.py b/mmrazor/structures/graph/base_graph.py
@@ -90,7 +90,7 @@ def copy_from(cls,
 
         # connect
         for old in graph:
-            for pre in old.pre:
+            for pre in old.prev_nodes:
                 new_graph.connect(old2new[pre], old2new[old])
         return new_graph
 

diff --git a/mmrazor/structures/graph/module_graph.py b/mmrazor/structures/graph/module_graph.py
@@ -47,7 +47,7 @@ def __init__(self,
             >>> class Pool(nn.Module):
                     def forward(x):
                         return F.adaptive_avg_pool2d(x,2).flatten(1)
-            >>> node= ModuleNode('pass_0',Pool(),expadn_ratio=4)
+            >>> node= ModuleNode('pass_0',Pool(),expand_ratio=4)
             >>> assert node.out_channels == node.in_channels*4
         """
 
@@ -207,6 +207,10 @@ def check_type(self):
 class ModuleGraph(BaseGraph[MODULENODE]):
     """Computatation Graph."""
 
+    def __init__(self) -> None:
+        super().__init__()
+        self._model = None
+
     # functions to generate module graph.
 
     @staticmethod