MCPcopy Create free account
hub / github.com/CompVis/diff2flow / Normalize

Function Normalize

diff2flow/kl_autoencoder.py:41–42  ·  view source on GitHub ↗
(in_channels, num_groups=32)

Source from the content-addressed store, hash-verified

39
40
41def Normalize(in_channels, num_groups=32):
42 return nn.GroupNorm(num_groups=num_groups, num_channels=in_channels, eps=1e-6, affine=True)
43
44
45class DiagonalGaussianDistribution:

Callers 4

__init__Method · 0.70
__init__Method · 0.70
__init__Method · 0.70
__init__Method · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected