mindspore.dataset.Dataset.get_dataset_size
- mindspore.dataset.Dataset.get_dataset_size()[源代码]
返回一个epoch中的batch数。
- 返回:
int,batch的数目。
样例:
>>> 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