提交 7038df8b 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2159 add virtualDatasetCell in eval

Merge pull request !2159 from yao_yf/add_virtualdataset_in_eval
......@@ -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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册