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: 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.