diff --git a/imperative/python/megengine/traced_module/node.py b/imperative/python/megengine/traced_module/node.py index 7bc15705..96091b36 100644 --- a/imperative/python/megengine/traced_module/node.py +++ b/imperative/python/megengine/traced_module/node.py @@ -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):