Files
burn/examples/custom-learning-strategy
David M. 6a5b0e64ce feat(train): add training and evaluation progress logger and add them to the event processors (#4980)
* feat: add TrainingProgressLogger

* feat: add a debug implementation of TrainingProgressLogger

* feat: expose new optional progress_logger

* feat: rename trait function end_eval to end_epoch and update mnist example

* feat: add EvaluationProgressLogger trait

* fix: fix evaluation_logger override

* doc: add documentation for EvaluationProgressLogger

* feat: refactor training and evaluation logger to be more general and structured

* feat: put the new methods in full.rs pipeline and evaluation pipeline

* feat: put the new methods in full.rs pipeline and evaluation pipeline

* chore: removed boolean for FullEventTrainingProcessor

* chore: remove is_test_started from EvaluationEventPorcessor and add TestStart Event

* feat: add EndEpoch event and add TrainingProgressLogger in minimal.rs

* doc: updated doc

* doc: update documentation in pardigm.rs

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* fix: address pr comments

* feat: update method signature from new traits

* fix: change output directory in mnist and add total_test to  EvalEven::start

* feat: add EndTest event and refactor evaluation pipeline to ugrade logging

* fix: reset default values in mnist example

* feat: add new event of TrainingProgressLogger to ddp algorithm

* fix: doc test and lint
2026-05-21 15:29:57 -04:00
..