OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

!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
This commit is contained in:
mindspore-ci-bot 2020-09-22 10:54:34 +08:00 committed by Gitee
parent c0e678af2c 0bcf4c0872
commit 00d254415a
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/nn/cell.py
View File

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