Skip to content

pfnet/pytorch-pfn-extras

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

pytorch-pfn-extras

PyPI Docs License

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

Requirements

  • Python 3.6+
  • PyTorch 1.8+

Optional dependencies:

  • CuPy 8.0+ for PyTorch/CuPy interoperatbility

Documentation

Refer to Read The Docs for the complete documentation.

Below are some quick-links to the most important features of the library.

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