DeepSpeedExamples/cifar at v0.1.0 · DistributedDeepLearning/DeepSpeedExamples · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
LICENSE		LICENSE
NOTICE.txt		NOTICE.txt
README.md		README.md
cifar10_deepspeed.py		cifar10_deepspeed.py
cifar10_tutorial.py		cifar10_tutorial.py
ds_config.json		ds_config.json
requirements.txt		requirements.txt
run_ds.sh		run_ds.sh

README.md

Thanks Gopi Kumar for contributing this example, demonstrating how to apply DeepSpeed to CIFAR-10 model.

cifar10_tutorial.py Baseline CIFAR-10 model.

cifar10_deepspeed.py DeepSpeed applied CIFAR-10 model.

ds_config.json DeepSpeed configuration file.

run_ds.sh Script for running DeepSpeed applied model.

To run baseline CIFAR-10 model - "python cifar10_tutorial.py"
To run DeepSpeed CIFAR-10 model - "bash run_ds.sh"