floor divide update - latest torch

rhysdg · Jul 12, 2022 · e8db658 · e8db658
1 parent ef6d92c
commit e8db658
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -184,7 +184,7 @@ def pred_squares(image,
     flatten_acc_map = acc_map.reshape([-1, ])
 
     scores, indices = torch.topk(flatten_acc_map, len(pts), dim=-1, largest=True)
-    yy = torch.floor_divide(indices, w).unsqueeze(-1)
+    yy = torch.div(indices, w, rounding_mode='floor').unsqueeze(-1)
     xx = torch.fmod(indices, w).unsqueeze(-1)
     yx = torch.cat((yy, xx), dim=-1)