Commit ab5e2e8e authored by yan.yan's avatar yan.yan
Browse files

v2.1.9: fix a small bug (assert isn't correct)

parent 27fb91bb
...@@ -144,7 +144,7 @@ class SparseConvTensor(metaclass=SpConvTensorMeta): ...@@ -144,7 +144,7 @@ class SparseConvTensor(metaclass=SpConvTensorMeta):
"""we need to replace x.features = F.relu(x.features) with x = x.replace_feature(F.relu(x.features)) """we need to replace x.features = F.relu(x.features) with x = x.replace_feature(F.relu(x.features))
due to limit of torch.fx due to limit of torch.fx
""" """
assert feature.shape[0] == self.indices.shape[0], "replaced num of features not equal to indices" # assert feature.shape[0] == self.indices.shape[0], "replaced num of features not equal to indices"
new_spt = SparseConvTensor(feature, self.indices, self.spatial_shape, new_spt = SparseConvTensor(feature, self.indices, self.spatial_shape,
self.batch_size, self.grid, self.voxel_num, self.batch_size, self.grid, self.voxel_num,
self.indice_dict) self.indice_dict)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment