提交 babecba2 编写于 作者: M Megvii Engine Team

fix(mge/traced_module): fix visualization bug by adding Node to non-const leaf

GitOrigin-RevId: af87f7007e4276c5b27e1d2f1402dfe3c120fb8d
上级 034c7787
...@@ -189,7 +189,7 @@ def _leaf_type(node): ...@@ -189,7 +189,7 @@ def _leaf_type(node):
def _is_const_leaf(node): def _is_const_leaf(node):
if isinstance(node, (RawTensor, NodeMixin, Module)): if isinstance(node, (RawTensor, Node, NodeMixin, Module)):
return False return False
return True return True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册