Skip to content

Latest commit

 

History

History
33 lines (27 loc) · 781 Bytes

README.md

File metadata and controls

33 lines (27 loc) · 781 Bytes

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

  1. Python 3.7
  2. Tensorflow v1.14.x
  3. tqdm
  4. 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