fashion-mnist example (#180)

* fashion mnist example

* fix from review
This commit is contained in:
Kashif Rasul
2023-12-23 16:34:45 +01:00
committed by GitHub
parent 848f118ac5
commit 0371d90ccb
5 changed files with 46 additions and 11 deletions

View File

@@ -14,10 +14,16 @@ Run the example with:
python main.py
```
By default the example runs on the CPU. To run on the GPU, use:
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.