联邦学习FedAvg代码

代码来源：https://github.com/shaoxiongji/federated-learning

文章结构来源：【联邦学习新手必看！】手把手教你读懂FedAvg代码并实现运行一下一些内容也是摘录此处

总览

运行

运行cifar

python main_fed.py --dataset cifar --epoch 10 --num_channel 3 --gpu 0 --model cnn --iid

运行mnist

python main_fed.py --dataset mnist --iid --num_channels 1 --model cnn --epochs 50 --gpu 0

跑non-iid

把--iid去掉就可以

查看结果

结果可以直接通过控制面板看，也会有图像形式的看