Add auto loss logging - #27

Merged

wli51 merged 12 commits into

WayScience: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
5cbdf05
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
d891e75
Add auto loss group logging functionality to MlflowLogger
wli51
committed
7e0eae7
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
b124f32
Add tests for automatic loss-group config logging in MlflowLogger
wli51
committed
2ae005c

Commits on Apr 30, 2026