haoyu94 / RoITr

Rotation-Invariant Transformer for Point Cloud Matching
MIT License
108 stars 13 forks source link

Batchsize > 1 broken #8

Open Parskatt opened 11 months ago

Parskatt commented 11 months ago

Hi, running the code with batchsize > 1 does not work as one might expect (for example the nearest neighbour calculation will leak over the batch). It might be good to put in an explicit error if using batch size > 1.