forked from mindspore-Ecosystem/mindspore
!9066 [MD][PSSERVER] Fix psserver mode problem
From: @xiefangqi Reviewed-by: @limingqi107,@pandoublefeng Signed-off-by: @pandoublefeng
This commit is contained in:
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…
Reference in New Issue