We provide several examples using ignite to display how it helps to write compact and full-featured training loops in several lines of code:

MNIST example#

Basic neural network training on MNIST dataset with/without ignite.contrib module:

These examples are ported from pytorch/examples.

Distributed examples#

Training a ResNet on CIFAR10 in various configurations:

  1. single gpu

  2. single node multiple gpus

  3. multiple nodes and multiple gpus

  4. single or multiple TPUs

Other examples#

These examples are ported from pytorch/examples.


All notebooks can be opened on Google Colab with a link:<notebook-name>

Reproducible trainings#

Inspired by torchvision/references, we provide several reproducible baselines for vision tasks: