Training doesnt stop

[akshan-main]

Hey, I have been trying to run the qwen_image_edit finetune code, but the training doesn't stop. It keeps on running, the max_training_steps in the config doesn't work as intended

[akshan-main]

I have fixed this in my code and also added validation loss tracking. Will create a PR soon