Update 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)
x[:, 0:self.in_channels] += shortcut
return x