mindspore.dataset.Dataset.get_dataset_size

Dataset.get_dataset_size()

Return the number of batches in an epoch.

Returns

int, number of batches.

Examples

>>> import mindspore.dataset as ds
>>> import numpy as np
>>>
>>> # A generator return 66 samples
>>> def generator1():
...     for i in range(66):
...         yield np.ones((16, 83, 83)), np.array([i])
>>>
>>> dataset = ds.GeneratorDataset(generator1, ["data1", "data2"])
>>> dataset_size = dataset.get_dataset_size()
>>> print(dataset_size)
66