finegrain-ai / refiners

A microframework on top of PyTorch with first-class citizen APIs for foundation model adaptation
https://refine.rs
MIT License
718 stars 52 forks source link

Fix clock step inconsistencies on batch end #350

Closed hugojarkoff closed 7 months ago

hugojarkoff commented 7 months ago

Previously, the logger for step end was called after incrementing the step, resulting in inconsistencies in the log messages:

Capture d’écran 2024-03-28 à 16 58 30

This PR fixes this issue:

Capture d’écran 2024-04-05 à 15 49 50