Merge pull request #685 from dyhan0920/master

Update rexnet.py
3 years ago · 4907f8f70d
parent 54a6cca27a ded1671483
commit 4907f8f70d
1 changed files with 4 additions and 3 deletions
--- a/timm/models/rexnet.py
+++ b/timm/models/rexnet.py
@ -89,10 +89,11 @@ class LinearBottleneck(nn.Module):
            x = self.se(x)
        x = self.act_dw(x)
        x = self.conv_pwl(x)
-        if self.drop_path is not None:
-            x = self.drop_path(x)
        if self.use_shortcut:
-            x[:, 0:self.in_channels] += shortcut
+            if self.drop_path is not None:
+                x = self.drop_path(x)
+            
+	    x[:, 0:self.in_channels] += shortcut
        return x