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.

previous

classification

next

load_dataset