Browse Source

!9203 fix bug for upgrading python3.8

From: @simson_wu
Reviewed-by: @ginfung,@zh_qh
Signed-off-by: @zh_qh
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
7e1f2e2d61
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      mindspore/common/parameter.py

+ 6
- 0
mindspore/common/parameter.py View File

@@ -128,6 +128,12 @@ class Parameter(MetaTensor_):
self.init_in_server = False
self._unique = False
self.is_in_parallel = _is_in_parallel_mode()
if isinstance(default_input, (MetaTensor, Tensor)):
MetaTensor_.__init__(self, default_input.dtype, default_input.shape)
elif isinstance(default_input, int):
MetaTensor_.__init__(self, mstype.int64, ())
elif isinstance(default_input, float):
MetaTensor_.__init__(self, mstype.float32, ())

@staticmethod
def _get_base_class(input_class):


Loading…
Cancel
Save