Unverified Commit bca23057 authored by kouml's avatar kouml Committed by GitHub
Browse files

Add requirements for cifar10 (#79)



* add install requirements command line

* add pillow library to fix version

* modify to uppercase
Co-authored-by: default avatarJeff Rasley <jerasley@microsoft.com>
parent 807480a0
...@@ -18,6 +18,12 @@ Original model code from [CIFAR-10 Tutorial](https://github.com/pytorch/tutorial ...@@ -18,6 +18,12 @@ Original model code from [CIFAR-10 Tutorial](https://github.com/pytorch/tutorial
git submodule update --init --recursive git submodule update --init --recursive
``` ```
To install requirements for CIFAR-10:
```
cd DeepSpeedExamples/cifar
pip install -r requirements.txt
```
Run `python cifar10_tutorial.py`, it downloads the training data set at first run. Run `python cifar10_tutorial.py`, it downloads the training data set at first run.
```less ```less
Downloading https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz to ./data/cifar-10-python.tar.gz Downloading https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz to ./data/cifar-10-python.tar.gz
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment