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) super().__init__(expr, name, qualname)
self._shape = shape self._shape = shape
self._dtype = shape
self._dtype = dtype
self._qparams = qparams self._qparams = qparams


def __getstate__(self): def __getstate__(self):


Loading…
Cancel
Save