Shortcuts

Examples#

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.

Notebooks#

All notebooks can be opened on Google Colab with a link:

https://colab.research.google.com/github/pytorch/ignite/blob/master/examples/notebooks/<notebook-name>

Reproducible trainings#

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

Features: