Skip to content

Commit 007054f

Browse files
committed
cifar10 experiment logs are included in the README
1 parent 065af98 commit 007054f

3 files changed

Lines changed: 7 additions & 5 deletions

File tree

README.md

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,14 +11,16 @@ scenario the clients are edge devices and the training is running in parallel.
1111

1212
In this setup the client trainings are running sequentially and you can use only your CPU or just 1 GPU.
1313

14-
## Cifar10 experiments
14+
## Cifar10 - "Shallow" VGG16
1515

16-
Training on Cifar10 with IID data where we had 100 clients and for each round (global epoch) we used only
17-
10% of them (selected randomly). Every client fitted 1 epoch on "their part" of the data with the batch size of 64.
16+
Training with a shallow version of VGG16 on Cifar10 with IID data where we had 100 clients and for each round (global epoch) we used only
17+
10% of them (selected randomly at each communication round). Every client fitted 1 epoch on "their part" of the data with the batch size of `[blue: 8, orange: 64, gray: 256]` and with learning rate of `0.1`.
1818

19-
`python fl.py -e 200 -c 100 -f 0.1 -lr 0.2 -b 64 -ce 1`
19+
A "single model" training (1 client with all the data) is also shown (`red`) on the graph. Batch size was `256` and the learning rate was: `0.05`.
2020

21-
<img src="art/fl_3_clients_accuracy.png" width="350">
21+
<img src="art/cifar_10_experiment.png" width="400">
22+
23+
(The Tensorboard logs are (for each experiment) included in the release, so you can easily visualize them.)
2224

2325
## About
2426

art/cifar_10_experiment.png

55.1 KB
Loading

art/fl_3_clients_accuracy.png

-43.5 KB
Binary file not shown.

0 commit comments

Comments
 (0)