From f494025db860560791630f98ca7be77346755756 Mon Sep 17 00:00:00 2001 From: chujinjin Date: Fri, 13 Nov 2020 15:31:28 +0800 Subject: [PATCH] revert pynative getnext --- mindspore/train/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 7c58d28eacf..63f3b73ddc1 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -366,8 +366,8 @@ class Model: with _CallbackManager(callbacks) as list_callback: if not dataset_sink_mode: self._train_process(epoch, train_dataset, list_callback, cb_params) - elif context.get_context("device_target") == "CPU": - logger.warning("The CPU cannot support dataset sink mode currently." + elif context.get_context("device_target") == "CPU" or context.get_context("mode") == context.PYNATIVE_MODE: + logger.warning("The CPU or PyNative mode cannot support dataset sink mode currently." "So the training process will be performed with dataset not sink.") self._train_process(epoch, train_dataset, list_callback, cb_params) else: