"launch/dynamo-run/src/lib.rs" did not exist on "72064d849ebe0a00983ff9baefa639085687f699"
Unverified Commit 58745850 authored by jihan.yang's avatar jihan.yang Committed by GitHub
Browse files

Merge pull request #437 from Starrah/fix_contiguous

fixbug: AssertionError features.is_contiguous() in pointnet2_utils.py
parents b56ab159 ca0a9505
......@@ -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