!6672 [bug]fix bug for tensor store, while switch between pynative mode and graph mode

Merge pull request !6672 from vlne-v1/code_style_r1.0
pull/6672/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 00d254415a

@ -228,11 +228,12 @@ class Cell(Cell_):
cells = self.__dict__['_cells']
if name in cells:
return cells[name]
if context.get_context("mode") == context.PYNATIVE_MODE and '_params_list' in self.__dict__:
params_list = self.__dict__['_params_list']
if '_tensor_list' in self.__dict__:
tensor_list = self.__dict__['_tensor_list']
if name in tensor_list:
return self.cast_param(tensor_list[name])
if '_params_list' in self.__dict__:
params_list = self.__dict__['_params_list']
if name in params_list:
para_list = params_list[name]
cast_list = list()
@ -253,6 +254,10 @@ class Cell(Cell_):
elif name in self._cells:
del self._cells[name]
else:
if '_params_list' in self.__dict__ and name in self._params_list:
del self._params_list[name]
elif '_tensor_list' in self.__dict__ and name in self._tensor_list:
del self._tensor_list[name]
object.__delattr__(self, name)
self._attr_synced = False

Loading…
Cancel
Save