Update demo_support/retrieval.py

demo_support/retrieval.py

@@ -38,7 +38,8 @@ feats = deser['feats']
 def retrieve(embedding, top, sim_th=0.0, filter_fn=None):
     sims = []
     embedding = F.normalize(embedding.detach().cpu(), dim=-1).squeeze()
-    for chunk in torch.split(feats, 10240):
+    #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, idx = torch.sort(sims, descending=True)