mindspore.dataset.config.set_sending_batches

mindspore.dataset.config.set_sending_batches(batch_num)

Set the default sending batches when training with sink_mode=True in Ascend device.

Parameters

batch_num (int) – the total sending batches, when batch_num is set, it will wait unless sending batches increase, 0 means will send all batches in dataset.

Raises

TypeError – If batch_num is not of type int.

Examples

>>> # Set a new global configuration value for the sending batches
>>> import mindspore.dataset as ds
>>> ds.config.set_sending_batches(10)