Browse Source

add tdt transfer data limit notice

tags/v0.2.0-alpha
Yanjun Peng 6 years ago
parent
commit
b08a90ab9b
1 changed files with 9 additions and 1 deletions
  1. +9
    -1
      mindspore/dataset/engine/datasets.py

+ 9
- 1
mindspore/dataset/engine/datasets.py View File

@@ -548,12 +548,16 @@ class Dataset:

def device_que(self, prefetch_size=None):
"""
Returns a transferredDataset that transfer data through tdt.
Returns a transferredDataset that transfer data through device.

Args:
prefetch_size (int, optional): prefetch number of records ahead of the
user's request (default=None).

Note:
If device is Ascend, features of data will be transferred one by one. The limitation
of data transferation per time is 256M.

Return:
TransferDataset, dataset for transferring.
"""
@@ -566,6 +570,10 @@ class Dataset:
Args:
num_batch (int, optional): limit the number of batch to be sent to device (default=None).

Note:
If device is Ascend, features of data will be transferred one by one. The limitation
of data transferation per time is 256M.

Returns:
TransferDataset, dataset for transferring.



Loading…
Cancel
Save