template-pytorch-lightning-hydra-mlflow-poetry
template-pytorch-lightning-hydra-mlflow-poetry copied to clipboard
Published
20 hours ago •
hppRC
hppRC
An easy to use ML template project using Pytorch-Lightning, Hydra, Mlflow, and Poetry.
Template Pytorch-Lightning Hydra Mlflow Poetry
The code in this repository is based on pytorch/examples.
Template Design
Configurations written in yaml are in configs directory.
You can easily overwrite these configurations through command line arguments.
Instllation
poetry install
Run Training
# single run with default settings
poetry run python main.py
# single run
poetry run python main.py gpus=[0,1,2,3] batch_size=128 trainer.accelerator=ddp trainer.precision=16 optimizer=sgd scheduler.step_size=1
# multi runs
poetry run python main.py -m optimizer=adam,rmsprop,sgd trainer.precision=16,32 scheduler.step_size=1
Start Mlflow Server
poetry run mlflow ui
# access http://127.0.0.1:5000
Run Formatter
poetry run pysen run format
hppRC