MCPcopy
hub / github.com/openai/improved-diffusion / SiLU

Class SiLU

improved_diffusion/nn.py:12–14  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

10
11# PyTorch 1.7 has SiLU, but we support PyTorch 1.5.
12class SiLU(nn.Module):
13 def forward(self, x):
14 return x * th.sigmoid(x)
15
16
17class GroupNorm32(nn.GroupNorm):

Callers 2

__init__Method · 0.85
__init__Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected