dinov2 icon indicating copy to clipboard operation
dinov2 copied to clipboard

Why is ibot loss divided by 2 in the logger?

Open bkanaki opened this issue 5 months ago • 0 comments

Hello, I'm trying to understand the behaviour of the individual losses and when I look at the implementation, I see that the loss_dict is what is logging different losses for dino local crops, global crops, ibot loss and koleo loss.

In the loss dict, here the ibot loss is divided by 2. But when calculatin the gradients and when the backward call happens, the ibot loss isn't divided by 2 here and then ultimately in the total_loss here metric too, this loss is not divided by 2.

So why is this loss during metric logging divided by 2? Is this just a mistake in logging or the actual ibot loss should be divided by 2?

bkanaki avatar Sep 03 '24 05:09 bkanaki