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)