提交 5519af55 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!361 Add TDT transfer data limit description

Merge pull request !361 from pengyanjun/add_tdt_transfer_data_limit_doc
...@@ -548,12 +548,16 @@ class Dataset: ...@@ -548,12 +548,16 @@ class Dataset:
def device_que(self, prefetch_size=None): def device_que(self, prefetch_size=None):
""" """
Returns a transferredDataset that transfer data through tdt. Returns a transferredDataset that transfer data through device.
Args: Args:
prefetch_size (int, optional): prefetch number of records ahead of the prefetch_size (int, optional): prefetch number of records ahead of the
user's request (default=None). 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: Return:
TransferDataset, dataset for transferring. TransferDataset, dataset for transferring.
""" """
...@@ -566,6 +570,10 @@ class Dataset: ...@@ -566,6 +570,10 @@ class Dataset:
Args: Args:
num_batch (int, optional): limit the number of batch to be sent to device (default=None). 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: Returns:
TransferDataset, dataset for transferring. TransferDataset, dataset for transferring.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册