BrianPugh / pytorch-lightning-template

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch Lightning Template

A quick template for getting a reasonably well organized pytorch-lightning experiment running quickly.

Usage

  1. Click the green "Use this template" button in the Github web interface.
  2. git clone your newly created repo.
  3. cd into the project, and then run python bootstrap.py
  4. Follow the onscreen prompts.

Features

  • Run python train.py --debug while developing for making everything single-threaded and as simple (but slow) as possible.
  • pre-commit and github actions for running pre-commit on PRs is setup.

About

License:Apache License 2.0


Languages

Language:Python 100.0%