Skip to content

GitLab

"...utils/git@developer.sourcefind.cn:OpenDAS/lmdeploy.git" did not exist on "6939e47c9659331214f3dea76951fa481c0ebc6f"
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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/ops/furthest_point_sample/furthest_point_sample.py
View file @ ca0faaf0
...@@ -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