From 798c5f604f4e1107532f2ab1a8d8ce6d77bab768 Mon Sep 17 00:00:00 2001 From: Xiaoda Zhang Date: Mon, 14 Sep 2020 21:20:45 +0800 Subject: [PATCH] remove the multi_subgraphs setting --- mindspore/train/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index a523d44f9a7..d6b4fb332af 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):