This is source code of the python implementation of the paper "CSER: Communication-efficient SGD with Error Reset".
- The following python packages needs to be installed:
- MXNet (a modified version in the folder "incubator-mxnet", read the instructions inside)
- GluonCV (a modified version in the folder "gluon-cv", read the instructions inside)
- Horovod
-
To install MXNet, please follow the instructions in README.md in the "incubator-mxnet" folder
-
To install GluonCV, run inside the folder "gluon-cv":
python3 setup.py install --user
- The detailed instructions are in README.md in the "gluon-cv" folder