Class GroupNorm32

diff2flow/models/unet/util.py:110–112 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

108
109
110	class GroupNorm32(nn.GroupNorm):
111	def forward(self, x):
112	return super().forward(x.float()).type(x.dtype)
113
114
115	def conv_nd(dims, args, *kwargs):

normalizationFunction · 0.85

no outgoing calls

no test coverage detected