Ignite Your Networks!#
ignite is a high-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Click on the image to see complete code
Less code than pure PyTorch while ensuring maximum control and simplicity
Library approach and no program’s control inversion - Use ignite where and when you need
Extensible API for metrics, experiment managers, and other components
pip install pytorch-ignite
conda install ignite -c pytorch
pip install git+https://github.com/pytorch/ignite
pip install --pre pytorch-ignite
From conda (this suggests to install pytorch nightly release instead of stable version as dependency):
conda install ignite -c pytorch-nightly
To get started, please, read Quick start and Concepts.
- Quick start
ignite: Core of the library, contains an engine for training and evaluating, most of the classic machine learning metrics and a variety of handlers to ease the pain of training and validation of neural networks.
ignite.contrib: The contrib directory contains additional modules that can require extra dependencies. Modules vary from TBPTT engine, various optimisation parameter schedulers, experiment tracking system handlers and a metrics module containing many regression metrics.