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

GitOrigin-RevId: af87f7007e4276c5b27e1d2f1402dfe3c120fb8d

fix(mge/traced_module): fix visualization bug by adding Node to non-const leaf
GitOrigin-RevId: af87f7007e4276c5b27e1d2f1402dfe3c120fb8d
babecba2 · Megvii Engine Team · 034c7787 · babecba2
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

imperative/python/megengine/traced_module/pytree.py imperative/python/megengine/traced_module/pytree.py +1 -1

未找到文件。
--- a/imperative/python/megengine/traced_module/pytree.py
+++ b/imperative/python/megengine/traced_module/pytree.py
@@ -189,7 +189,7 @@ def _leaf_type(node):
 def _is_const_leaf(node):
-    if isinstance(node, (RawTensor, NodeMixin, Module)):
+    if isinstance(node, (RawTensor, Node, NodeMixin, Module)):
        return False
    return True