Update demo_support/retrieval.py

demo_support/retrieval.py

@@ -41,7 +41,8 @@ def retrieve(embedding, top, sim_th=0.0, filter_fn=None):
     #for chunk in torch.split(feats, 10240):
     for chunk in feats:
         sims.append(embedding @ F.normalize(chunk.float(), dim=-1).T)
-    sims = torch.cat(sims)
+    #sims = torch.cat(sims)
+    sims = torch.tensor(sims) 
     sims, idx = torch.sort(sims, descending=True)
     sim_mask = sims > sim_th
     sims = sims[sim_mask]