提交 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:
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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册