Pytorch Lantern

Lantern contains our process of bringing a project to fruition as efficiently as possible. This is subject to change as we iterate and improve. This package implements tools and missing features to help bridge the gap between frameworks and libraries that we utilize.

The main packages and tools that we build around are:

• pytorch
• pytorch-datastream
• guild

See the documentation for more information.

Create new project with template

Install cookiecutter and poetry:

pip install cookiecutter
curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/get-poetry.py | python

Setup project:

cookiecutter https://github.com/aiwizo/pytorch-lantern-template.git
cd <new-project>
poetry install

You can now train the placeholder model and inspect the results:

guild run prepare
guild run train
guild tensorboard

Use lantern without templates

Install lantern from pypi using pip or poetry:

poetry add pytorch-lantern[training]

pip install pytorch-lantern[training]