Skip to content

Commit

Permalink
Fixed issue unpacking size in furthest_point_sample (#248)
Browse files Browse the repository at this point in the history
The tuple unpacked into three arguments when four were given when
running VoteNet on sunrgbd
  • Loading branch information
EricWiener authored Jan 19, 2021
1 parent cda3e0e commit ca0faaf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmdet3d/ops/furthest_point_sample/furthest_point_sample.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def forward(ctx, points_xyz: torch.Tensor,
"""
assert points_xyz.is_contiguous()

B, N, _ = points_xyz.size()
B, N = points_xyz.size()[:2]
output = torch.cuda.IntTensor(B, num_points)
temp = torch.cuda.FloatTensor(B, N).fill_(1e10)

Expand Down

0 comments on commit ca0faaf

Please sign in to comment.