fix psserver problem

4 years ago · 8cef90f2de
parent d6f6269ff1
commit 8cef90f2de
1 changed files with 2 additions and 0 deletions
--- 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)