[Bug] (suggested fix) `nn.Parameter` are not added to root after being traced in `mmrazor.models.task_utils.tracer.fx.custom_tracer.build_graphmodule()` about mmrazor HOT 2 OPEN

@@ -139,7 +139,10 @@ def _prepare_module_dict(model: torch.nn.Module, fx_graph):
     for nose in fx_graph.nodes:
         if node.op == 'get_attr':
             attr = _get_attrs(model, node.target)
             if isinstance(attr, nn.Module):
                 module_dict[node.target] = nn.Module()
                 special_nodes.append(node)
+            else:
+                module_dict[node.target] = attr
+                special_nodes.append(node)

@@ -139,7 +139,10 @@ def _prepare_module_dict(model: torch.nn.Module, fx_graph):
     for nose in fx_graph.nodes:
         if node.op == 'get_attr':
             attr = _get_attrs(model, node.target)
             if isinstance(attr, nn.Module):
                 module_dict[node.target] = nn.Module()
                 special_nodes.append(node)
+            elif isinstance(attr, nn.Parameter):
+                module_dict[node.target] = attr
+                special_nodes.append(node)