IRMVLab / RegFormer

[ICCV2023]RegFormer: An Efficient Projection-Aware Transformer Network for Large-Scale Point Cloud Registration
93 stars 5 forks source link

neighbor_idx out of bounds #6

Closed zlynpu closed 8 months ago

zlynpu commented 9 months ago

When I was debugging, I found that neighbor_idx would become garbled as shown in the picture. What is the reason for this? image This is in lines 549~554 of conv_util.py. And I would like to ask what is the difference between fused_conv_select_k and fused_conv_select_k_add in ops_torch. Thank you for your reply!

zlynpu commented 9 months ago

My environment is CUDA11.3, pytorch1.10.0, and the device is NVIDIA RTX3090, which is almost the same as in the README.

liujiuming123 commented 9 months ago

We haven't met the same problem that neighbor idx is out of bounds as stated. It is quite confusing. The environment is okay and 'fused_conv_select_k_add' is another version which we didn't use here. We only use 'fused_conv_select_k' for CUDA-based KNN searching.

zlynpu commented 9 months ago

We haven't met the same problem that neighbor idx is out of bounds as stated. It is quite confusing. The environment is okay and 'fused_conv_select_k_add' is another version which we didn't use here. We only use 'fused_conv_select_k' for CUDA-based KNN searching.

Okay, thanks anyway

liujiuming123 commented 8 months ago

We found that the index setting inside the CUDA operator was wrong, and have corrected it now. You can try to compile them again.

zlynpu commented 8 months ago

We found that the index setting inside the CUDA operator was wrong, and have corrected it now. You can try to compile them again.

Ok now the problem is solved, thank you very much!