From 7089260e24cafc8a4de7a291b30ca61c41fa7d3d Mon Sep 17 00:00:00 2001 From: YangLuo Date: Wed, 15 Dec 2021 17:20:58 +0800 Subject: [PATCH] fix dataset_helper doc --- mindspore/train/dataset_helper.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/mindspore/train/dataset_helper.py b/mindspore/train/dataset_helper.py index 061ea2a2975..e3d8a7c0b73 100644 --- a/mindspore/train/dataset_helper.py +++ b/mindspore/train/dataset_helper.py @@ -287,7 +287,7 @@ class DatasetHelper: >>> from mindspore import DatasetHelper >>> >>> train_dataset = create_custom_dataset() - >>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False) + >>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=True) >>> >>> types, shapes = dataset_helper.types_shapes() """ @@ -324,14 +324,6 @@ class DatasetHelper: """ In sink mode, it returns the types and shapes of the current data. Generally, it works in dynamic shape scenarios. - - Examples: - >>> from mindspore import DatasetHelper - >>> - >>> train_dataset = create_custom_dataset() - >>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=True) - >>> - >>> types, shapes = dataset_helper.get_data_info() """ return self.iter.get_data_info()