!9066 [MD][PSSERVER] Fix psserver mode problem

From: @xiefangqi
Reviewed-by: @limingqi107,@pandoublefeng
Signed-off-by: @pandoublefeng
This commit is contained in:
mindspore-ci-bot 2020-11-26 19:50:34 +08:00 committed by Gitee
commit e4d29455ee
1 changed files with 2 additions and 0 deletions

View File

@ -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)