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, default is 0 which 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 >>> ds.config.set_sending_batches(10)