Skip to content

A federated learning demo for AlexNet on CIFAR-10 dataset, basing on Tensorflow.

License

Notifications You must be signed in to change notification settings

Zing22/tf-fed-demo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

A federated learning demo for AlexNet on CIFAR-10 dataset, basing on Tensorflow.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published