updates

2019-11-18 10:15:17 -08:00 · 2019-11-18 10:15:17 -08:00 · 7ebb7d1310
parent 9c716a39c3
commit 7ebb7d1310
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -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))