[Minor] fix spatial_range in building kernel map (#299)

mit-han-lab · May 31, 2024 · f9ea9c0 · f9ea9c0
1 parent 462dea4
commit f9ea9c0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchsparse/nn/functional/conv/kmap/build_kmap.py b/torchsparse/nn/functional/conv/kmap/build_kmap.py
@@ -69,6 +69,7 @@ def build_kernel_map(
                 spatial_range[i + 1] + 2 * padding[i] - (kernel_size[i] - 1) - 1
             ) // stride[i] + 1
         new_spatial_range = spatial_range[:1] + tuple(new_spatial_range)
+        kmap["spatial_range"] = new_spatial_range
     else:
         new_spatial_range = None
     subm = not (any(s > 1 for s in stride))