/FederatedLearningOnRaspberry

Federated Learning on Raspberry

Primary LanguagePython

FederatedLearningOnRaspberry

可部署在硬件平台上进行多机训练的同步联邦学习框架

项目介绍

  • 联邦平均算法(Fedavg)
  • MNIST数据集(独立同分布)
  • 逻辑回归模型与卷积神经网络模型(Logistic&CNN)
  • 兼顾文件传输功能(便于收集客户端上的数据)
  • 代码包含原有冗余部分未删除,部分代码块未使用
  • 测试集数据集请使用MNIST_test2.pkl,可兼容两种神经网络模型
  • 环境配置请参考environment.txt

使用须知

  • 客户端与服务器程序为client_demo_filetrans.py,server_demo_filetrans.py
  • 请先运行服务器端文件再运行客户端文件
  • 默认客户端数量为20,请在服务器端文件中修改n_nodes的值
  • 联邦学习框架各项设置在服务器端文件read_options中修改

如有任何问题,请在issues区中提出