There is a bug in inference stage

[lenyou]

In the training stage, the samplepoint func return a tensor while in the inference stage it is asked to return a tuple with point_idx and point