apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2019-11-18 10:15:17 -08:00
parent 9c716a39c3
commit 7ebb7d1310
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -115,7 +115,7 @@ def create_modules(module_defs, img_size, arc):
class Swish(nn.Module):
def __init__(self):
super(Swish, self).__init__()
super().__init__()
def forward(self, x):
return x.mul_(torch.sigmoid(x))