### Bug summary As shown below, eta in step 1 is obviously different from others. <img width="1695" height="1142" alt="Image" src="https://github.com/user-attachments/assets/e916f58d-8a9b-4c81-9cb8-b225c5e598f7" /> ### DeePMD-kit Version devel ### Backend and its version PyTorch 2.7 ### How did you download the software? pip ### Input Files, Running Commands, Error Log, etc. See above. ### Steps to Reproduce Training any model could reproduce the result. ### Further Information, Files, and Links cc @HydrogenSulfate