fix dataset example

4 years ago · 22e57648c8
parent 02aa0c4b0a
commit 22e57648c8
1 changed files with 1 additions and 1 deletions
--- a/mindspore/train/dataset_helper.py
+++ b/mindspore/train/dataset_helper.py
@ -155,7 +155,7 @@ class DatasetHelper:
        >>> network = Net()
        >>> net_loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean")
        >>> network = nn.WithLossCell(network, net_loss)
-        >>> train_dataset = create_custom_dataset()
+        >>> train_dataset = create_custom_dataset(sparse=True)
        >>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False)
        >>> for next_element in dataset_helper:
        ...     outputs = network(*next_element)