updates
This commit is contained in:
parent
9c716a39c3
commit
7ebb7d1310
|
@ -115,7 +115,7 @@ def create_modules(module_defs, img_size, arc):
|
||||||
|
|
||||||
class Swish(nn.Module):
|
class Swish(nn.Module):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super(Swish, self).__init__()
|
super().__init__()
|
||||||
|
|
||||||
def forward(self, x):
|
def forward(self, x):
|
||||||
return x.mul_(torch.sigmoid(x))
|
return x.mul_(torch.sigmoid(x))
|
||||||
|
|
Loading…
Reference in New Issue