diff --git a/mindspore/train/dataset_helper.py b/mindspore/train/dataset_helper.py
index a95ec9d266..22f0da2ea5 100644
--- 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)