Browse Source

!7553 Modify warning for data_sink_mode in model.eval

Merge pull request !7553 from liuyang/sink_mode
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
baa4241b82
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/model.py

+ 1
- 1
mindspore/train/model.py View File

@@ -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.")


Loading…
Cancel
Save