Skip to content

TODO tracker for ongoing training loop refactors #7938

Closed
@awaelchli

Description

📚 TODOs around the loop refactor

This issue tracks the work planned for and after the loop refactor #7700.
Core contributors feel free to update the description here directly.

Hygiene and code quality

  • Typing for full mypy compatibility (see ignored files in setup.cfg)
  • Documentation for the public interface + address [Refactor] Remove _run_evaluation + 3 EvaluationLoop #8065 (comment)
  • Evaluate which methods should be public and which should be protected
  • Rename Trainer.train_loop to Trainer.fit_loop and deprecate the former access.
  • Formalize how arguments are forwarded to lower loops (especially in connect

Feature integration and follow-up refactors

Customization

Miscellaneous

cc: @justusschock @carmocca @ananthsub @tchaton

Metadata

Labels

docsDocumentation relatedfeatureIs an improvement or enhancementlet's do it!approved to implementrefactor

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions