mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #685 from dyhan0920/master

Browse Source 
Update rexnet.py
This commit is contained in:
Ross Wightman 2021-06-07 14:08:45 -07:00 committed by GitHub
commit 4907f8f70d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
timm/models/rexnet.py
View File

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