Hi, I retrain stage 1 in FitDit model but seems that loss hasn't changed in training process. Any solution 