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
pip install pytorch-ignite
conda install ignite -c pytorch
pip install git+https://github.com/pytorch/ignite
- 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.