Home

Awesome

TmPL: Template for Pytorch Lightning

Lightning Docs  •  Installation  •  Usage  •  Reference  •  Contributing  •  License

A template for rapid & flexible DL experimentation development, built upon Lightning & Hydra with best practice.

What's New

v2.0 was released on Sep 5 '23.

Installation

pip install -r requirements.txt

It is recommended to manually install PyTorch and Torchvision before running the installation command, referring to the official PyTorch website for instructions.

Usage

  1. Setup

    export PYTHONPATH=`pwd`:$PYTHONPATH
    
  2. Training

    python tools/train.py [--config-name config[.yaml]] [trainer.devices=4] [data.loader.batch_size=16]
    
  3. Tips for Further Development

    The code is designed to be flexible and customizable to meet your specific needs.
    Useful comments can be found in the source code.

Reference

Contributing

Contributions are welcome and appreciated!
Feel free to open an issue or PR! 🎉

License

Released under the MIT License.