fix SE layer of basicblock in resnet (#375)

Summary: SE layer defined in `__init__` but not used in `forward` close #375
2021-01-04 10:46:54 +08:00 · 2021-01-04 10:46:54 +08:00 · c49414bb9f
parent 63c952f3cf
commit c49414bb9f
1 changed files with 1 additions and 0 deletions
--- a/fastreid/modeling/backbones/resnet.py
+++ b/fastreid/modeling/backbones/resnet.py
@ -65,6 +65,7 @@ class BasicBlock(nn.Module):

        out = self.conv2(out)
        out = self.bn2(out)
+        out = self.se(out)

        if self.downsample is not None:
            identity = self.downsample(x)