Browse Source

fix(traced_module): fix tensor node init

GitOrigin-RevId: bdd8259ce7
release-1.7
Megvii Engine Team “wenjuan” 3 years ago
parent
commit
fefd4f0928
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/traced_module/node.py

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

@@ -231,7 +231,7 @@ class TensorNode(Node):
):
super().__init__(expr, name, qualname)
self._shape = shape
self._dtype = shape
self._dtype = dtype
self._qparams = qparams

def __getstate__(self):


Loading…
Cancel
Save