"git@developer.sourcefind.cn:OpenDAS/Uni-Core.git" did not exist on "5551a79082697d3cc26cdb7f4a06c734e16d6f3c"
Commit ca0a9505 authored by Starrah's avatar Starrah
Browse files

fix: AssertionError features.is_contiguous() in pointnet2_utils.py

caused by `features` in pointnet2_backbone.py:78, which is possible to be a non-contiguous tensor, being directly passed into SA_modules without calling .contiguous()
parent a7cf5368
......@@ -75,7 +75,7 @@ class PointNet2MSG(nn.Module):
assert xyz_batch_cnt.min() == xyz_batch_cnt.max()
xyz = xyz.view(batch_size, -1, 3)
features = features.view(batch_size, -1, features.shape[-1]).permute(0, 2, 1) if features is not None else None
features = features.view(batch_size, -1, features.shape[-1]).permute(0, 2, 1).contiguous() if features is not None else None
l_xyz, l_features = [xyz], [features]
for i in range(len(self.SA_modules)):
......
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