bugfixed: cumsum floordiv in vectorpool

pcdet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py

@@ -215,7 +215,7 @@ class VectorPoolLocalInterpolateModule(nn.Module):
                 self.max_neighbour_distance, self.nsample, self.neighbor_type,
                 self.num_avg_length_of_neighbor_idxs, self.num_total_grids, self.neighbor_distance_multiplier
             )
-        self.num_avg_length_of_neighbor_idxs = max(self.num_avg_length_of_neighbor_idxs, num_avg_length_of_neighbor_idxs.item())
+        self.num_avg_length_of_neighbor_idxs = max(self.num_avg_length_of_neighbor_idxs, num_avg_length_of_neighbor_idxs)
 
         dist_recip = 1.0 / (dist + 1e-8)
         norm = torch.sum(dist_recip, dim=-1, keepdim=True)

pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py

@@ -337,7 +337,7 @@ class ThreeNNForVectorPoolByTwoStep(Function):
                 avg_length_of_neighbor_idxs, max_neighbour_distance * neighbor_distance_multiplier,
                 nsample, neighbor_type
             )
-            avg_length_of_neighbor_idxs = cumsum[0] // num_new_xyz + int(cumsum[0] % num_new_xyz > 0)
+            avg_length_of_neighbor_idxs = cumsum[0].item() // num_new_xyz + int(cumsum[0].item() % num_new_xyz > 0)
 
             if cumsum[0] <= num_max_sum_points:
                 break