Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.
(channels, num_groups=32)
| 85 | |
| 86 | |
| 87 | def normalization(channels, num_groups=32): |
| 88 | """ |
| 89 | Make a standard normalization layer. |
| 90 | :param channels: number of input channels. |
| 91 | :return: an nn.Module for normalization. |
| 92 | """ |
| 93 | return GroupNormSpecific(num_groups, channels) |
| 94 | |
| 95 | |
| 96 | class HybridConditioner(nn.Module): |
no test coverage detected