tf-fed-demo
tf-fed-demo copied to clipboard
A federated learning demo for AlexNet on CIFAR-10 dataset, basing on Tensorflow.
tf-fed-demo
./src
: A federated learning demo for AlexNet on CIFAR-10 dataset, basing on Tensorflow.\
./src_grad
: A simple TF demo for federated LR with uploading the gradients from clients to server.
Dependence
- Python 3.7
- Tensorflow v1.14.x
- tqdm
- jupyter
Usage
Upload updated models version:
cd ./src
python Server.py
Upload model update gradients version:
cd ./src_grad
python main.py
or
cd ./src_grad
jupyter notebook
Blog
- Upload updated models: https://blog.csdn.net/Mr_Zing/article/details/101938334
- Upload model update gradients: https://blog.csdn.net/Mr_Zing/article/details/109496824