mindspore.dataset.config.set_sending_batches

mindspore.dataset.config.set_sending_batches(batch_num)

设置Host向Device发送数据的批数上限。

可用于实现自定义的数据发送控制逻辑，以解决Device内存不足的问题。在每个epoch中，当实际向Device发送的批数达到该值时， Host将停止继续发送，直到用户再次通过该接口增大这个上限。

当前仅支持在Ascend后端的下沉模式训练时使用，下沉模式可通过 mindspore.train.Model.train 接口开启。

参数：

• batch_num (int) - Host向Device发送数据的批数上限。 0 表示没有发送上限。

异常：

• TypeError - batch_num 不是int类型。

样例：

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