(num_channels, *args, **kwargs)
| 139 | def get_norm_layer(config): |
| 140 | if config.type == 'GroupNorm': |
| 141 | def NormLayer(num_channels, *args, **kwargs): |
| 142 | return torch.nn.GroupNorm( |
| 143 | num_channels=num_channels, *args, **kwargs, **config.kwargs) |
| 144 | else: |
| 145 | assert False |
| 146 | return NormLayer |
nothing calls this directly
no outgoing calls
no test coverage detected