From 22c243860b92c1c9408dd940e1de958a4a0927c3 Mon Sep 17 00:00:00 2001
From: yao_yf <yaoyifan1@huawei.com>
Date: Tue, 16 Jun 2020 11:29:45 +0800
Subject: [PATCH] add virtualdatasetcell in eval

---
 mindspore/train/model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/train/model.py b/mindspore/train/model.py
index b711cf675e..9ac49002cb 100755
--- 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):