⚡ 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

0. Setup

   export PYTHONPATH=`pwd`:$PYTHONPATH

1. Training

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

   • Specify the configuration file using --config-name.
   • Refer to the following for detailed information on Hydra's override syntax.
     • https://hydra.cc/docs/advanced/hydra-command-line-flags/
     • https://hydra.cc/docs/advanced/override_grammar/basic/

2. 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

• PyTorch Lightning Docs ↗
• Hydra Docs ↗

Contributing

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

License

Released under the MIT License.