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

From: @xiefangqi
Reviewed-by: @limingqi107,@pandoublefeng
Signed-off-by: @pandoublefeng
pull/9066/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit e4d29455ee

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

Loading…
Cancel
Save