set output_featuremaps_only=False in osnet

2019-08-03 21:52:52 +01:00 · 2019-08-03 21:52:52 +01:00 · d72b85a1b8
parent a08a85d8e4
commit d72b85a1b8
1 changed files with 1 additions and 1 deletions
--- a/torchreid/models/osnet.py
+++ b/torchreid/models/osnet.py
@ -292,7 +292,7 @@ class OSNet(nn.Module):
        x = self.conv5(x)
        return x

-    def forward(self, x, output_featuremaps_only=True):
+    def forward(self, x, output_featuremaps_only=False):
        x = self.featuremaps(x)
        if output_featuremaps_only:
            return x