An implementation of the NIPS 2018 paper by Nazari et.al. https://arxiv.org/abs/1802.04240v2
- Pytorch==0.4.1
- matplotlib
- tqdm
To run the default version
python3 main.py
To run specific scenario
python3 main.py --nodes 10 --test True
Please check main.py to see the various args that can be used.