Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
cifar10_eval.py		cifar10_eval.py
cifar10_eval_mpi4py.py		cifar10_eval_mpi4py.py
cifar10_eval_torch_dist.py		cifar10_eval_torch_dist.py

README.md

CIFAR-10 Evaluation Example

Single process evaluation

python cifar10_eval.py

Multiple processes evaluation with torch.distributed

python cifar10_eval_torch_dist.py

Multiple processes evaluation with MPI4Py

mpirun -np 3 python3 cifar10_eval_mpi4py.py