Browse Source

fix psserver problem

tags/v1.1.0
xiefangqi 5 years ago
parent
commit
8cef90f2de
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/dataset/engine/datasets.py

+ 2
- 0
mindspore/dataset/engine/datasets.py 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)


Loading…
Cancel
Save