update

2022-03-19 22:47:08 +08:00
parent 25a65ff60b
commit 9efbe03d3f
8 changed files with 1503 additions and 137 deletions
@@ -167,8 +167,8 @@ class PatchNCELoss(nn.Module):

    def forward(self, feat_q, feat_k):
        num_patches = feat_q.shape[0]
-        dim = feat_q.shape[1]
-        feat_k = feat_k.detach()
+        dim         = feat_q.shape[1]
+        feat_k      = feat_k.detach()

        # pos logit
        l_pos = torch.bmm(