nn-template is opinionated so you don't have to be
Generic template to bootstrap your PyTorch project, read more in the documentation.
If you already know cookiecutter, just generate your project with:
cookiecutter https://github.com/grok-ai/nn-templateOtherwise
Cookiecutter manages the setup stages and delivers to you a personalized ready to run project.Install it with:
pip install cookiecutter
More details in the documentation.
- Actually works for research!
- Guided setup to customize project bootstrapping;
- Fast prototyping of new ideas, no need to build a new code base from scratch;
- Less boilerplate with no impact on the learning curve (as long as you know the integrated tools);
- Automatize via GitHub actions: testing, stylish documentation deploy, PyPi upload;
- Enforce Python best practices;
Avoid writing boilerplate code to integrate:
- PyTorch Lightning, lightweight PyTorch wrapper for high-performance AI research.
- Hydra, a framework for elegantly configuring complex applications.
- Weights and Biases, organize and analyze machine learning experiments. (educational account available)
- Streamlit, turns data scripts into shareable web apps in minutes.
- MkDocs and Material for MkDocs, a fast, simple and downright gorgeous static site generator.
- DVC, track large files, directories, or ML models. Think "Git for data".
- GitHub Actions, to run the tests, publish the documentation and to PyPI automatically.
- Python best practices for developing and publishing research projects.