MCPcopy Create free account
hub / github.com/csxmli2016/DFDNet / convert_model

Function convert_model

sync_batchnorm/batchnorm.py:353–394  ·  view source on GitHub ↗

Traverse the input module and its child recursively and replace all instance of torch.nn.modules.batchnorm.BatchNorm*N*d to SynchronizedBatchNorm*N*d Args: module: the input module needs to be convert to SyncBN model Examples: >>> import torch.nn as nn

(module)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

init_netFunction · 0.90

Calls 1

Tested by

no test coverage detected