pytorch-pfn-extras
Supplementary components to accelerate research and development in PyTorch.
Installation
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
Documentation
- Extensions Manager
- Reporting
- Lazy Modules
- Distributed Snapshot
- Config System
- ONNX Utils
- CUDA Utils (CuPy Interoperability)
Examples
Contribution Guide
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.
License
MIT License