From 7c01ac8c33195fd81f257c2aa0bfc341f8481155 Mon Sep 17 00:00:00 2001 From: chujinjin Date: Tue, 3 Nov 2020 15:48:18 +0800 Subject: [PATCH] pynative support 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 2dce1a9917..7c58d28eac 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("mode") == context.PYNATIVE_MODE or context.get_context("device_target") == "CPU": - logger.warning("The pynative mode and CPU cannot support dataset sink mode currently." + elif context.get_context("device_target") == "CPU": + logger.warning("The CPU 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: