Dinov2 semantic segmentation error after the first epoch

[ogencoglu]

Thanks for the Dinov2 fine-tuning example.

If you continue training until the first epoch ends, the following error occurs

Just wanted to let you know.

[NielsRogge]

Thanks for notifying, could you print the shapes of the logits and labels before calculating the loss?

[SonnetSaif]

@ogencoglu could you solve the issue? I'm also facing the same one.

@NielsRogge, will this screenshot be of any help?

[xyclone10]

Hello, this problem hasn't solved yet? Is there any solution for this? Thank you.