Function Normalize

taming/modules/diffusionmodules/model.py:34–35 · view source on GitHub ↗

(in_channels)

Source from the content-addressed store, hash-verified

32
33
34	def Normalize(in_channels):
35	return torch.nn.GroupNorm(num_groups=32, num_channels=in_channels, eps=1e-6, affine=True)
36
37
38	class Upsample(nn.Module):

__init__Method · 0.70

no outgoing calls

no test coverage detected