mindspore.dataset.BatchInfo

class mindspore.dataset.BatchInfo[源代码]

此类提供了两种方法获取数据集的批处理数量(batch size)和迭代数(epoch)属性,这些属性可以用于 batch 操作中的输入参数 batch_sizeper_batch_map

get_batch_num()[源代码]

返回数据集的批处理数量(batch size)。

get_epoch_num()[源代码]

返回数据集的迭代数(epoch)。