mindspore.dataset.config.set_auto_offload

mindspore.dataset.config.set_auto_offload(offload)

设置是否开启数据异构加速。

数据异构加速可以自动将数据处理的部分运算分配到不同的异构硬件(GPU或Ascend)上,以提高数据处理的速度。

参数:
  • offload (bool) - 是否开启数据异构加速。

异常:
  • TypeError - 当 offload 的类型不为bool。

样例:

>>> # Enable automatic offload feature
>>> import mindspore.dataset as ds
>>> ds.config.set_auto_offload(True)