timekiller
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix python parameter del coredump if raise an error before init finished

This commit is contained in:
huanghui 2021-07-08 09:58:48 +08:00
parent 2d0fdf3904
commit 3cdcb80809
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/common/parameter.py
View File

@ -136,6 +136,7 @@ class Parameter(Tensor_):
def __init__(self, default_input, name=None, requires_grad=True, layerwise_parallel=False, parallel_optimizer=True):
self.param_info = ParamInfo()
self.init_param_info = True
self.init_in_server = False
self.cache_enable = False
self.name = name
@ -581,8 +582,9 @@ class Parameter(Tensor_):
return obj
def __del__(self):
if context.get_context("mode") == context.GRAPH_MODE:
self.param_info = None
if hasattr(self, "init_param_info"):
if self.init_param_info is True and context.get_context("mode") == context.GRAPH_MODE:
self.param_info = None
class ParameterTuple(tuple):