Shortcuts

mmengine.model.convert_sync_batchnorm

mmengine.model.convert_sync_batchnorm(module, implementation='torch')[source]

Helper function to convert all BatchNorm layers in the model to SyncBatchNorm (SyncBN) or mmcv.ops.sync_bn.SyncBatchNorm (MMSyncBN) layers. Adapted from PyTorch convert sync batchnorm.

Parameters:
  • module (nn.Module) – The module containing SyncBatchNorm layers.

  • implementation (str) –

    The type of SyncBatchNorm to convert to.

    • ’torch’: convert to torch.nn.modules.batchnorm.SyncBatchNorm.

    • ’mmcv’: convert to mmcv.ops.sync_bn.SyncBatchNorm.

Returns:

The converted module with SyncBatchNorm layers.

Return type:

nn.Module

Read the Docs v: latest
Versions
latest
stable
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.