Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

xinyandai/mpi-tensorflow

Repository files navigation

mpi-tensorflow

Implement MPI as tensorflow operation

setup

cd mpi-tensorflow
# build inplace
python setup.py build_ext --inplace
# install 
python setup.py install
# use system ld rather than anaconda's ld

run

using classic MPI_allreduce

mpirun -np 4 python example_distributed_allreduce_optimizer.py

compress gradient using HSQ in the paper is under review

mpirun -np 4 python example_distributed_hsq_optimizer.py

compress gradient using HSQ in the paper is under review

mpirun -np 4 python example_distributed_nn_optimizer.py

compress gradient using QSGD: Communication-Efficient SGD via Gradient Quantization and Encoding

mpirun -np 4 python example_distributed_qsgd_optimizer.py

About

😂Distributed optimizer implemented with TensorFlow MPI operation

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

AltStyle によって変換されたページ (->オリジナル) /