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

!6214 [Auto parallel] Remove the 'multi_subgraphs' setting in model.py 

Merge pull request !6214 from Xiaoda/22-remove-multi-subgraphs-setting
This commit is contained in:
mindspore-ci-bot 2020-09-15 11:42:20 +08:00 committed by Gitee
parent 8b0793eb84 798c5f604f
commit fd31352aba
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/train/model.py
View File

@ -206,7 +206,7 @@ class Model:
self._predict_network = self._network
if self._parallel_mode in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
self._predict_network = _VirtualDatasetCell(self._network)
_set_multi_subgraphs()
# Unlike the cases in build_train_network() and build_eval_network(), 'multi_subgraphs' is not set
self._predict_network.set_auto_parallel()
def _clear_metrics(self):