If anyone else gets stuck - this solution worked for me: https://github.com/erikwijmans/Pointnet2_PyTorch/issues/174#issuecomment-2232300080