fix bnneck

This commit is contained in:
liaoxingyu 2020-08-20 16:21:14 +08:00
parent ac8409a7da
commit f4305b0964

View File

@ -36,7 +36,6 @@ class BNneckHead(nn.Module):
See :class:`ReIDHeads.forward`.
"""
global_feat = self.pool_layer(features)
global_feat = torch.clamp(global_feat, min=0., max=1.)
bn_feat = self.bnneck(global_feat)
bn_feat = bn_feat[..., 0, 0]