Add auto loss logging#27

Merged

wli51 merged 12 commits intoWayScience:mainfrom

wli51:dev-auto-loss-log

Apr 30, 2026

Commits on Apr 25, 2026

Refactored current loss group by adding 1) a currently unused but potentially useful optional parameter, epoch, that would help future addition of loss weight scheudling functionality; and 2) addin…
wli51
committed
Refactored current realizations of logging trainers to 1) pass self.epoch as parameter to modified loss group (currently not used for anything) that will be useful for the future; and 2) add loss g…
wli51
committed
Add auto loss group logging functionality to MlflowLogger
wli51
committed
Refactor test fixture by moving fixtures only under the trainer scope to global fixture, and modified in test_abstract_trainer tests to use fixtures as opposed to rely on helper class import. This …
wli51
committed
Add tests for automatic loss-group config logging in MlflowLogger
wli51
committed

Commits on Apr 30, 2026