Function nonlinearity

sat/sgm/modules/diffusionmodules/model.py:44–46 · view source on GitHub ↗

(x)

Source from the content-addressed store, hash-verified

42
43
44	def nonlinearity(x):
45	# swish
46	return x * torch.sigmoid(x)
47
48
49	def Normalize(in_channels, num_groups=32):

forwardMethod · 0.70

no outgoing calls

no test coverage detected