From babecba2062972c2cff3109bb62b9daf79265522 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 28 Sep 2022 17:41:35 +0800 Subject: [PATCH] fix(mge/traced_module): fix visualization bug by adding Node to non-const leaf GitOrigin-RevId: af87f7007e4276c5b27e1d2f1402dfe3c120fb8d --- imperative/python/megengine/traced_module/pytree.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imperative/python/megengine/traced_module/pytree.py b/imperative/python/megengine/traced_module/pytree.py index 5a59fc1fd..6cfcffde5 100644 --- 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 -- GitLab