Fix integer division error (#163)

mit-han-lab · Jul 29, 2022 · 69c1034 · 69c1034
1 parent 4fa67d2
commit 69c1034
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torchsparse/nn/functional/downsample.py b/torchsparse/nn/functional/downsample.py
@@ -24,8 +24,9 @@ def spdownsample(
 
     if all(stride[k] in [1, kernel_size[k]] for k in range(3)):
         coords = coords.clone()
-        coords[:, :3] = torch.div(coords[:, :3],
-                                  sample_stride).trunc() * sample_stride
+        coords[:, :3] = torch.div(
+            coords[:, :3],
+            sample_stride.float()).trunc() * sample_stride  # type: ignore
     else:
         offsets = get_kernel_offsets(kernel_size,
                                      tensor_stride,