Skip to content

Commit

Permalink
Merge pull request open-mmlab#437 from Starrah/fix_contiguous
Browse files Browse the repository at this point in the history
fixbug: AssertionError features.is_contiguous() in pointnet2_utils.py
  • Loading branch information
jihanyang authored Jan 27, 2021
2 parents b56ab15 + ca0a950 commit 5874585
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pcdet/models/backbones_3d/pointnet2_backbone.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ def forward(self, batch_dict):

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)):
Expand Down

0 comments on commit 5874585

Please sign in to comment.