git clone https://github.com/anhvth/litclassifier
pip install -e ./litclassifier

lit_trainer --help
# Demo training with simple mnist model
# To change model please take a look at `configs/00_mnist_vanila.py`
lit_trainer configs/00_mnist_vanila.py