How do I stop running code 

I ran your code and found that lr_anneal_steps remained unchanged while the loop could not be stopped. Can you help me with this problem? Thank you
![stop condition](https://github.com/user-attachments/assets/496285cf-f041-4487-bf0c-e0e094f44642)
