diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index 72fb8c45c5..276437b606 100644 --- a/mindspore/dataset/engine/datasets.py +++ b/mindspore/dataset/engine/datasets.py @@ -2726,6 +2726,8 @@ class TransferDataset(Dataset): """ Send to device """ + if Dataset._noop_mode(): + return if self._to_device is not None: del self._to_device self._to_device = _ToDevice(self, num_epochs)