修改PCB最后一个全连接层unit: from 2 to 1

2021-11-10 16:56:39 +08:00 · 2021-11-10 16:56:39 +08:00 · a080930464
parent dfd7e5f61e
commit a080930464
1 changed files with 1 additions and 1 deletions
--- 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()