Skip to content

Latest commit

 

History

History
 
 

mnist

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 

MNIST

This example shows how to run some simple models on MNIST.

Install the dependencies:

pip install -r requirements.txt

Run the example with:

python main.py

By default, the example runs on the CPU. To run on the GPU, use:

python main.py --gpu

For a full list of options run:

python main.py --help

To run the PyTorch or JAX examples install the respective framework.