Browse Source

fix(traced_module): fix tensor node init

GitOrigin-RevId: bdd8259ce7
tags/v1.7.0.m1
Megvii Engine Team 3 years ago
parent
commit
d3c7c56317
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