mmengine.model.revert_sync_batchnorm — mmengine 0.10.7 documentation (original) (raw)

• Docs >
• mmengine.model >
• mmengine.model.revert_sync_batchnorm
• 以中文阅读

Shortcuts

mmengine.model.revert_sync_batchnorm(module)[source]¶

Helper function to convert all SyncBatchNorm (SyncBN) andmmcv.ops.sync_bn.SyncBatchNorm`(MMSyncBN) layers in the model to `BatchNormXd layers.

Adapted from @kapily’s work: (https://github.com/pytorch/pytorch/issues/41081#issuecomment-783961547)

Parameters:

module (nn.Module) – The module containing SyncBatchNorm layers.

Returns:

The converted module with BatchNormXd layers.

Return type:

module_output

• mmengine.model.revert_sync_batchnorm
  • revert_sync_batchnorm()