And you’re sure that these outputs here: https://github.com/vinits5/pcrnet_pytorch/blob/5ee2a2b5c12faa9c43c4c7ae1e503e098ecd4b46/pcrnet/losses/cuda/chamfer_distance/chamfer_distance.py#L61 38 are all floats?