diff --git a/mindspore/train/model.py b/mindspore/train/model.py index a523d44f9a..d6b4fb332a 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -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):