From 66d67f1eb2107510aaa8c29613b03bece7b9d6a1 Mon Sep 17 00:00:00 2001 From: liuyang_655 Date: Wed, 21 Oct 2020 11:32:33 +0800 Subject: [PATCH] modify data sink mode for model.eval --- 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 0ef0a766a5..bba8a26c4e 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -680,7 +680,7 @@ class Model: self._clear_metrics() - if context.get_context("device_target") == "CPU": + if context.get_context("device_target") == "CPU" and dataset_sink_mode: dataset_sink_mode = False logger.warning("CPU cannot support dataset sink mode currently." "So the evaluating process will be performed with dataset non-sink mode.")