load checkpoint issue

Thanks for the great work! 

Why is checkpoint loading enforced at this step, which appears to invalidate the parameter updates and training progress from the prior training iteration?

<img width="1898" height="626" alt="Image" src="https://github.com/user-attachments/assets/16abacbd-745c-498b-93ff-ae439f3c11f0" />