Skip to content

Latest commit

 

History

History
41 lines (30 loc) · 1.44 KB

README.md

File metadata and controls

41 lines (30 loc) · 1.44 KB

Pytorch Lightning seed project with hydra

codecov Actions Status

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