96433d03ea
* Initial changes * Add broken example for now * Fix reference * Fix format * Code runs * Fixes * Clear up files * Add tests, helpers, fixes * Small cleanups * Refactors based on review * Swap to special tests * Add special tests * Add source * Cleanups * Add logic to attach/detach model from devices * Fixes for tests * Fixes for tests * Move earlier * Cleanups * Add check for nvcc * Add tests, cleanups * Fix errors * fix * Try condition * Add missing annotation * Clearer * Clearer message * Fix variable * Cleanups * Add comment * CHANGELOG.md * Add simple selection test * Remove special=True to see what happens * Fix test * Update tests/accelerators/test_ipu.py Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com> * Convert ipu_cores -> ipus * Add typing, fail earlier * simplify precision * Add test, add helper * fix accum * Update pytorch_lightning/plugins/training_type/ipu.py Co-authored-by: thomas chaton <thomas@grid.ai> * Use stages * Make sure warning message returned * thorw error * Add more tests, use fs * add comment * Clean * Address feedback, add IPU tests * Fixes * Fix signature * Add types * Remove autoround * Add docstring * ipu_cores -> ipus * Add test, remove unnecessary precision set * Add optimizer test * Add precision back with test * Address code review * Change to probs * Move some of the asserts earlier Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com> Co-authored-by: thomas chaton <thomas@grid.ai> |
||
|---|---|---|
| .. | ||
| basic_examples | ||
| domain_templates | ||
| ipu_examples | ||
| README.md | ||
| __init__.py | ||
| bug_report_model.py | ||
| run_ddp-examples.sh | ||
| run_examples-args.sh | ||
| test_examples.py | ||
README.md
Examples
Our most robust examples showing all sorts of implementations can be found in our sister library lightning-bolts.
Basic examples
In this folder we add 3 simple examples:
- MNIST Classifier (defines the model inside the
LightningModule). - Image Classifier (trains arbitrary datasets with arbitrary backbones).
- Autoencoder (shows how the
LightningModulecan be used as a system)
Domain examples
This folder contains older examples. You should instead use the examples in lightning-bolts for advanced use cases.