I was fine tuning curie model for specific domain but the validation loss was nan after the first epoch How to resolve this