Class GroupNorm32

improved_diffusion/nn.py:17–19 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

15
16
17	class GroupNorm32(nn.GroupNorm):
18	def forward(self, x):
19	return super().forward(x.float()).type(x.dtype)
20
21
22	def conv_nd(dims, args, *kwargs):

normalizationFunction · 0.85

no outgoing calls

no test coverage detected