p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix dataset_helper doc

This commit is contained in:
YangLuo 2021-12-15 17:20:58 +08:00
parent d750fce904
commit 7089260e24
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/train/dataset_helper.py
View File

@ -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()