2023-11-29 08:17:26 -08:00
|
|
|
# MNIST
|
|
|
|
|
2023-12-11 20:45:39 -06:00
|
|
|
This example shows how to run some simple models on MNIST.
|
|
|
|
|
|
|
|
Install the dependencies:
|
|
|
|
|
|
|
|
```
|
|
|
|
pip install -r requirements.txt
|
|
|
|
```
|
2023-11-29 08:17:26 -08:00
|
|
|
|
|
|
|
Run the example with:
|
|
|
|
|
|
|
|
```
|
|
|
|
python main.py
|
|
|
|
```
|
|
|
|
|
2023-12-23 16:34:45 +01:00
|
|
|
By default, the example runs on the CPU. To run on the GPU, use:
|
2023-11-29 08:17:26 -08:00
|
|
|
|
|
|
|
```
|
|
|
|
python main.py --gpu
|
|
|
|
```
|
|
|
|
|
2023-12-23 16:34:45 +01:00
|
|
|
For a full list of options run:
|
|
|
|
|
|
|
|
```
|
|
|
|
python main.py --help
|
|
|
|
```
|
|
|
|
|
2023-12-24 19:49:53 +05:30
|
|
|
To run the PyTorch or JAX examples install the respective framework.
|