Swish activation function: x * sigmoid(x).
Definition at line 49 of file effnet.py.
◆ __init__()
| fastreid.modeling.backbones.regnet.effnet.Swish.__init__ |
( |
| self | ) |
|
Definition at line 52 of file effnet.py.
52 def __init__(self):
53 super(Swish, self).__init__()
54
◆ forward()
| fastreid.modeling.backbones.regnet.effnet.Swish.forward |
( |
| self, |
|
|
| x ) |
Definition at line 55 of file effnet.py.
55 def forward(self, x):
56 return x * torch.sigmoid(x)
57
58
The documentation for this class was generated from the following file:
- smreid/fastreid/modeling/backbones/regnet/effnet.py