This is the Lightning Library - collection of Lightning related notebooks which are pulled back to the main repo as submodule and rendered inside the main documentations. The key features/highlights:
- we keep the repo light-weighted - notebooks are stored in rich script format
- all scripts/notebooks are tested to be fully executable
- fully reproducible by saving runtime env. details
For more details read our blogpost - Best Practices for Publishing PyTorch Lightning Tutorial Notebooks
This repo in main branch contain only python scripts with markdown extensions, and notebooks are generated in special publication branch, so no raw notebooks are accepted as PR. On the other hand we highly recommend creating a notebooks and convert it script with jupytext as
jupytext --set-formats ipynb,py:percent my-notebook.ipynb
The addition has to formed as new folder
- the folder name is used for the future notebooks
- single python scripts with converted notebooks (name does not matter)
- metadata named
.meta.yaml
including following info:title: Sample notebooks author: [User](contact) created: YYYY-MM-DD updated: YYYY-MM-DD license: CC BY-SA # multi-line description: | This notebook will walk you through ... requirements: - package # with version if needed # define supported - CPU|GPU|TPU accelerator: - CPU
It is quite common to use some public or competition's dataset for your example. We facilitate this via defining the data sources in the metafile. There are two basic options, download a file from web or pul Kaggle dataset:
datasets:
web:
- https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
kaggle:
- titanic
In both cases, the downloaded archive (Kaggle dataset is originally downloaded as zip file) is extracted to the default dataset folder under sub-folder with the same name as the downloaded file.
To get path to this dataset folder, please use environment variable PATH_DATASETS
, so in your script use:
import os
data_path = os.environ.get("PATH_DATASETS", "_datasets")
path_titanic = os.path.join(data_path, "titatnic")
Warning: some Kaggle datasets can be quite large and the process is - downloading and extracting, which means that particular runner needs to have double free space. For this reason, the CPU runner is limited to 3GB datasets.
- For inserting images into text cells use MarkDown formatting, so we can insert inline images to the notebooks directly and drop eventual dependency on internet connection -> generated notebooks could be better shared offline
- If your images need special sizes, use
![Cation](my-image.png){height="60px" width="240px"}
- If your notebook is computational or any other resource (CPU/RAM) demanding use only GPU accelerator option in meta config
- Nothing major at this moment
On the back side of publishing workflow you can find in principle these three steps
# 1) convert script to notebooks
jupytext --set-formats ipynb,py:percent notebook.py
# 2) testing the created notebook
pytest -v notebook.ipynb --nbval
# 3) generating notebooks outputs
papermill in-notebook.ipynb out-notebook.ipynb