Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!14619 modify datasethepler comment 

From: @liuyang_655
Reviewed-by: @zhoufeng54,@xsmq
Signed-off-by: @xsmq
This commit is contained in:
mindspore-ci-bot 2021-04-02 17:11:43 +08:00 committed by Gitee
parent ad9c04cc4e d604334652
commit 44416db990
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mindspore/train/dataset_helper.py
View File

@ -178,15 +178,12 @@ class DatasetHelper:
epoch_num (int): Control the number of epoch data to send. Default: 1.
Examples:
>>> from mindspore import nn, DatasetHelper
>>> from mindspore import DatasetHelper
>>>
>>> network = Net()
>>> net_loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean")
>>> network = nn.WithLossCell(network, net_loss)
>>> train_dataset = create_custom_dataset()
>>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False)
>>> for next_element in dataset_helper:
... outputs = network(*next_element)
>>> set_helper = DatasetHelper(train_dataset, dataset_sink_mode=False)
>>> for next_element in set_helper:
>>> print(next_element)
"""
def __init__(self, dataset, dataset_sink_mode=True, sink_size=-1, epoch_num=1):