Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.
(channels)
| 223 | |
| 224 | |
| 225 | def normalization(channels): |
| 226 | """ |
| 227 | Make a standard normalization layer. |
| 228 | :param channels: number of input channels. |
| 229 | :return: an nn.Module for normalization. |
| 230 | """ |
| 231 | return GroupNorm32(32, channels) |
| 232 | |
| 233 | |
| 234 | # PyTorch 1.7 has SiLU, but we support PyTorch 1.5. |