p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!17664 fix I3TGGM: set_train in step loop, performance slows down. 

From: @zhangzhaoju
Reviewed-by: @ginfung,@zh_qh
Signed-off-by: @zh_qh
This commit is contained in:
mindspore-ci-bot 2021-06-08 08:33:03 +08:00 committed by Gitee
parent 23749345f7 45172c48e6
commit f89f811dc3
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/nn/cell.py
View File

@ -1018,12 +1018,14 @@ class Cell(Cell_):
def add_flags_recursive(self, **flags):
self.add_flags(**flags)
if hasattr(self, '_cell_init_args'):
self._cell_init_args += str({**flags})
for cell in self.cells():
cell.add_flags_recursive(**flags)
return self
def _add_init_args(self, **args):
if hasattr(self, '_cell_init_args'):
self._cell_init_args += str({**args})
def get_flags(self):
if not hasattr(self, "_mindspore_flags"):
self._mindspore_flags = {}