Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.
(channels)
| 91 | |
| 92 | |
| 93 | def normalization(channels): |
| 94 | """ |
| 95 | Make a standard normalization layer. |
| 96 | |
| 97 | :param channels: number of input channels. |
| 98 | :return: an nn.Module for normalization. |
| 99 | """ |
| 100 | return GroupNorm32(32, channels) |
| 101 | |
| 102 | |
| 103 | def timestep_embedding(timesteps, dim, max_period=10000): |
no test coverage detected