Federated Learning - MNIST

Requirements

• kaggle
• torch
• opencv-python
• matplotlib
• numpy

Download Dataset

kaggle datasets download hojjatk/mnist-dataset
mkdir raw_data
unzip mnist-dataset.zip -d raw_data

Data Preprocess

python test_dataset_preprocess.py
python train_dataset_preprocess.py

Train

Centralized Training

python train.py

Fedrated Learning

python fedrated_learning.py

Test

python test.py