From 2ed34ad064d1fb2c1ba217b6fd0f9da21276515c Mon Sep 17 00:00:00 2001 From: caifubi Date: Tue, 23 Feb 2021 21:38:39 +0800 Subject: [PATCH] fix pynative amp bug --- mindspore/common/parameter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/common/parameter.py b/mindspore/common/parameter.py index b5e065f45d..1cd31597da 100644 --- a/mindspore/common/parameter.py +++ b/mindspore/common/parameter.py @@ -463,7 +463,8 @@ class Parameter(Tensor_): if mstype.implicit_conversion_seq[self.dtype] < mstype.implicit_conversion_seq[data.dtype]: raise_type_error(data.dtype) else: - data = Tensor(data, self.dtype) + from mindspore.ops import functional as F + data = F.cast(data, self.dtype) if isinstance(data, Tensor) and data.has_init: # The parameter has been initializered, directly update by the data if current_tensor_is_init: