Fix stochastic depth working only with a shortcut

timm/models/rexnet.py

@@ -89,9 +89,10 @@ class LinearBottleneck(nn.Module):
             x = self.se(x)
         x = self.act_dw(x)
         x = self.conv_pwl(x)
+        if self.use_shortcut:
             if self.drop_path is not None:
                 x = self.drop_path(x)
-        if self.use_shortcut:
+            
 	    x[:, 0:self.in_channels] += shortcut
         return x