mmengine.model.revert_sync_batchnorm¶
- mmengine.model.revert_sync_batchnorm(module)[source]¶
Helper function to convert all SyncBatchNorm (SyncBN) and mmcv.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