6e5f232f5c
* add MNIST DALI example, update README.md * Fix PEP8 warnings * reformatted using black * add mnist_dali to test_examples.py * Add documentation as docstrings * add nvidia-pyindex and nvidia-dali-cuda100 * replace nvidia-pyindex with --extra-index-url * mark mnist_dali test as Linux and GPU only * adjust CUDA docker and examples.txt, fix import error in test_examples.py * adjust the GPU check * Exit when DALI is not available * remove requirements-examples.txt and DALI pip install * Refactored example, moved to new logging api, added runtime check for test and dali script * Patch to reflect the mnist example module * add req. * Apply suggestions from code review * Removed requirement as it breaks CPU install, added note in README to install DALI * add DALI to Drone * test examples * Apply suggestions from code review * imports * ABC * cuda * cuda * pip DALI * Move build into init function Co-authored-by: SeanNaren <sean@grid.ai> Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Sean Naren <sean.narenthiran@gmail.com> |
||
|---|---|---|
| .. | ||
| basic_examples | ||
| domain_templates | ||
| README.md | ||
| __init__.py | ||
| bug_report_model.py | ||
| test_examples.py | ||
README.md
Examples
Our most robust examples showing all sorts of implementations can be found in our sister library PyTorch-Lightning-Bolts.
Basic examples
In this folder we add 3 simple examples:
- Image Classifier (trains arbitrary datasets with arbitrary backbones).
- MNIST classifier (defined the model inside the lightningModule).
- Autoencoder (shows how the LightningModule is meant to be used as a system)
Domain examples
This folder contains older examples. You should instead use the examples in PyTorch-Lightning-Bolts for advanced use cases.