ml-aim icon indicating copy to clipboard operation
ml-aim copied to clipboard

AIMV2 as the encoder, unfreezing it and setting the learning rate to 1e-6 results in the LLaVA model achieving a loss of 0,grad_norm of NAN.

Open shengyuwoo opened this issue 1 year ago • 0 comments

When using AIMV2 as the encoder, unfreezing it and setting the learning rate to 1e-6 leads to the LLaVA model reaching a loss of 0 after 5000 steps. The original paper kept the encoder frozen. Why is it not recommended to unfreeze it for training? If I decide to unfreeze it, What should I do?

shengyuwoo avatar Jan 07 '25 05:01 shengyuwoo