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