From 4d3f294cb111a016624424f973a39e4939ad0cba Mon Sep 17 00:00:00 2001 From: "wangnan39@huawei.com" Date: Sat, 12 Sep 2020 11:24:43 +0800 Subject: [PATCH] optimize the example in connect_network_with_dataset --- 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 85f58590e5..93dba6239a 100644 --- a/mindspore/train/dataset_helper.py +++ b/mindspore/train/dataset_helper.py @@ -61,7 +61,7 @@ def connect_network_with_dataset(network, dataset_helper): Examples: >>> # call create_dataset function to create a regular dataset, refer to mindspore.dataset >>> train_dataset = create_dataset() - >>> dataset_helper = mindspore.DatasetHelper(train_dataset) + >>> dataset_helper = mindspore.DatasetHelper(train_dataset, dataset_sink_mode=True) >>> net = Net() >>> net_with_get_next = connect_network_with_dataset(net, dataset_helper) """