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.
Leave a Reply