Skip to content

Commit e574840

Browse files
authored
[Minor] fix spatial_range in building kernel map
1 parent 462dea4 commit e574840

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchsparse/nn/functional/conv/kmap/build_kmap.py

+1
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ def build_kernel_map(
6969
spatial_range[i + 1] + 2 * padding[i] - (kernel_size[i] - 1) - 1
7070
) // stride[i] + 1
7171
new_spatial_range = spatial_range[:1] + tuple(new_spatial_range)
72+
kmap["spatial_range"] = new_spatial_range
7273
else:
7374
new_spatial_range = None
7475
subm = not (any(s > 1 for s in stride))

0 commit comments

Comments
 (0)