OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

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

GitOrigin-RevId: af87f7007e
release-1.11.1
Megvii Engine Team 2 years ago
parent
034c7787fa
commit
babecba206
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      imperative/python/megengine/traced_module/pytree.py

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

@@ -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



Write Preview
Loading…
Cancel
Save