Shortcuts

mmengine.dist.get_comm_device

mmengine.dist.get_comm_device(group=None)[source]

Return the device for communication among groups.

Parameters

group (ProcessGroup, optional) – The process group to work on.

Returns

The device of backend.

Return type

torch.device

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.