Skip to content

Latest commit

 

History

History
66 lines (45 loc) · 1.96 KB

File metadata and controls

66 lines (45 loc) · 1.96 KB

TmPL: Template for Pytorch Lightning

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

A template for rapid & flexible DL experimentation development, powered by Lightning & Hydra following best practice.

Installation

pip install -r requirements.txt

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

    This framework is designed for flexibility and easy customization to meet users' specific needs. Useful comments and details on extending the framework can be found within the source code files.

Reference

Contributing

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

License

Released under the MIT License.