From a08093046487b9d70dc6a6712dfec7d940593daf Mon Sep 17 00:00:00 2001
From: "zuchen.wang" <zuchen.wang@vipshop.com>
Date: Wed, 10 Nov 2021 16:56:39 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9PCB=E6=9C=80=E5=90=8E?=
=?UTF-8?q?=E4=B8=80=E4=B8=AA=E5=85=A8=E8=BF=9E=E6=8E=A5=E5=B1=82unit:=20f?=
=?UTF-8?q?rom=202=20to=201?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
fastreid/modeling/heads/pcb_head.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/fastreid/modeling/heads/pcb_head.py b/fastreid/modeling/heads/pcb_head.py
index 5c426cc..3322894 100644
--- a/fastreid/modeling/heads/pcb_head.py
+++ b/fastreid/modeling/heads/pcb_head.py
@@ -73,7 +73,7 @@ class PcbHead(nn.Module):
# Get similarity
self.match_all = nn.Sequential(
nn.Dropout(p=0.5),
- nn.Linear(self.embedding_dim * 4, 2)
+ nn.Linear(self.embedding_dim * 4, 1)
)
self.reset_parameters()