mindspore.dataset.config.set_sending_batches
- mindspore.dataset.config.set_sending_batches(batch_num)
在昇腾设备中使用sink_mode=True进行训练时,设置默认的发送批次。
- 参数:
batch_num (int) - 表示总的发送批次。当设置了 batch_num 时,它将会等待,除非增加发送批次。
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)