MCPcopy Create free account
hub / github.com/Yasoz/DiffTraj / Normalize

Function Normalize

utils/module.py:32–36  ·  view source on GitHub ↗
(in_channels)

Source from the content-addressed store, hash-verified

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

Callers 1

__init__Method · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected