You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.DatasetHelper.rst 2.2 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. mindspore.DatasetHelper
  2. ========================
  3. .. py:class:: mindspore.DatasetHelper(dataset, dataset_sink_mode=True, sink_size=-1, epoch_num=1)
  4. DatasetHelperһMindDataݼ࣬ṩݼϢ
  5. ݲͬģıݼĵڲͬʹͬĵ
  6. .. note::
  7. DatasetHelperĵṩһepochݡ
  8. ****
  9. - **dataset** (Dataset) - ѵݼݼݼAPI :class:`mindspore.dataset` ɣ :class:`mindspore.dataset.ImageFolderDataset`
  10. - **dataset_sink_mode** (bool) - ֵΪTrueʹ :class:`mindspore.ops.GetNext` 豸DeviceͨͨлȡݣֱӱݼȡݡĬֵTrue
  11. - **sink_size** (int) - ÿ³е `sink_size` Ϊ-1³ÿepochݼ `sink_size` 0³ÿepoch `sink_size` ݡĬֵ-1
  12. - **epoch_num** (int) - ƴ͵epochĬֵ1
  13. ****
  14. >>> from mindspore import DatasetHelper
  15. >>>
  16. >>> train_dataset = create_custom_dataset()
  17. >>> set_helper = DatasetHelper(train_dataset, dataset_sink_mode=False)
  18. >>> # DatasetHelperǿɵ
  19. >>> for next_element in set_helper:
  20. ... next_element
  21. .. py:method:: continue_send()
  22. epochʼʱ豸ݡ
  23. .. py:method:: dynamic_min_max_shapes()
  24. ض̬ݵ״(shape)ΧС״(shape)״(shape)
  25. .. py:method:: get_data_info()
  26. ³ģʽ£ȡǰݵͺ״(shape)ͨ״(shape)̬仯ijʹá
  27. ****
  28. >>> from mindspore import DatasetHelper
  29. >>>
  30. >>> train_dataset = create_custom_dataset()
  31. >>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=True)
  32. >>>
  33. >>> types, shapes = dataset_helper.get_data_info()
  34. .. py:method:: release()
  35. ͷ³Դ
  36. .. py:method:: sink_size()
  37. ȡÿε `sink_size`
  38. .. py:method:: stop_send()
  39. ֹͣ³ݡ
  40. .. py:method:: types_shapes()
  41. ӵǰеݼȡͺ״(shape)