Skip to content

GitLab

"tools/cfgs/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "0f4d3f1f5c1fbe551c35917220e75eb90e28035f"
Unverified Commit 846cf3ed authored by xiangruhuang's avatar xiangruhuang Committed by GitHub
Browse files

skip backprop in case no grouped indices are available (#926)

parent 878ad2e0
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py
View file @ 846cf3ed
...@@ -441,10 +441,11 @@ class VectorPoolWithVoxelQuery(Function): ...@@ -441,10 +441,11 @@ class VectorPoolWithVoxelQuery(Function):
point_cnt_of_grid, grouped_idxs, N, num_c_in = ctx.vector_pool_for_backward point_cnt_of_grid, grouped_idxs, N, num_c_in = ctx.vector_pool_for_backward
grad_support_features = grad_new_features.new_zeros((N, num_c_in)) grad_support_features = grad_new_features.new_zeros((N, num_c_in))
pointnet2.vector_pool_grad_wrapper( if grouped_idxs.shape[0] > 0:
grad_new_features.contiguous(), point_cnt_of_grid, grouped_idxs, pointnet2.vector_pool_grad_wrapper(
grad_support_features grad_new_features.contiguous(), point_cnt_of_grid, grouped_idxs,
) grad_support_features
)
return None, None, grad_support_features, None, None, None, None, None, None, None, None, None, None, None, None return None, None, grad_support_features, None, None, None, None, None, None, None, None, None, None, None, None
......
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