Unverified Commit ca0faaf0 authored by Eric Wiener's avatar Eric Wiener Committed by GitHub
Browse files

Fixed issue unpacking size in furthest_point_sample (#248)

The tuple unpacked into three arguments when four were given when
running VoteNet on sunrgbd
parent cda3e0eb
...@@ -25,7 +25,7 @@ class FurthestPointSampling(Function): ...@@ -25,7 +25,7 @@ class FurthestPointSampling(Function):
""" """
assert points_xyz.is_contiguous() assert points_xyz.is_contiguous()
B, N, _ = points_xyz.size() B, N = points_xyz.size()[:2]
output = torch.cuda.IntTensor(B, num_points) output = torch.cuda.IntTensor(B, num_points)
temp = torch.cuda.FloatTensor(B, N).fill_(1e10) temp = torch.cuda.FloatTensor(B, N).fill_(1e10)
......
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