Browse Source

fix pynative amp bug

tags/v1.2.0-rc1
caifubi 4 years ago
parent
commit
2ed34ad064
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/common/parameter.py

+ 2
- 1
mindspore/common/parameter.py View File

@@ -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:


Loading…
Cancel
Save