edits

app.py

@@ -65,7 +65,7 @@ def match(query_feat, pos_feat, LoweRatioTh=0.9):
     dist = torch.cdist(query_feat, pos_feat)
     best1 = torch.argmin(dist, dim=1)
     best2 = torch.argmin(dist, dim=0)
-    arange = torch.arange(best2.size(0), device=best2.device)
+    arange = torch.arange(best2.size(0))
     reciprocal = best1[best2]==arange
     # check Lowe ratio test
     dist2 = dist.clone()