DataParallel API
DataParallel class
keras.distribution.DataParallel(device_mesh=None, devices=None)
Distribution for data parallelism.
You can choose to create this instance by either specifying
the device_mesh or devices arguments (but not both).
The device_mesh argument is expected to be a DeviceMesh instance,
and is expected to be 1D only. In case that the mesh has multiple axes,
then the first axis will be treated as the data parallel dimension
(and a warning will be raised).
When a list of devices are provided, they will be used to construct a
1D mesh.
When both mesh and devices are absent, then list_devices()
will be used to detect any available devices and create a 1D mesh from
them.
Arguments
- device_mesh: Optional
DeviceMeshinstance. - devices: Optional list of devices.