selameab / lightning-hydra-seed

Pytorch Lightning seed project with hydra

Pytorch Lightning seed project with hydra

Setup

pyenv install 3.7.8
pyenv local 3.7.8
python -m venv .venv
source .venv/bin/activate
python -m pip install -U pip
python -m pip install poetry
python -m poetry install

PRINCIPAL CMD

python train.py model={{MODEL}} dataset={{DATASET}} loggers={{LOGGERS}} log={{LOG}} notes={{NOTES}} name={{NAME}} jit={{JIT}}

• LOGGERS str: Configuration file to log to Wandb, currently using mine as thomas-chaton
• LOG bool: Wheter to log training to wandb
• NOTES str: A note associated to the training
• NAME str: Training name appearing on Wandb.
• LOG bool: Wheter to make model jittable.

Current Demo Command

python train.py task=categorical_classification model=simple_mlp dataset=mnist loggers=thomas-chaton log=False

python train.py task=categorical_classification model=simple_mlp dataset=mnist loggers=thomas-chaton log=False notes="My Simple MLP with Adam" notes="My Simple MLP with Adam" name="My Simple MLP with Adam" jit=False

Register your project on codecov

Set your secret key in settings with CODECOV_TOKEN