diff --git a/torchreid/models/osnet.py b/torchreid/models/osnet.py index 9ad753e..764c0a6 100644 --- 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