Browse Source

add virtualdatasetcell in eval

tags/v0.5.0-beta
yao_yf 5 years ago
parent
commit
22c243860b
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/train/model.py

+ 2
- 0
mindspore/train/model.py View File

@@ -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):


Loading…
Cancel
Save