Skip to content

Commit

Permalink
fix sp ut
Browse files Browse the repository at this point in the history
  • Loading branch information
VVsssssk committed Apr 12, 2022
1 parent cc23ff5 commit 02f3f20
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/test_ops/test_spconv.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,11 +89,11 @@ def make_sparse_convmodule_forward():

# test
input_sp_tensor = SparseConvTensor(voxel_features, coordinates,
[41, 1600, 1408], 1)
[41, 1600, 1408], 2)

sparse_block0 = make_sparse_convmodule(
4,
1,
2,
3,
'test0',
stride=1,
Expand All @@ -103,7 +103,7 @@ def make_sparse_convmodule_forward():
order=('conv', 'norm', 'act')).cuda()
assert isinstance(sparse_block0[0], SubMConv3d)
assert sparse_block0[0].in_channels == 4
assert sparse_block0[0].out_channels == 1
assert sparse_block0[0].out_channels == 2
assert isinstance(sparse_block0[1], torch.nn.BatchNorm1d)
assert sparse_block0[1].eps == 0.001
assert sparse_block0[1].momentum == 0.01
Expand All @@ -115,7 +115,7 @@ def make_sparse_convmodule_forward():

sparse_block1 = make_sparse_convmodule(
4,
1,
2,
3,
'test1',
stride=1,
Expand Down

0 comments on commit 02f3f20

Please sign in to comment.