RandLA for Point Regression Task

[asadabbas09]

Thanks for the awesome work.

I'm trying to use RandLA for a regression task.

I'm actually using the PyTorch implementation and tried replacing the final layers with linear layers but the network doesn't seem to be converging.

Do you have a min example of how to use RandLA for point regression?

Thanks