diff --git a/mindspore/train/model.py b/mindspore/train/model.py
index fe655433faa..4174632cece 100755
--- a/mindspore/train/model.py
+++ b/mindspore/train/model.py
@@ -372,7 +372,7 @@ class Model:
             >>> model.train(2, dataset)
         """
         repeat_count = train_dataset.get_repeat_count()
-        if epoch != repeat_count:
+        if epoch != repeat_count and dataset_sink_mode is True:
             logger.warning(f"The epoch_size {epoch} is not the same with dataset repeat_count {repeat_count}")
         check_bool(dataset_sink_mode)
         _device_number_check(self._parallel_mode, self._device_number)