From 22e57648c8420fccfa644eb36d34f7c6d3d1970a Mon Sep 17 00:00:00 2001 From: caozhou Date: Thu, 17 Dec 2020 19:25:14 +0800 Subject: [PATCH] fix dataset example --- mindspore/train/dataset_helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)