mindspore.dataset.config.set_auto_offload

mindspore.dataset.config.set_auto_offload(offload)

Set the automatic offload flag of the dataset. If set_auto_offload is True, automatically offload as many dataset operations from the CPU to the Device (GPU or Ascend).

Parameters

offload (bool) – Whether to use the automatic offload feature.

Raises

TypeError – If offload is not a boolean data type.

Examples

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