mindspore.dataset.Dataset.get_dataset_size

Dataset.get_dataset_size()[source]

Return the number of batches in an epoch.

Returns

int, number of batches.

Examples

>>> # dataset is an instance object of Dataset
>>> dataset_size = dataset.get_dataset_size()