Supplementary components to accelerate research and development in PyTorch.
pip install pytorch-pfn-extras # Use `[onnx]` to use onnx submodule like: # pip install "pytorch-pfn-extras[onnx]" ### Optinal dependencies # For PlotReport / VariableStatisticsPlot extensions pip install matplotlib # For IgniteExtensionsManager pip install pytorch-ignite torchvision # For CuPy interoperability (see: https://docs.cupy.dev/en/stable/install.html) pip install cupy # or cupy-cudaXXX
- Python 3.6+
- PyTorch 1.9+
- CuPy 8.0+ for PyTorch/CuPy interoperatbility
Refer to Read The Docs for the complete documentation.
Below are some quick-links to the most important features of the library.
- Extensions Manager
- Lazy Modules
- Distributed Snapshot
- Config System
- ONNX Utils
- CUDA Utils (CuPy Interoperability)
You can contribute to this project by sending a pull request. After approval, the pull request will be merged by the reviewer.
Before making a contribution, please confirm that:
- Code quality stays consistent across the script, module or package.
- Code is covered by unit tests.
- API is maintainable.