About
Tools used in the project
-
Data pre-processing and analysis
-
Model Training
- Pytorch and Lightning for training
- MLflow for experiment monitoring and model registry
- Tensorboard for in-run loss and metrics monitoring
-
Documentation
- Mkdocs: to generate static documentation websites from markdown files
- Mkdocs-Material: a theme for Mkdocs
-
Miscellaneous
- Typer: for a fancy CLI
- Pydantic: for config validation
- Github Actions: for CI/CD
- Pytest: for testing some parts of the project
- Precommit: for quality checks before committing to the repo