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

pip install -r requirements.txt

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.

• PyTorch Lightning Docs ↗
• Hydra Docs ↗

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

Released under the MIT License.