Hanzi Mao

Did you check the shape of your `transformed_points` and `transformed_labels`? transformed_points should be a a BxNx2 tensor and transformed_labels should be a BxN tensor. N is the number of sampled...

@calebrob6 Thank you for the contribution! Could you please rebase the pr to the latest main? I could merge it afterwards.

Please ensure that you provide the right model-type. In the first screenshot, you were using `vit_l` without specifying the model-type (the default is `vit_h`). In the second screenshot, you were...