Add auto logging of trainer metadata and optimizer state saving to help with resumation of training - #39

Open

wli51 wants to merge 9 commits into

WayScience:mainfrom

wli51:resume-training

Commits on Aug 6, 2026

Add optimizer state saving functionality as new trainer contract, update existing properties by best effort inferring some. Remove some confusing and unused properties. Adjust model saving defaults.
wli51
committed
b8000e8
Implement AutoTrainerLogger for adding auto logging of certain useful trainer metadata and couple logging of model weights by logging of optimizer state to facilitate resuming of complete trainings…
wli51
committed
3c19922
Add unit tests for model and optimizer state saving functionality
wli51
committed
bd3036b

Commits on Aug 7, 2026

Fix MlflowLogger to use _log_trainer_artifact for model weight logging and add unit tests for AbstractTrainer save methods
wli51
committed
7689071

Commits on Sep 9, 2026