zaburo-ch / pytorch-pfn-extras

Supplementary components to accelerate research and development in PyTorch

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

Documentation

• Extensions Manager
• Reporting
• Lazy Modules
• Distributed Snapshot
• Config System
• ONNX Utils

Examples

• Custom training loop
• Ignite integration

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