Shortcuts

mmengine.dist.get_comm_device

mmengine.dist.get_comm_device(group=None)[源代码]

Return the device for communication among groups.

参数:

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

返回:

The device of backend.

返回类型:

torch.device

Read the Docs v: stable
Versions
latest
stable
v0.10.4
v0.10.3
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
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
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.