mnist_torch#

Module representing the MNIST dataset in pytorch.

Functions

load_dataset([train, batch_size, shuffle, ...])

Download MNIST dataset.

load_model([pretrained, path, device])

Load MNIST model.

Classes

MnistModel

Represent a simple MNIST network.

TorchMnistDataset

MNIST Dataset.

IterableTorchMnistDataset

Iterable MNIST <http://yann.lecun.com/exdb/mnist/>`_ Dataset.