mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
fix bnneck
This commit is contained in:
parent
ac8409a7da
commit
f4305b0964
@ -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]
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user