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

Method forward

diff2flow/models/unet/util.py:111–112  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

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

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected