add virtualdatasetcell in eval

5 years ago · 22c243860b
parent f10e297498
commit 22c243860b
1 changed files with 2 additions and 0 deletions
--- a/mindspore/train/model.py
+++ b/mindspore/train/model.py
@ -171,6 +171,8 @@ class Model:
            self._eval_indexes = [0, 1, 2]

        if self._parallel_mode in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):
+            if self._optimizer:
+                self._eval_network = _VirtualDatasetCell(self._eval_network)
            self._eval_network.set_auto_parallel()

    def _build_predict_network(self):