Training MNIST benchmark with PyTorch

This notebook reproduces a simple benchmark experiment that trains a convolutional neural network based on the MNIST dataset. In particular, the neural network is built and trained with PyTorch.

MNIST is a database with 60,000 training images and 10,000 testing images that contains hand-written digits. Please see for more details.

1 1 Dec. 17, 2020, 5:45 p.m.


  • Xinyu Liu

Digital Object Identifier (DOI)

Launch on Chameleon

Launching this artifact will open it within Chameleon’s shared Jupyter experiment environment, which is accessible to all Chameleon users with an active allocation.