fix pynative amp bug

2021-02-23 21:38:39 +08:00 · 2021-02-23 21:38:39 +08:00 · 2ed34ad064
parent 761e5daaac
commit 2ed34ad064
1 changed files with 2 additions and 1 deletions
--- 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: