We simulate the Cloud-Edge-Client FL framework.
python==3.7
pytorch==1.4.0
Federated learning with ResNet is produced by:
python main.py
See the arguments in main.py.
For example:
python main.py --edge_num=10 --client_num=100 --ratio1=0.5 --ratio2=0.3 --optim=adam --lr=0.001 --bs=128
Note: We only consider IID setting on CIFAR10.