search

ywyue / AGILE3D

[ICLR 2024] AGILE3D: Attention Guided Interactive Multi-object 3D Segmentation
https://ywyue.github.io/AGILE3D/
MIT License
88 stars 5 forks source link

Training problem - list index out of range #6

Closed AshkanAlami closed 2 months ago

AshkanAlami commented 2 months ago

Hi,

Firstly, thank you for your amazing work!

I'm having a problem with the training script. When I run it, I get this error:

pos_enc = pos_encodings_pcd[hlevel][0][b]# [num_points, 128] IndexError: list index out of range

I noticed that the shape of pos_enc is (5, 1, 1). This causes the error with a batch size more than 1 (by default batch size 5). It works fine with a batch size 1.

Do you have any suggestions on how to fix this so it works with a batch size of more than 1?

Thanks!

ywyue commented 2 months ago

Hey thanks for your interests! I recently updated the code to make the interactive tool also run on CPU but it seems that it broke the training. I will fix this problem soon.

For a quick solution, could you please revert the code to the previous version: git reset --hard 843ead4? It should solve your problem. If not, let me know. Thanks.

AshkanAlami commented 2 months ago

Thanks for your quick reply! The previous version is working, thanks.