mindspore.dataset.BatchInfo

class mindspore.dataset.BatchInfo[source]

Only the batch size function and per_batch_map of the batch operator can dynamically adjust parameters based on the number of batches and epochs during training.

get_batch_num()[source]

Return the batch number of the current batch.

get_epoch_num()[source]

Return the epoch number of the current batch.