Introduction

A generic project template based on pytorch lightning and lightning-template

Feature

See feature doc from lightning-template

Installation

See installation docs for details.

Usage

CLI

See cli doc from lightning-template

Models, Datasets and Optimizers

As pytorch lightning, we use LightningModule to implement the model and train, val and test loop, use LightningDataModule to implement dataset and dataloaders, for detail, see model doc, dataset doc and optimizer doc