Browse Source

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

Merge pull request !6214 from Xiaoda/22-remove-multi-subgraphs-setting
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
fd31352aba
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/model.py

+ 1
- 1
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):


Loading…
Cancel
Save