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

Fix stochastic depth working only with a shortcut

Browse Source
This commit is contained in:
Dongyoon Han 2021-06-07 23:08:55 +09:00
parent 54a6cca27a
commit ded1671483
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
timm/models/rexnet.py
View File

@ -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