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
-
Setup
export PYTHONPATH=`pwd`:$PYTHONPATH
-
Training
python tools/train.py [--config-name config[.yaml]] [trainer.devices=4] [data.loader.batch_size=16]
- Override the default config file with
--config-name
. - You can also override any value in the loaded config from the command line, refer to the following for more infomation.
- Override the default config file with
-
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.