Focus on your deep learning experiments and forget about (re)writing code. lighter
is:
-
Task-agnostic
Whether you’re working on classification, segmentation, or self-supervised learning,
lighter
provides generalized training logic that you can use out-of-the-box. -
Configuration-based
Easily define, track, and reproduce experiments with
lighter
’s configuration-driven approach, keeping all your hyperparameters organized. -
Customizable
Seamlessly integrate your custom models, datasets, or transformations into
lighter
’s flexible framework.
lighter
stands on the shoulder of these two giants:
- MONAI Bundle - Configuration system. Similar to Hydra, but with additional features.
- PyTorch Lightning - Our
LighterSystem
is based on the PyTorch LightningLightningModule
and implements all the necessary training logic for you. Couple it with the PyTorch Lightning Trainer and you're good to go.
Simply put,
lighter = config(trainer + system)
😇Install:
pip install project-lighter
Pre-release (up-to-date with the main branch):
pip install project-lighter --pre
For development:
make setup
make install # Install lighter via Poetry
make pre-commit-install # Set up the pre-commit hook for code formatting
poetry shell # Once installed, activate the poetry shell
Projects that use lighter
:
Project | Description |
---|---|
Foundation Models for Quantitative Imaging Biomarker Discovery in Cancer Imaging | A foundation model for lesions on CT scans that can be applied to down-stream tasks related to tumor radiomics, nodule classification, etc. |
@software{lighter,
author = {Ibrahim Hadzic and
Suraj Pai and
Keno Bressem and
Hugo Aerts},
title = {Lighter},
publisher = {Zenodo},
doi = {10.5281/zenodo.8007711},
url = {https://doi.org/10.5281/zenodo.8007711}
}