Function Normalize

models/diffusion.py:32–33 · view source on GitHub ↗

(in_channels)

Source from the content-addressed store, hash-verified

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

__init__Method · 0.85

no outgoing calls

no test coverage detected