diffusion_models.models.nn.normalization

diffusion_models.models.nn.normalization(channels)

Make a standard normalization layer.

Parameters:

channels – number of input channels.

Returns:

an nn.Module for normalization.