This repository holds the code for baseline and federated mnist experiments. The config files for the experiments can be found in ./config folder.
Training for baseline mnist experiment can be run by the following command:
python main.py train_base.yml