Skip to content

GitLab

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
Show 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,6 +441,7 @@ class VectorPoolWithVoxelQuery(Function): ...@@ -441,6 +441,7 @@ 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))
if grouped_idxs.shape[0] > 0:
pointnet2.vector_pool_grad_wrapper( pointnet2.vector_pool_grad_wrapper(
grad_new_features.contiguous(), point_cnt_of_grid, grouped_idxs, grad_new_features.contiguous(), point_cnt_of_grid, grouped_idxs,
grad_support_features grad_support_features
......
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