From 8cef90f2de66c841e99582117141be69058747b6 Mon Sep 17 00:00:00 2001 From: xiefangqi Date: Thu, 26 Nov 2020 16:45:01 +0800 Subject: [PATCH] fix psserver problem --- mindspore/dataset/engine/datasets.py | 2 ++ 1 file changed, 2 insertions(+) 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)