diff --git a/mindspore/train/model.py b/mindspore/train/model.py
index 7c58d28eac..63f3b73ddc 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: